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
)
45 #define PSI_ALIGNOF(T) case PSI_T_## T: return ALIGNOF_## T ##_T;
58 return ALIGNOF_DOUBLE
;
60 return ALIGNOF_VOID_P
;
61 EMPTY_SWITCH_DEFAULT_CASE();
65 size_t psi_t_size(token_t t
)
67 #define PSI_SIZEOF(T) case PSI_T_## T : return SIZEOF_## T ##_T;
83 EMPTY_SWITCH_DEFAULT_CASE();
87 size_t psi_t_align(token_t t
, size_t s
)
89 size_t a
= psi_t_alignment(t
);
90 return ((s
- 1) | (a
- 1)) + 1;
93 int psi_internal_type(impl_type
*type
)
112 zend_internal_arg_info
*psi_internal_arginfo(impl
*impl
)
115 zend_internal_arg_info
*aip
;
116 zend_internal_function_info
*fi
;
118 aip
= calloc(impl
->func
->args
->count
+ 1, sizeof(*aip
));
120 fi
= (zend_internal_function_info
*) &aip
[0];
121 fi
->required_num_args
= psi_num_min_args(impl
);
122 fi
->return_reference
= impl
->func
->return_reference
;
123 fi
->type_hint
= psi_internal_type(impl
->func
->return_type
);
125 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
126 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
127 zend_internal_arg_info
*ai
= &aip
[i
+1];
129 ai
->name
= iarg
->var
->name
;
130 ai
->type_hint
= psi_internal_type(iarg
->type
);
131 if (iarg
->var
->reference
) {
132 ai
->pass_by_reference
= 1;
134 if (iarg
->var
->reference
|| (iarg
->def
&& iarg
->def
->type
== PSI_T_NULL
)) {
142 size_t psi_num_min_args(impl
*impl
)
144 size_t i
, n
= impl
->func
->args
->count
;
146 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
147 if (impl
->func
->args
->args
[i
]->def
) {
154 void psi_to_bool(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
156 psi_to_int(return_value
, set
, ret_val
);
157 convert_to_boolean(return_value
);
160 void psi_to_int(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
162 decl_var
*var
= set
->vars
->vars
[0];
163 token_t t
= real_decl_type(var
->arg
->type
)->type
;
164 impl_val
*v
= deref_impl_val(ret_val
, var
);
168 RETVAL_DOUBLE((double) v
->fval
);
169 convert_to_long(return_value
);
172 RETVAL_DOUBLE(v
->dval
);
173 convert_to_long(return_value
);
191 #if UINT32_MAX >= ZEND_LONG_MAX
192 if (v
->u32
> ZEND_LONG_MAX
) {
195 RETVAL_STRING(zend_print_ulong_to_buf(&d
[10], v
->u32
));
199 #if UINT32_MAX >= ZEND_LONG_MAX
207 if (v
->u64
> ZEND_LONG_MAX
) {
210 RETVAL_STRING(zend_print_ulong_to_buf(&d
[22], v
->u64
));
215 EMPTY_SWITCH_DEFAULT_CASE();
219 void psi_to_double(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
221 decl_var
*var
= set
->vars
->vars
[0];
222 token_t t
= real_decl_type(var
->arg
->type
)->type
;
223 impl_val
*v
= deref_impl_val(ret_val
, var
);
227 RETVAL_DOUBLE((double) v
->fval
);
230 RETVAL_DOUBLE(v
->dval
);
233 RETVAL_DOUBLE((double) v
->i8
);
236 RETVAL_DOUBLE((double) v
->u8
);
239 RETVAL_DOUBLE((double) v
->i16
);
242 RETVAL_DOUBLE((double) v
->u16
);
245 RETVAL_DOUBLE((double) v
->i32
);
248 RETVAL_DOUBLE((double) v
->u32
);
251 RETVAL_DOUBLE((double) v
->i64
);
254 RETVAL_DOUBLE((double) v
->u64
);
256 EMPTY_SWITCH_DEFAULT_CASE();
260 void psi_to_string(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
262 decl_var
*var
= set
->vars
->vars
[0];
263 token_t t
= real_decl_type(var
->arg
->type
)->type
;
268 if (!var
->arg
->var
->pointer_level
) {
269 RETVAL_STRINGL(&ret_val
->cval
, 1);
271 ret_val
= deref_impl_val(ret_val
, var
);
272 if (ret_val
&& ret_val
->ptr
) {
273 RETVAL_STRING(ret_val
->ptr
);
275 RETVAL_EMPTY_STRING();
280 RETVAL_DOUBLE((double) deref_impl_val(ret_val
, var
)->fval
);
283 RETVAL_DOUBLE(deref_impl_val(ret_val
, var
)->dval
);
286 psi_to_int(return_value
, set
, ret_val
);
289 convert_to_string(return_value
);
293 static impl_val
*iterate(impl_val
*val
, size_t size
, unsigned i
, impl_val
*tmp
)
295 memset(tmp
, 0, sizeof(*tmp
));
296 memcpy(tmp
, ((void*) val
->ptr
) + size
* i
, size
);
300 void psi_from_zval(impl_val
*mem
, decl_arg
*spec
, zval
*zv
, void **tmp
)
302 decl_type
*type
= real_decl_type(spec
->type
);
304 switch (type
->type
) {
306 mem
->fval
= (float) zval_get_double(zv
);
309 mem
->dval
= zval_get_double(zv
);
313 if (spec
->var
->pointer_level
) {
314 zend_string
*zs
= zval_get_string(zv
);
315 *tmp
= mem
->ptr
= estrndup(zs
->val
, zs
->len
);
316 zend_string_release(zs
);
321 mem
->zend
.lval
= zval_get_long(zv
);
326 void *psi_array_to_struct(decl_struct
*s
, HashTable
*arr
)
329 char *mem
= ecalloc(1, s
->size
+ s
->args
->count
* sizeof(void *));
331 if (arr
) for (i
= 0; i
< s
->args
->count
; ++i
) {
332 decl_arg
*darg
= s
->args
->args
[i
];
333 zval
*entry
= zend_hash_str_find_ind(arr
, darg
->var
->name
, strlen(darg
->var
->name
));
339 memset(&tmp
, 0, sizeof(tmp
));
340 psi_from_zval(&val
, darg
, entry
, &tmp
);
341 memcpy(mem
+ darg
->layout
->pos
, &val
, darg
->layout
->len
);
343 ((void **)(mem
+ s
->size
))[j
++] = tmp
;
350 static inline impl_val
*struct_member_ref(decl_arg
*set_arg
, impl_val
*struct_ptr
, impl_val
**to_free
) {
351 void *ptr
= (char *) struct_ptr
->ptr
+ set_arg
->layout
->pos
;
352 impl_val
*val
= enref_impl_val(ptr
, set_arg
->var
);
360 void psi_to_array(zval
*return_value
, set_value
*set
, impl_val
*r_val
)
363 decl_var
*var
= set
->vars
->vars
[0];
364 token_t t
= real_decl_type(var
->arg
->type
)->type
;
365 impl_val tmp
, *ret_val
= deref_impl_val(r_val
, var
);
367 array_init(return_value
);
369 if (t
== PSI_T_STRUCT
) {
370 // decl_struct *s = real_decl_type(var->arg->type)->strct;
373 /* explicit member casts */
374 for (i
= 0; i
< set
->count
; ++i
) {
375 set_value
*sub_set
= set
->inner
[i
];
376 decl_var
*sub_var
= sub_set
->vars
->vars
[0];
378 sub_set
->outer
.val
= r_val
;
381 impl_val
*tmp
= NULL
, *val
;
384 val
= deref_impl_val(struct_member_ref(sub_var
->arg
, ret_val
, &tmp
), sub_var
);
385 sub_set
->func
->handler(&ztmp
, sub_set
, val
);
386 add_assoc_zval(return_value
, sub_var
->name
, &ztmp
);
397 if (var
->arg
->var
->array_size
) {
398 /* to_array(foo[NUMBER]) */
399 for (i
= 0; i
< var
->arg
->var
->array_size
; ++i
) {
400 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
> 1 ? PSI_T_POINTER
: t
);
401 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
406 ZVAL_DOUBLE(&ele
, (double) ptr
->fval
);
409 ZVAL_DOUBLE(&ele
, ptr
->dval
);
412 ZVAL_LONG(&ele
, ptr
->lval
);
416 add_next_index_zval(return_value
, &ele
);
419 } else if (set
->vars
->count
> 1) {
420 /* to_array(arr_var, cnt_var[, cnt_var...], to_int(*arr_var))
421 * check for length in second var
426 if (set
->outer
.set
) {
428 for (i
= 1; i
< set
->vars
->count
; ++i
) {
429 impl_val
*tmp
= NULL
, *cnt_val
;
430 decl_var
*cnt_var
= set
->vars
->vars
[i
];
432 cnt_val
= struct_member_ref(cnt_var
->arg
, set
->outer
.val
, &tmp
);
433 count
+= deref_impl_val(cnt_val
, cnt_var
)->lval
;
443 for (i
= 0; i
< count
; ++i
) {
444 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
? PSI_T_POINTER
: t
);
445 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
447 set
->inner
[0]->func
->handler(&ele
, set
->inner
[0], ptr
);
448 add_next_index_zval(return_value
, &ele
);
456 static inline ZEND_RESULT_CODE
psi_parse_args(zend_execute_data
*execute_data
, impl
*impl
)
460 if (!impl
->func
->args
->count
) {
461 return zend_parse_parameters_none();
464 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl
), impl
->func
->args
->count
)
466 iarg
= impl
->func
->args
->args
[_i
];
470 if (PSI_T_BOOL
== iarg
->type
->type
) {
472 iarg
->val
.zend
.bval
= iarg
->def
->type
== PSI_T_TRUE
? 1 : 0;
474 Z_PARAM_BOOL(iarg
->val
.zend
.bval
);
475 } else if (PSI_T_INT
== iarg
->type
->type
) {
477 iarg
->val
.zend
.lval
= zend_atol(iarg
->def
->text
, strlen(iarg
->def
->text
));
479 Z_PARAM_LONG(iarg
->val
.zend
.lval
);
480 } else if (PSI_T_FLOAT
== iarg
->type
->type
|| PSI_T_DOUBLE
== iarg
->type
->type
) {
482 iarg
->val
.dval
= zend_strtod(iarg
->def
->text
, NULL
);
484 Z_PARAM_DOUBLE(iarg
->val
.dval
);
485 } else if (PSI_T_STRING
== iarg
->type
->type
) {
486 struct {char *val
; size_t len
;} str
;
489 str
.len
= strlen(iarg
->def
->text
) - 2;
490 str
.val
= &iarg
->def
->text
[1];
492 Z_PARAM_STR_EX(iarg
->val
.zend
.str
, 1, 0);
493 if (iarg
->val
.zend
.str
) {
494 zend_string_addref(iarg
->val
.zend
.str
);
495 } else if (iarg
->def
) {
496 iarg
->val
.zend
.str
= zend_string_init(str
.val
, str
.len
, 0);
498 } else if (PSI_T_ARRAY
== iarg
->type
->type
) {
499 /* handled as _zv in let or set */
502 error_code
= ZPP_ERROR_FAILURE
;
506 if (_i
< _max_num_args
) {
509 ZEND_PARSE_PARAMETERS_END_EX(return FAILURE
);
514 static inline void *psi_do_calloc(let_calloc
*alloc
)
516 decl_type
*type
= real_decl_type(alloc
->type
);
519 if (type
->type
== PSI_T_STRUCT
) {
520 /* psi_do_clean expects at least one NULL pointer after the struct */
521 size
= type
->strct
->size
+ sizeof(void *);
523 size
= psi_t_size(type
->type
);
526 return ecalloc(alloc
->n
, size
);
529 static inline void *psi_do_let(decl_arg
*darg
)
531 impl_arg
*iarg
= darg
->let
->arg
;
534 darg
->let
->ptr
= &darg
->let
->out
;
535 arg_val
= darg
->let
->ptr
;
538 /* let foo = calloc(1, long);
542 if (darg
->let
->val
->func
&& darg
->let
->val
->func
->type
== PSI_T_CALLOC
) {
543 arg_val
->ptr
= psi_do_calloc(darg
->let
->val
->func
->alloc
);
544 darg
->let
->mem
= arg_val
->ptr
;
545 } else if (darg
->var
->array_size
) {
546 arg_val
->ptr
= ecalloc(darg
->var
->array_size
, sizeof(*arg_val
));
547 darg
->let
->mem
= arg_val
->ptr
;
549 memset(arg_val
, 0, sizeof(*arg_val
));
553 switch (darg
->let
->val
->func
->type
) {
555 if (iarg
->type
->type
== PSI_T_BOOL
) {
556 arg_val
->cval
= iarg
->val
.zend
.bval
;
558 arg_val
->cval
= zend_is_true(iarg
->_zv
);
562 if (iarg
->type
->type
== PSI_T_INT
) {
563 arg_val
->lval
= iarg
->val
.zend
.lval
;
565 arg_val
->lval
= zval_get_long(iarg
->_zv
);
569 if (iarg
->type
->type
== PSI_T_STRING
) {
570 arg_val
->ptr
= estrdup(iarg
->val
.zend
.str
->val
);
571 darg
->let
->mem
= arg_val
->ptr
;
572 zend_string_release(iarg
->val
.zend
.str
);
574 zend_string
*zs
= zval_get_string(iarg
->_zv
);
575 arg_val
->ptr
= estrdup(zs
->val
);
576 darg
->let
->mem
= arg_val
->ptr
;
577 zend_string_release(zs
);
581 if (iarg
->type
->type
== PSI_T_STRING
) {
582 arg_val
->lval
= iarg
->val
.zend
.str
->len
;
583 zend_string_release(iarg
->val
.zend
.str
);
585 zend_string
*zs
= zval_get_string(iarg
->_zv
);
586 arg_val
->lval
= zs
->len
;
587 zend_string_release(zs
);
591 if (iarg
->type
->type
== PSI_T_ARRAY
) {
592 decl_type
*type
= real_decl_type(darg
->type
);
594 switch (type
->type
) {
596 arg_val
->ptr
= psi_array_to_struct(type
->strct
, HASH_OF(iarg
->_zv
));
597 darg
->let
->mem
= arg_val
->ptr
;
602 EMPTY_SWITCH_DEFAULT_CASE();
606 if (darg
->let
->val
&& darg
->let
->val
->is_reference
) {
607 return &darg
->let
->ptr
;
609 return darg
->let
->ptr
;
613 static inline void psi_do_set(zval
*return_value
, set_value
*set
)
615 ZVAL_DEREF(return_value
);
616 zval_dtor(return_value
);
618 set
->func
->handler(return_value
, set
, set
->vars
->vars
[0]->arg
->let
->ptr
);
621 static inline void psi_do_return(zval
*return_value
, return_stmt
*ret
, impl_val
*ret_val
)
623 ret
->set
->func
->handler(return_value
, ret
->set
, ret_val
);
626 static inline void psi_do_free(free_stmt
*fre
)
631 for (i
= 0; i
< fre
->calls
->count
; ++i
) {
632 free_call
*f
= fre
->calls
->list
[i
];
634 for (j
= 0; j
< f
->vars
->count
; ++j
) {
635 decl_var
*dvar
= f
->vars
->vars
[j
];
636 decl_arg
*darg
= dvar
->arg
;
638 f
->decl
->call
.args
[j
] = &darg
->let
->out
;
641 /* FIXME: check in validate_* that free functions return scalar */
642 PSI_ContextCall(&PSI_G(context
), &dummy
, f
->decl
);
646 static inline void psi_do_clean(impl
*impl
)
650 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
651 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
653 switch (iarg
->type
->type
) {
655 if (iarg
->val
.zend
.str
) {
656 zend_string_release(iarg
->val
.zend
.str
);
662 if (impl
->decl
->args
) for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
663 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
665 if (darg
->let
&& darg
->let
->mem
) {
666 decl_type
*type
= real_decl_type(darg
->type
);
668 if (type
->type
== PSI_T_STRUCT
) {
669 void **ptr
= (void **) ((char *) darg
->let
->mem
+ type
->strct
->size
);
675 efree(darg
->let
->mem
);
676 darg
->let
->mem
= NULL
;
681 void psi_call(zend_execute_data
*execute_data
, zval
*return_value
, impl
*impl
)
686 if (SUCCESS
!= psi_parse_args(execute_data
, impl
)) {
690 if (impl
->decl
->args
) {
691 for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
692 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
694 impl
->decl
->call
.args
[i
] = psi_do_let(darg
);
698 memset(&ret_val
, 0, sizeof(ret_val
));
699 PSI_ContextCall(&PSI_G(context
), &ret_val
, impl
->decl
);
701 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
703 for (i
= 0; i
< impl
->stmts
->set
.count
; ++i
) {
704 set_stmt
*set
= impl
->stmts
->set
.list
[i
];
707 psi_do_set(set
->arg
->_zv
, set
->val
);
711 for (i
= 0; i
< impl
->stmts
->fre
.count
; ++i
) {
712 free_stmt
*fre
= impl
->stmts
->fre
.list
[i
];
720 PHP_MINIT_FUNCTION(psi
)
722 PSI_ContextOps
*ops
= NULL
;
724 REGISTER_INI_ENTRIES();
727 if (!strcasecmp(PSI_G(engine
), "jit")) {
736 php_error(E_WARNING
, "No PSI engine found");
740 PSI_ContextInit(&PSI_G(context
), ops
, psi_error
);
741 PSI_ContextBuild(&PSI_G(context
), PSI_G(directory
));
743 if (psi_check_env("PSI_DUMP")) {
744 PSI_ContextDump(&PSI_G(context
), STDOUT_FILENO
);
749 PHP_MSHUTDOWN_FUNCTION(psi
)
751 PSI_ContextDtor(&PSI_G(context
));
753 UNREGISTER_INI_ENTRIES();
758 /* Remove if there's nothing to do at request start */
759 /* {{{ PHP_RINIT_FUNCTION
761 PHP_RINIT_FUNCTION(psi
)
763 #if defined(COMPILE_DL_PSI) && defined(ZTS)
764 ZEND_TSRMLS_CACHE_UPDATE();
770 /* Remove if there's nothing to do at request end */
771 /* {{{ PHP_RSHUTDOWN_FUNCTION
773 PHP_RSHUTDOWN_FUNCTION(psi
)
779 PHP_MINFO_FUNCTION(psi
)
781 php_info_print_table_start();
782 php_info_print_table_header(2, "psi support", "enabled");
783 php_info_print_table_end();
785 DISPLAY_INI_ENTRIES();
787 const zend_function_entry psi_functions
[] = {
791 zend_module_entry psi_module_entry
= {
792 STANDARD_MODULE_HEADER
,
797 PHP_RINIT(psi
), /* Replace with NULL if there's nothing to do at request start */
798 PHP_RSHUTDOWN(psi
), /* Replace with NULL if there's nothing to do at request end */
801 STANDARD_MODULE_PROPERTIES
804 #ifdef COMPILE_DL_PSI
806 ZEND_TSRMLS_CACHE_DEFINE();
816 * vim600: noet sw=4 ts=4 fdm=marker
817 * vim<600: noet sw=4 ts=4