8 #include "ext/standard/info.h"
9 #include "zend_exceptions.h"
17 # define PSI_ENGINE "jit"
22 # define PSI_ENGINE "ffi"
25 ZEND_DECLARE_MODULE_GLOBALS(psi
);
28 STD_PHP_INI_ENTRY("psi.engine", PSI_ENGINE
, PHP_INI_SYSTEM
, OnUpdateString
, engine
, zend_psi_globals
, psi_globals
)
29 STD_PHP_INI_ENTRY("psi.directory", "psi.d", PHP_INI_SYSTEM
, OnUpdateString
, directory
, zend_psi_globals
, psi_globals
)
32 static zend_object_handlers psi_object_handlers
;
33 static zend_class_entry
*psi_class_entry
;
35 void psi_error(int type
, const char *msg
, ...)
41 vslprintf(buf
, 0x1000, msg
, argv
);
47 size_t psi_t_alignment(token_t t
)
49 #define PSI_ALIGNOF(T) case PSI_T_## T: return ALIGNOF_## T ##_T;
62 return ALIGNOF_DOUBLE
;
64 return ALIGNOF_VOID_P
;
65 EMPTY_SWITCH_DEFAULT_CASE();
69 size_t psi_t_size(token_t t
)
71 #define PSI_SIZEOF(T) case PSI_T_## T : return SIZEOF_## T ##_T;
87 EMPTY_SWITCH_DEFAULT_CASE();
91 size_t psi_t_align(token_t t
, size_t s
)
93 size_t a
= psi_t_alignment(t
);
94 return ((s
- 1) | (a
- 1)) + 1;
97 int psi_internal_type(impl_type
*type
)
116 zend_internal_arg_info
*psi_internal_arginfo(impl
*impl
)
119 zend_internal_arg_info
*aip
;
120 zend_internal_function_info
*fi
;
122 aip
= calloc(impl
->func
->args
->count
+ 1, sizeof(*aip
));
124 fi
= (zend_internal_function_info
*) &aip
[0];
125 fi
->required_num_args
= psi_num_min_args(impl
);
126 fi
->return_reference
= impl
->func
->return_reference
;
127 fi
->type_hint
= psi_internal_type(impl
->func
->return_type
);
129 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
130 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
131 zend_internal_arg_info
*ai
= &aip
[i
+1];
133 ai
->name
= iarg
->var
->name
;
134 ai
->type_hint
= psi_internal_type(iarg
->type
);
135 if (iarg
->var
->reference
) {
136 ai
->pass_by_reference
= 1;
138 if (iarg
->var
->reference
|| (iarg
->def
&& iarg
->def
->type
== PSI_T_NULL
)) {
146 size_t psi_num_min_args(impl
*impl
)
148 size_t i
, n
= impl
->func
->args
->count
;
150 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
151 if (impl
->func
->args
->args
[i
]->def
) {
158 void psi_to_void(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
163 void psi_to_bool(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
165 psi_to_int(return_value
, set
, ret_val
);
166 convert_to_boolean(return_value
);
169 void psi_to_int(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
171 decl_var
*var
= set
->vars
->vars
[0];
172 token_t t
= real_decl_type(var
->arg
->type
)->type
;
173 impl_val
*v
= deref_impl_val(ret_val
, var
);
177 RETVAL_DOUBLE((double) v
->fval
);
178 convert_to_long(return_value
);
181 RETVAL_DOUBLE(v
->dval
);
182 convert_to_long(return_value
);
200 #if UINT32_MAX >= ZEND_LONG_MAX
201 if (v
->u32
> ZEND_LONG_MAX
) {
204 RETVAL_STRING(zend_print_ulong_to_buf(&d
[10], v
->u32
));
208 #if UINT32_MAX >= ZEND_LONG_MAX
216 if (v
->u64
> ZEND_LONG_MAX
) {
219 RETVAL_STRING(zend_print_ulong_to_buf(&d
[22], v
->u64
));
224 EMPTY_SWITCH_DEFAULT_CASE();
228 void psi_to_double(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
230 decl_var
*var
= set
->vars
->vars
[0];
231 token_t t
= real_decl_type(var
->arg
->type
)->type
;
232 impl_val
*v
= deref_impl_val(ret_val
, var
);
236 RETVAL_DOUBLE((double) v
->fval
);
239 RETVAL_DOUBLE(v
->dval
);
242 RETVAL_DOUBLE((double) v
->i8
);
245 RETVAL_DOUBLE((double) v
->u8
);
248 RETVAL_DOUBLE((double) v
->i16
);
251 RETVAL_DOUBLE((double) v
->u16
);
254 RETVAL_DOUBLE((double) v
->i32
);
257 RETVAL_DOUBLE((double) v
->u32
);
260 RETVAL_DOUBLE((double) v
->i64
);
263 RETVAL_DOUBLE((double) v
->u64
);
265 EMPTY_SWITCH_DEFAULT_CASE();
269 void psi_to_string(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
271 decl_var
*var
= set
->vars
->vars
[0];
272 token_t t
= real_decl_type(var
->arg
->type
)->type
;
278 if (!var
->arg
->var
->pointer_level
) {
279 RETVAL_STRINGL(&ret_val
->cval
, 1);
281 ret_val
= deref_impl_val(ret_val
, var
);
282 if (ret_val
&& ret_val
->ptr
) {
283 RETVAL_STRING(ret_val
->ptr
);
285 RETVAL_EMPTY_STRING();
290 RETVAL_DOUBLE((double) deref_impl_val(ret_val
, var
)->fval
);
293 RETVAL_DOUBLE(deref_impl_val(ret_val
, var
)->dval
);
296 psi_to_int(return_value
, set
, ret_val
);
299 convert_to_string(return_value
);
303 static impl_val
*iterate(impl_val
*val
, size_t size
, unsigned i
, impl_val
*tmp
)
305 memset(tmp
, 0, sizeof(*tmp
));
306 memcpy(tmp
, ((void*) val
->ptr
) + size
* i
, size
);
310 void psi_from_zval(impl_val
*mem
, decl_arg
*spec
, zval
*zv
, void **tmp
)
312 decl_type
*type
= real_decl_type(spec
->type
);
314 switch (type
->type
) {
316 mem
->fval
= (float) zval_get_double(zv
);
319 mem
->dval
= zval_get_double(zv
);
323 if (spec
->var
->pointer_level
) {
324 zend_string
*zs
= zval_get_string(zv
);
325 *tmp
= mem
->ptr
= estrndup(zs
->val
, zs
->len
);
326 zend_string_release(zs
);
331 mem
->zend
.lval
= zval_get_long(zv
);
336 void *psi_array_to_struct(decl_struct
*s
, HashTable
*arr
)
339 char *mem
= ecalloc(1, s
->size
+ s
->args
->count
* sizeof(void *));
341 if (arr
) for (i
= 0; i
< s
->args
->count
; ++i
) {
342 decl_arg
*darg
= s
->args
->args
[i
];
343 zval
*entry
= zend_hash_str_find_ind(arr
, darg
->var
->name
, strlen(darg
->var
->name
));
349 memset(&tmp
, 0, sizeof(tmp
));
350 psi_from_zval(&val
, darg
, entry
, &tmp
);
351 memcpy(mem
+ darg
->layout
->pos
, &val
, darg
->layout
->len
);
353 ((void **)(mem
+ s
->size
))[j
++] = tmp
;
360 static inline impl_val
*struct_member_ref(decl_arg
*set_arg
, impl_val
*struct_ptr
, impl_val
**to_free
) {
361 void *ptr
= (char *) struct_ptr
->ptr
+ set_arg
->layout
->pos
;
362 impl_val
*val
= enref_impl_val(ptr
, set_arg
->var
);
370 void psi_to_array(zval
*return_value
, set_value
*set
, impl_val
*r_val
)
373 decl_var
*var
= set
->vars
->vars
[0];
374 token_t t
= real_decl_type(var
->arg
->type
)->type
;
375 impl_val tmp
, *ret_val
= deref_impl_val(r_val
, var
);
377 array_init(return_value
);
379 if (t
== PSI_T_STRUCT
) {
380 // decl_struct *s = real_decl_type(var->arg->type)->strct;
383 /* explicit member casts */
384 for (i
= 0; i
< set
->count
; ++i
) {
385 set_value
*sub_set
= set
->inner
[i
];
386 decl_var
*sub_var
= sub_set
->vars
->vars
[0];
388 sub_set
->outer
.val
= r_val
;
391 impl_val
*tmp
= NULL
, *val
;
394 val
= deref_impl_val(struct_member_ref(sub_var
->arg
, ret_val
, &tmp
), sub_var
);
395 sub_set
->func
->handler(&ztmp
, sub_set
, val
);
396 add_assoc_zval(return_value
, sub_var
->name
, &ztmp
);
407 if (var
->arg
->var
->array_size
) {
408 /* to_array(foo[NUMBER]) */
409 for (i
= 0; i
< var
->arg
->var
->array_size
; ++i
) {
410 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
> 1 ? PSI_T_POINTER
: t
);
411 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
416 ZVAL_DOUBLE(&ele
, (double) ptr
->fval
);
419 ZVAL_DOUBLE(&ele
, ptr
->dval
);
422 ZVAL_LONG(&ele
, ptr
->lval
);
426 add_next_index_zval(return_value
, &ele
);
429 } else if (set
->vars
->count
> 1) {
430 /* to_array(arr_var, cnt_var[, cnt_var...], to_int(*arr_var))
431 * check for length in second var
436 if (set
->outer
.set
) {
438 for (i
= 1; i
< set
->vars
->count
; ++i
) {
439 impl_val
*tmp
= NULL
, *cnt_val
;
440 decl_var
*cnt_var
= set
->vars
->vars
[i
];
442 cnt_val
= struct_member_ref(cnt_var
->arg
, set
->outer
.val
, &tmp
);
443 count
+= deref_impl_val(cnt_val
, cnt_var
)->lval
;
453 for (i
= 0; i
< count
; ++i
) {
454 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
? PSI_T_POINTER
: t
);
455 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
457 set
->inner
[0]->func
->handler(&ele
, set
->inner
[0], ptr
);
458 add_next_index_zval(return_value
, &ele
);
465 void psi_to_object(zval
*return_value
, set_value
*set
, impl_val
*r_val
)
467 decl_var
*var
= set
->vars
->vars
[0];
468 impl_val
*ret_val
= deref_impl_val(r_val
, var
);
472 object_init_ex(return_value
, psi_class_entry
);
473 obj
= PSI_OBJ(return_value
, NULL
);
474 obj
->data
= ret_val
->ptr
;
480 static inline ZEND_RESULT_CODE
psi_parse_args(zend_execute_data
*execute_data
, impl
*impl
)
483 zend_error_handling zeh
;
485 zend_replace_error_handling(EH_THROW
, zend_exception_get_default(), &zeh
);
487 if (!impl
->func
->args
->count
) {
490 rv
= zend_parse_parameters_none();
491 zend_restore_error_handling(&zeh
);
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 */
531 Z_PARAM_ARRAY_EX(iarg
->_zv
, 1, 0);
532 } else if (PSI_T_OBJECT
== iarg
->type
->type
) {
533 Z_PARAM_OBJECT_EX(iarg
->_zv
, 1, 0);
535 error_code
= ZPP_ERROR_FAILURE
;
539 if (_i
< _max_num_args
) {
542 ZEND_PARSE_PARAMETERS_END_EX(
543 zend_restore_error_handling(&zeh
);
547 zend_restore_error_handling(&zeh
);
551 static inline void *psi_do_calloc(let_calloc
*alloc
)
553 decl_type
*type
= real_decl_type(alloc
->type
);
556 if (type
->type
== PSI_T_STRUCT
) {
557 /* psi_do_clean expects at least one NULL pointer after the struct */
558 size
= type
->strct
->size
+ sizeof(void *);
560 size
= psi_t_size(type
->type
);
563 return ecalloc(alloc
->n
, size
);
566 static inline void *psi_do_let(decl_arg
*darg
)
568 impl_arg
*iarg
= darg
->let
->arg
;
571 darg
->let
->ptr
= &darg
->let
->out
;
572 arg_val
= darg
->let
->ptr
;
575 /* let foo = calloc(1, long);
579 if (darg
->let
->val
->func
&& darg
->let
->val
->func
->type
== PSI_T_CALLOC
) {
580 arg_val
->ptr
= psi_do_calloc(darg
->let
->val
->func
->alloc
);
581 darg
->let
->mem
= arg_val
->ptr
;
582 } else if (darg
->var
->array_size
) {
583 arg_val
->ptr
= ecalloc(darg
->var
->array_size
, sizeof(*arg_val
));
584 darg
->let
->mem
= arg_val
->ptr
;
586 memset(arg_val
, 0, sizeof(*arg_val
));
590 switch (darg
->let
->val
->func
->type
) {
592 if (iarg
->type
->type
== PSI_T_BOOL
) {
593 arg_val
->cval
= iarg
->val
.zend
.bval
;
595 arg_val
->cval
= zend_is_true(iarg
->_zv
);
599 if (iarg
->type
->type
== PSI_T_INT
) {
600 arg_val
->lval
= iarg
->val
.zend
.lval
;
602 arg_val
->lval
= zval_get_long(iarg
->_zv
);
607 if (iarg
->type
->type
== PSI_T_STRING
) {
608 arg_val
->ptr
= estrdup(iarg
->val
.zend
.str
->val
);
609 darg
->let
->mem
= arg_val
->ptr
;
610 zend_string_release(iarg
->val
.zend
.str
);
612 zend_string
*zs
= zval_get_string(iarg
->_zv
);
613 arg_val
->ptr
= estrdup(zs
->val
);
614 darg
->let
->mem
= arg_val
->ptr
;
615 zend_string_release(zs
);
617 if (PSI_T_PATHVAL
== darg
->let
->val
->func
->type
) {
618 if (SUCCESS
!= php_check_open_basedir(arg_val
->ptr
)) {
625 if (iarg
->type
->type
== PSI_T_STRING
) {
626 arg_val
->lval
= iarg
->val
.zend
.str
->len
;
627 zend_string_release(iarg
->val
.zend
.str
);
629 zend_string
*zs
= zval_get_string(iarg
->_zv
);
630 arg_val
->lval
= zs
->len
;
631 zend_string_release(zs
);
635 if (iarg
->type
->type
== PSI_T_ARRAY
) {
636 decl_type
*type
= real_decl_type(darg
->type
);
638 switch (type
->type
) {
640 arg_val
->ptr
= psi_array_to_struct(type
->strct
, HASH_OF(iarg
->_zv
));
641 darg
->let
->mem
= arg_val
->ptr
;
647 if (iarg
->type
->type
== PSI_T_OBJECT
) {
650 if (!instanceof_function(Z_OBJCE_P(iarg
->_zv
), psi_class_entry
)) {
654 obj
= PSI_OBJ(iarg
->_zv
, NULL
);
655 arg_val
->ptr
= obj
->data
;
658 EMPTY_SWITCH_DEFAULT_CASE();
662 if (darg
->let
->val
&& darg
->let
->val
->is_reference
) {
663 return &darg
->let
->ptr
;
665 return darg
->let
->ptr
;
669 static inline void psi_do_set(zval
*return_value
, set_value
*set
)
671 ZVAL_DEREF(return_value
);
672 zval_dtor(return_value
);
674 set
->func
->handler(return_value
, set
, set
->vars
->vars
[0]->arg
->let
->ptr
);
677 static inline void psi_do_return(zval
*return_value
, return_stmt
*ret
, impl_val
*ret_val
)
679 ret
->set
->func
->handler(return_value
, ret
->set
, ret_val
);
682 static inline void psi_do_free(free_stmt
*fre
)
687 for (i
= 0; i
< fre
->calls
->count
; ++i
) {
688 free_call
*f
= fre
->calls
->list
[i
];
690 for (j
= 0; j
< f
->vars
->count
; ++j
) {
691 decl_var
*dvar
= f
->vars
->vars
[j
];
692 decl_arg
*darg
= dvar
->arg
;
694 f
->decl
->call
.args
[j
] = &darg
->let
->out
;
697 /* FIXME: check in validate_* that free functions return scalar */
698 PSI_ContextCall(&PSI_G(context
), &dummy
, f
->decl
);
702 static inline void psi_do_clean(impl
*impl
)
706 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
707 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
709 switch (iarg
->type
->type
) {
711 if (iarg
->val
.zend
.str
) {
712 zend_string_release(iarg
->val
.zend
.str
);
718 if (impl
->decl
->args
) for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
719 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
721 if (darg
->let
&& darg
->let
->mem
) {
722 decl_type
*type
= real_decl_type(darg
->type
);
724 if (type
->type
== PSI_T_STRUCT
) {
725 void **ptr
= (void **) ((char *) darg
->let
->mem
+ type
->strct
->size
);
731 efree(darg
->let
->mem
);
732 darg
->let
->mem
= NULL
;
737 void psi_call(zend_execute_data
*execute_data
, zval
*return_value
, impl
*impl
)
742 if (SUCCESS
!= psi_parse_args(execute_data
, impl
)) {
746 if (impl
->decl
->args
) {
747 for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
748 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
750 if (!(impl
->decl
->call
.args
[i
] = psi_do_let(darg
))) {
756 memset(&ret_val
, 0, sizeof(ret_val
));
757 PSI_ContextCall(&PSI_G(context
), &ret_val
, impl
->decl
);
759 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
761 for (i
= 0; i
< impl
->stmts
->set
.count
; ++i
) {
762 set_stmt
*set
= impl
->stmts
->set
.list
[i
];
765 psi_do_set(set
->arg
->_zv
, set
->val
);
769 for (i
= 0; i
< impl
->stmts
->fre
.count
; ++i
) {
770 free_stmt
*fre
= impl
->stmts
->fre
.list
[i
];
778 memset(&ret_val
, 0, sizeof(ret_val
));
779 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
783 static void psi_object_free(zend_object
*o
)
785 psi_object
*obj
= PSI_OBJ(NULL
, o
);
791 zend_object_std_dtor(o
);
794 static zend_object
*psi_object_init(zend_class_entry
*ce
)
796 psi_object
*o
= ecalloc(1, sizeof(*o
) + zend_object_properties_size(ce
));
798 zend_object_std_init(&o
->std
, ce
);
799 object_properties_init(&o
->std
, ce
);
800 o
->std
.handlers
= &psi_object_handlers
;
804 PHP_MINIT_FUNCTION(psi
)
806 PSI_ContextOps
*ops
= NULL
;
807 zend_class_entry ce
= {0};
809 REGISTER_INI_ENTRIES();
811 INIT_NS_CLASS_ENTRY(ce
, "psi", "object", NULL
);
812 psi_class_entry
= zend_register_internal_class_ex(&ce
, NULL
);
813 psi_class_entry
->create_object
= psi_object_init
;
815 memcpy(&psi_object_handlers
, zend_get_std_object_handlers(), sizeof(zend_object_handlers
));
816 psi_object_handlers
.offset
= XtOffsetOf(psi_object
, std
);
817 psi_object_handlers
.free_obj
= psi_object_free
;
818 psi_object_handlers
.clone_obj
= NULL
;
821 if (!strcasecmp(PSI_G(engine
), "jit")) {
830 php_error(E_WARNING
, "No PSI engine found");
834 PSI_ContextInit(&PSI_G(context
), ops
, psi_error
);
835 PSI_ContextBuild(&PSI_G(context
), PSI_G(directory
));
837 if (psi_check_env("PSI_DUMP")) {
838 PSI_ContextDump(&PSI_G(context
), STDOUT_FILENO
);
843 PHP_MSHUTDOWN_FUNCTION(psi
)
845 PSI_ContextDtor(&PSI_G(context
));
847 UNREGISTER_INI_ENTRIES();
852 /* Remove if there's nothing to do at request start */
853 /* {{{ PHP_RINIT_FUNCTION
855 PHP_RINIT_FUNCTION(psi
)
857 #if defined(COMPILE_DL_PSI) && defined(ZTS)
858 ZEND_TSRMLS_CACHE_UPDATE();
864 /* Remove if there's nothing to do at request end */
865 /* {{{ PHP_RSHUTDOWN_FUNCTION
867 PHP_RSHUTDOWN_FUNCTION(psi
)
873 PHP_MINFO_FUNCTION(psi
)
875 php_info_print_table_start();
876 php_info_print_table_header(2, "psi support", "enabled");
877 php_info_print_table_end();
879 DISPLAY_INI_ENTRIES();
881 const zend_function_entry psi_functions
[] = {
885 zend_module_entry psi_module_entry
= {
886 STANDARD_MODULE_HEADER
,
891 PHP_RINIT(psi
), /* Replace with NULL if there's nothing to do at request start */
892 PHP_RSHUTDOWN(psi
), /* Replace with NULL if there's nothing to do at request end */
895 STANDARD_MODULE_PROPERTIES
898 #ifdef COMPILE_DL_PSI
900 ZEND_TSRMLS_CACHE_DEFINE();
910 * vim600: noet sw=4 ts=4 fdm=marker
911 * vim<600: noet sw=4 ts=4