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 decl_var
*var
= set
->vars
->vars
[0];
210 token_t t
= real_decl_type(var
->arg
->type
)->type
;
211 impl_val
*v
= deref_impl_val(ret_val
, var
);
215 RETVAL_DOUBLE((double) v
->fval
);
218 RETVAL_DOUBLE(v
->dval
);
221 RETVAL_LONG(v
->lval
);
224 convert_to_boolean(return_value
);
227 void psi_to_int(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
229 decl_var
*var
= set
->vars
->vars
[0];
230 token_t t
= real_decl_type(var
->arg
->type
)->type
;
231 impl_val
*v
= deref_impl_val(ret_val
, var
);
235 RETVAL_DOUBLE((double) v
->fval
);
238 RETVAL_DOUBLE(v
->dval
);
241 RETVAL_LONG(v
->lval
);
244 convert_to_long(return_value
);
247 void psi_to_double(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
249 decl_var
*var
= set
->vars
->vars
[0];
250 token_t t
= real_decl_type(var
->arg
->type
)->type
;
251 impl_val
*v
= deref_impl_val(ret_val
, var
);
255 RETVAL_DOUBLE((double) v
->fval
);
258 RETVAL_DOUBLE(v
->dval
);
261 RETVAL_DOUBLE((double) v
->lval
);
266 void psi_to_string(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
268 decl_var
*var
= set
->vars
->vars
[0];
269 token_t t
= real_decl_type(var
->arg
->type
)->type
;
274 if (!var
->arg
->var
->pointer_level
) {
275 RETVAL_STRINGL(&ret_val
->cval
, 1);
277 ret_val
= deref_impl_val(ret_val
, var
);
278 if (ret_val
&& ret_val
->ptr
) {
279 RETVAL_STRING(ret_val
->ptr
);
281 RETVAL_EMPTY_STRING();
286 RETVAL_DOUBLE((double) deref_impl_val(ret_val
, var
)->fval
);
289 RETVAL_DOUBLE(deref_impl_val(ret_val
, var
)->dval
);
292 RETVAL_LONG(deref_impl_val(ret_val
, var
)->lval
);
295 convert_to_string(return_value
);
299 static impl_val
*iterate(impl_val
*val
, size_t size
, unsigned i
, impl_val
*tmp
)
301 memset(tmp
, 0, sizeof(*tmp
));
302 memcpy(tmp
, ((void*) val
->ptr
) + size
* i
, size
);
306 void psi_from_zval(impl_val
*mem
, decl_arg
*spec
, zval
*zv
, void **tmp
)
308 decl_type
*type
= real_decl_type(spec
->type
);
310 switch (type
->type
) {
312 mem
->fval
= (float) zval_get_double(zv
);
315 mem
->dval
= zval_get_double(zv
);
319 if (spec
->var
->pointer_level
) {
320 zend_string
*zs
= zval_get_string(zv
);
321 *tmp
= mem
->ptr
= estrndup(zs
->val
, zs
->len
);
322 zend_string_release(zs
);
327 mem
->zend
.lval
= zval_get_long(zv
);
332 void *psi_array_to_struct(decl_struct
*s
, HashTable
*arr
)
335 char *mem
= ecalloc(1, s
->size
+ s
->args
->count
* sizeof(void *));
337 if (arr
) for (i
= 0; i
< s
->args
->count
; ++i
) {
338 decl_arg
*darg
= s
->args
->args
[i
];
339 zval
*entry
= zend_hash_str_find_ind(arr
, darg
->var
->name
, strlen(darg
->var
->name
));
345 memset(&tmp
, 0, sizeof(tmp
));
346 psi_from_zval(&val
, darg
, entry
, &tmp
);
347 memcpy(mem
+ darg
->layout
->pos
, &val
, darg
->layout
->len
);
349 ((void **)(mem
+ s
->size
))[j
++] = tmp
;
356 static inline impl_val
*struct_member_ref(decl_arg
*set_arg
, impl_val
*struct_ptr
, impl_val
**to_free
) {
357 void *ptr
= (char *) struct_ptr
->ptr
+ set_arg
->layout
->pos
;
358 impl_val
*val
= enref_impl_val(ptr
, set_arg
->var
);
366 void psi_to_array(zval
*return_value
, set_value
*set
, impl_val
*r_val
)
369 decl_var
*var
= set
->vars
->vars
[0];
370 token_t t
= real_decl_type(var
->arg
->type
)->type
;
371 impl_val tmp
, *ret_val
= deref_impl_val(r_val
, var
);
373 array_init(return_value
);
375 if (t
== PSI_T_STRUCT
) {
376 // decl_struct *s = real_decl_type(var->arg->type)->strct;
379 /* explicit member casts */
380 for (i
= 0; i
< set
->count
; ++i
) {
381 set_value
*sub_set
= set
->inner
[i
];
382 decl_var
*sub_var
= sub_set
->vars
->vars
[0];
384 sub_set
->outer
.val
= r_val
;
387 impl_val
*tmp
= NULL
, *val
;
390 val
= deref_impl_val(struct_member_ref(sub_var
->arg
, ret_val
, &tmp
), sub_var
);
391 sub_set
->func
->handler(&ztmp
, sub_set
, val
);
392 add_assoc_zval(return_value
, sub_var
->name
, &ztmp
);
403 if (var
->arg
->var
->array_size
) {
404 /* to_array(foo[NUMBER]) */
405 for (i
= 0; i
< var
->arg
->var
->array_size
; ++i
) {
406 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
> 1 ? PSI_T_POINTER
: t
);
407 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
412 ZVAL_DOUBLE(&ele
, (double) ptr
->fval
);
415 ZVAL_DOUBLE(&ele
, ptr
->dval
);
418 ZVAL_LONG(&ele
, ptr
->lval
);
422 add_next_index_zval(return_value
, &ele
);
425 } else if (set
->vars
->count
> 1) {
426 /* to_array(arr_var, cnt_var[, cnt_var...], to_int(*arr_var))
427 * check for length in second var
432 if (set
->outer
.set
) {
434 for (i
= 1; i
< set
->vars
->count
; ++i
) {
435 impl_val
*tmp
= NULL
, *cnt_val
;
436 decl_var
*cnt_var
= set
->vars
->vars
[i
];
438 cnt_val
= struct_member_ref(cnt_var
->arg
, set
->outer
.val
, &tmp
);
439 count
+= deref_impl_val(cnt_val
, cnt_var
)->lval
;
449 for (i
= 0; i
< count
; ++i
) {
450 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
? PSI_T_POINTER
: t
);
451 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
453 set
->inner
[0]->func
->handler(&ele
, set
->inner
[0], ptr
);
454 add_next_index_zval(return_value
, &ele
);
462 static inline ZEND_RESULT_CODE
psi_parse_args(zend_execute_data
*execute_data
, impl
*impl
)
466 if (!impl
->func
->args
->count
) {
467 return zend_parse_parameters_none();
470 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl
), impl
->func
->args
->count
)
472 iarg
= impl
->func
->args
->args
[_i
];
476 if (PSI_T_BOOL
== iarg
->type
->type
) {
478 iarg
->val
.zend
.bval
= iarg
->def
->type
== PSI_T_TRUE
? 1 : 0;
480 Z_PARAM_BOOL(iarg
->val
.zend
.bval
);
481 } else if (PSI_T_INT
== iarg
->type
->type
) {
483 iarg
->val
.zend
.lval
= zend_atol(iarg
->def
->text
, strlen(iarg
->def
->text
));
485 Z_PARAM_LONG(iarg
->val
.zend
.lval
);
486 } else if (PSI_T_FLOAT
== iarg
->type
->type
|| PSI_T_DOUBLE
== iarg
->type
->type
) {
488 iarg
->val
.dval
= zend_strtod(iarg
->def
->text
, NULL
);
490 Z_PARAM_DOUBLE(iarg
->val
.dval
);
491 } else if (PSI_T_STRING
== iarg
->type
->type
) {
492 struct {char *val
; size_t len
;} str
;
495 str
.len
= strlen(iarg
->def
->text
) - 2;
496 str
.val
= &iarg
->def
->text
[1];
498 Z_PARAM_STR_EX(iarg
->val
.zend
.str
, 1, 0);
499 if (iarg
->val
.zend
.str
) {
500 zend_string_addref(iarg
->val
.zend
.str
);
501 } else if (iarg
->def
) {
502 iarg
->val
.zend
.str
= zend_string_init(str
.val
, str
.len
, 0);
504 } else if (PSI_T_ARRAY
== iarg
->type
->type
) {
505 /* handled as _zv in let or set */
508 error_code
= ZPP_ERROR_FAILURE
;
512 if (_i
< _max_num_args
) {
515 ZEND_PARSE_PARAMETERS_END_EX(return FAILURE
);
520 static inline void *psi_do_calloc(let_calloc
*alloc
)
522 decl_type
*type
= real_decl_type(alloc
->type
);
525 if (type
->type
== PSI_T_STRUCT
) {
526 /* psi_do_clean expects at least one NULL pointer after the struct */
527 size
= type
->strct
->size
+ sizeof(void *);
529 size
= psi_t_size(type
->type
);
532 return ecalloc(alloc
->n
, size
);
535 static inline void *psi_do_let(decl_arg
*darg
)
537 impl_arg
*iarg
= darg
->let
->arg
;
540 darg
->let
->ptr
= &darg
->let
->out
;
541 arg_val
= darg
->let
->ptr
;
544 /* let foo = calloc(1, long);
548 if (darg
->let
->val
->func
&& darg
->let
->val
->func
->type
== PSI_T_CALLOC
) {
549 arg_val
->ptr
= psi_do_calloc(darg
->let
->val
->func
->alloc
);
550 darg
->let
->mem
= arg_val
->ptr
;
551 } else if (darg
->var
->array_size
) {
552 arg_val
->ptr
= ecalloc(darg
->var
->array_size
, sizeof(*arg_val
));
553 darg
->let
->mem
= arg_val
->ptr
;
555 memset(arg_val
, 0, sizeof(*arg_val
));
559 switch (darg
->let
->val
->func
->type
) {
561 if (iarg
->type
->type
== PSI_T_BOOL
) {
562 arg_val
->cval
= iarg
->val
.zend
.bval
;
564 arg_val
->cval
= zend_is_true(iarg
->_zv
);
568 if (iarg
->type
->type
== PSI_T_INT
) {
569 arg_val
->lval
= iarg
->val
.zend
.lval
;
571 arg_val
->lval
= zval_get_long(iarg
->_zv
);
575 if (iarg
->type
->type
== PSI_T_STRING
) {
576 arg_val
->ptr
= estrdup(iarg
->val
.zend
.str
->val
);
577 darg
->let
->mem
= arg_val
->ptr
;
578 zend_string_release(iarg
->val
.zend
.str
);
580 zend_string
*zs
= zval_get_string(iarg
->_zv
);
581 arg_val
->ptr
= estrdup(zs
->val
);
582 darg
->let
->mem
= arg_val
->ptr
;
583 zend_string_release(zs
);
587 if (iarg
->type
->type
== PSI_T_STRING
) {
588 arg_val
->lval
= iarg
->val
.zend
.str
->len
;
589 zend_string_release(iarg
->val
.zend
.str
);
591 zend_string
*zs
= zval_get_string(iarg
->_zv
);
592 arg_val
->lval
= zs
->len
;
593 zend_string_release(zs
);
597 if (iarg
->type
->type
== PSI_T_ARRAY
) {
598 decl_type
*type
= real_decl_type(darg
->type
);
600 switch (type
->type
) {
602 arg_val
->ptr
= psi_array_to_struct(type
->strct
, HASH_OF(iarg
->_zv
));
603 darg
->let
->mem
= arg_val
->ptr
;
608 EMPTY_SWITCH_DEFAULT_CASE();
612 if (darg
->let
->val
&& darg
->let
->val
->is_reference
) {
613 return &darg
->let
->ptr
;
615 return darg
->let
->ptr
;
619 static inline void psi_do_set(zval
*return_value
, set_value
*set
)
621 ZVAL_DEREF(return_value
);
622 zval_dtor(return_value
);
624 set
->func
->handler(return_value
, set
, set
->vars
->vars
[0]->arg
->let
->ptr
);
627 static inline void psi_do_return(zval
*return_value
, return_stmt
*ret
, impl_val
*ret_val
)
629 ret
->set
->func
->handler(return_value
, ret
->set
, ret_val
);
632 static inline void psi_do_free(free_stmt
*fre
)
637 for (i
= 0; i
< fre
->calls
->count
; ++i
) {
638 free_call
*f
= fre
->calls
->list
[i
];
640 for (j
= 0; j
< f
->vars
->count
; ++j
) {
641 decl_var
*dvar
= f
->vars
->vars
[j
];
642 decl_arg
*darg
= dvar
->arg
;
644 f
->decl
->call
.args
[j
] = &darg
->let
->out
;
647 /* FIXME: check in validate_* that free functions return scalar */
648 PSI_ContextCall(&PSI_G(context
), &dummy
, f
->decl
);
652 static inline void psi_do_clean(impl
*impl
)
656 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
657 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
659 switch (iarg
->type
->type
) {
661 if (iarg
->val
.zend
.str
) {
662 zend_string_release(iarg
->val
.zend
.str
);
668 if (impl
->decl
->args
) for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
669 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
671 if (darg
->let
&& darg
->let
->mem
) {
672 decl_type
*type
= real_decl_type(darg
->type
);
674 if (type
->type
== PSI_T_STRUCT
) {
675 void **ptr
= (void **) ((char *) darg
->let
->mem
+ type
->strct
->size
);
681 efree(darg
->let
->mem
);
682 darg
->let
->mem
= NULL
;
687 void psi_call(zend_execute_data
*execute_data
, zval
*return_value
, impl
*impl
)
692 if (SUCCESS
!= psi_parse_args(execute_data
, impl
)) {
696 if (impl
->decl
->args
) {
697 for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
698 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
700 impl
->decl
->call
.args
[i
] = psi_do_let(darg
);
704 memset(&ret_val
, 0, sizeof(ret_val
));
705 PSI_ContextCall(&PSI_G(context
), &ret_val
, impl
->decl
);
707 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
709 for (i
= 0; i
< impl
->stmts
->set
.count
; ++i
) {
710 set_stmt
*set
= impl
->stmts
->set
.list
[i
];
713 psi_do_set(set
->arg
->_zv
, set
->val
);
717 for (i
= 0; i
< impl
->stmts
->fre
.count
; ++i
) {
718 free_stmt
*fre
= impl
->stmts
->fre
.list
[i
];
726 PHP_MINIT_FUNCTION(psi
)
728 PSI_ContextOps
*ops
= NULL
;
730 REGISTER_INI_ENTRIES();
733 if (!strcasecmp(PSI_G(engine
), "jit")) {
742 php_error(E_WARNING
, "No PSI engine found");
746 PSI_ContextInit(&PSI_G(context
), ops
, psi_error
);
747 PSI_ContextBuild(&PSI_G(context
), PSI_G(directory
));
749 if (getenv("PSI_DUMP")) {
750 PSI_ContextDump(&PSI_G(context
), STDOUT_FILENO
);
755 PHP_MSHUTDOWN_FUNCTION(psi
)
757 PSI_ContextDtor(&PSI_G(context
));
759 UNREGISTER_INI_ENTRIES();
764 /* Remove if there's nothing to do at request start */
765 /* {{{ PHP_RINIT_FUNCTION
767 PHP_RINIT_FUNCTION(psi
)
769 #if defined(COMPILE_DL_PSI) && defined(ZTS)
770 ZEND_TSRMLS_CACHE_UPDATE();
776 /* Remove if there's nothing to do at request end */
777 /* {{{ PHP_RSHUTDOWN_FUNCTION
779 PHP_RSHUTDOWN_FUNCTION(psi
)
785 PHP_MINFO_FUNCTION(psi
)
787 php_info_print_table_start();
788 php_info_print_table_header(2, "psi support", "enabled");
789 php_info_print_table_end();
791 DISPLAY_INI_ENTRIES();
793 const zend_function_entry psi_functions
[] = {
797 zend_module_entry psi_module_entry
= {
798 STANDARD_MODULE_HEADER
,
803 PHP_RINIT(psi
), /* Replace with NULL if there's nothing to do at request start */
804 PHP_RSHUTDOWN(psi
), /* Replace with NULL if there's nothing to do at request end */
807 STANDARD_MODULE_PROPERTIES
810 #ifdef COMPILE_DL_PSI
812 ZEND_TSRMLS_CACHE_DEFINE();
822 * vim600: noet sw=4 ts=4 fdm=marker
823 * vim<600: noet sw=4 ts=4