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
, token_t t
, unsigned i
, impl_val
*tmp
)
301 size_t size
= psi_t_size(t
);
303 memset(tmp
, 0, sizeof(*tmp
));
304 memcpy(tmp
, val
->ptr
+ size
* i
, size
);
308 void psi_from_zval(impl_val
*mem
, decl_arg
*spec
, zval
*zv
, void **tmp
)
310 decl_type
*type
= real_decl_type(spec
->type
);
312 switch (type
->type
) {
314 mem
->fval
= (float) zval_get_double(zv
);
317 mem
->dval
= zval_get_double(zv
);
321 if (spec
->var
->pointer_level
) {
322 zend_string
*zs
= zval_get_string(zv
);
323 *tmp
= mem
->ptr
= estrndup(zs
->val
, zs
->len
);
324 zend_string_release(zs
);
329 mem
->zend
.lval
= zval_get_long(zv
);
334 void *psi_array_to_struct(decl_struct
*s
, HashTable
*arr
)
337 char *mem
= ecalloc(1, s
->size
+ s
->args
->count
* sizeof(void *));
339 if (arr
) for (i
= 0; i
< s
->args
->count
; ++i
) {
340 decl_arg
*darg
= s
->args
->args
[i
];
341 zval
*entry
= zend_hash_str_find_ind(arr
, darg
->var
->name
, strlen(darg
->var
->name
));
347 memset(&tmp
, 0, sizeof(tmp
));
348 psi_from_zval(&val
, darg
, entry
, &tmp
);
349 memcpy(mem
+ darg
->layout
->pos
, &val
, darg
->layout
->len
);
351 ((void **)(mem
+ s
->size
))[j
++] = tmp
;
358 void psi_to_array(zval
*return_value
, set_value
*set
, impl_val
*r_val
)
362 decl_var
*var
= set
->vars
->vars
[0];
363 token_t t
= real_decl_type(var
->arg
->type
)->type
;
364 impl_val tmp
, *ret_val
= deref_impl_val(r_val
, var
);
366 array_init(return_value
);
368 if (t
== PSI_T_STRUCT
) {
369 decl_struct
*s
= real_decl_type(var
->arg
->type
)->strct
;
374 /* explicit member casts */
375 for (i
= 0; i
< set
->count
; ++i
) {
378 set_value
*sub_set
= set
->inner
[i
];
379 decl_var
*sub_var
= sub_set
->vars
->vars
[0];
380 decl_arg
*sub_arg
= sub_var
->arg
;
383 void *ptr
= malloc(sub_arg
->layout
->len
);
385 memcpy(ptr
, (char *) ret_val
->ptr
+ sub_arg
->layout
->pos
,
386 sub_arg
->layout
->len
);
387 tmp_ptr
= enref_impl_val(ptr
, sub_arg
->var
);
388 sub_set
->func
->handler(&ztmp
, sub_set
, tmp_ptr
);
389 add_assoc_zval(return_value
, sub_var
->name
, &ztmp
);
391 if (tmp_ptr
!= ptr
) {
398 // for (i = 0; i < s->args->count; ++i) {
399 // decl_arg *darg = s->args->args[i];
400 // impl_val tmp, tmp_ptr;
402 // char *ptr = (char *) ret_val->ptr + darg->layout->pos;
404 // tmp_ptr.ptr = &tmp;
405 // memset(&tmp, 0, sizeof(tmp));
406 // memcpy(&tmp, ptr, darg->layout->len);
407 // switch (real_decl_type(darg->type)->type) {
409 // case PSI_T_DOUBLE:
410 // psi_to_double(&ztmp, real_decl_type(darg->type)->type, &tmp, darg->var);
414 // if (darg->var->pointer_level) {
415 // psi_to_string(&ztmp, real_decl_type(darg->type)->type, &tmp_ptr, darg->var);
420 // case PSI_T_UINT16:
422 // case PSI_T_UINT32:
424 // case PSI_T_UINT64:
425 // psi_to_int(&ztmp, real_decl_type(darg->type)->type, &tmp, darg->var);
427 // case PSI_T_STRUCT:
428 // psi_to_array(&ztmp, real_decl_type(darg->type)->type, &tmp_ptr, darg->var);
431 // printf("t=%d\n", real_decl_type(darg->type)->type);
434 // add_assoc_zval(return_value, darg->var->name, &ztmp);
437 if (var
->arg
->var
->array_size
) {
438 /* to_array(foo[NUMBER]) */
439 for (i
= 0; i
< var
->arg
->var
->array_size
; ++i
) {
440 impl_val
*ptr
= iterate(ret_val
, t
, i
, &tmp
);
444 ZVAL_DOUBLE(&ele
, (double) ptr
->fval
);
447 ZVAL_DOUBLE(&ele
, ptr
->dval
);
450 ZVAL_LONG(&ele
, ptr
->lval
);
454 add_next_index_zval(return_value
, &ele
);
458 /* pointer to something */
461 for (i
= 0; (ptr
= iterate(ret_val
, t
, i
, &tmp
)); ++i
) {
469 static inline ZEND_RESULT_CODE
psi_parse_args(zend_execute_data
*execute_data
, impl
*impl
)
473 if (!impl
->func
->args
->count
) {
474 return zend_parse_parameters_none();
477 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl
), impl
->func
->args
->count
)
479 iarg
= impl
->func
->args
->args
[_i
];
483 if (PSI_T_BOOL
== iarg
->type
->type
) {
485 iarg
->val
.zend
.bval
= iarg
->def
->type
== PSI_T_TRUE
? 1 : 0;
487 Z_PARAM_BOOL(iarg
->val
.zend
.bval
);
488 } else if (PSI_T_INT
== iarg
->type
->type
) {
490 iarg
->val
.zend
.lval
= zend_atol(iarg
->def
->text
, strlen(iarg
->def
->text
));
492 Z_PARAM_LONG(iarg
->val
.zend
.lval
);
493 } else if (PSI_T_FLOAT
== iarg
->type
->type
|| PSI_T_DOUBLE
== iarg
->type
->type
) {
495 iarg
->val
.dval
= zend_strtod(iarg
->def
->text
, NULL
);
497 Z_PARAM_DOUBLE(iarg
->val
.dval
);
498 } else if (PSI_T_STRING
== iarg
->type
->type
) {
499 struct {char *val
; size_t len
;} str
;
502 str
.len
= strlen(iarg
->def
->text
) - 2;
503 str
.val
= &iarg
->def
->text
[1];
505 Z_PARAM_STR_EX(iarg
->val
.zend
.str
, 1, 0);
506 if (iarg
->val
.zend
.str
) {
507 zend_string_addref(iarg
->val
.zend
.str
);
508 } else if (iarg
->def
) {
509 iarg
->val
.zend
.str
= zend_string_init(str
.val
, str
.len
, 0);
511 } else if (PSI_T_ARRAY
== iarg
->type
->type
) {
512 /* handled as _zv in let or set */
515 error_code
= ZPP_ERROR_FAILURE
;
519 if (_i
< _max_num_args
) {
522 ZEND_PARSE_PARAMETERS_END_EX(return FAILURE
);
527 static inline void *psi_do_calloc(let_calloc
*alloc
)
529 decl_type
*type
= real_decl_type(alloc
->type
);
532 if (type
->type
== PSI_T_STRUCT
) {
533 /* psi_do_clean expects at least one NULL pointer after the struct */
534 size
= type
->strct
->size
+ sizeof(void *);
536 size
= psi_t_size(type
->type
);
539 return ecalloc(alloc
->n
, size
);
542 static inline void *psi_do_let(decl_arg
*darg
)
544 impl_arg
*iarg
= darg
->let
->arg
;
547 darg
->let
->ptr
= &darg
->let
->out
;
548 arg_val
= darg
->let
->ptr
;
551 /* let foo = calloc(1, long);
555 if (darg
->let
->val
->func
&& darg
->let
->val
->func
->type
== PSI_T_CALLOC
) {
556 arg_val
->ptr
= psi_do_calloc(darg
->let
->val
->func
->alloc
);
557 darg
->let
->mem
= arg_val
->ptr
;
558 } else if (darg
->var
->array_size
) {
559 arg_val
->ptr
= ecalloc(darg
->var
->array_size
, sizeof(*arg_val
));
560 darg
->let
->mem
= arg_val
->ptr
;
562 memset(arg_val
, 0, sizeof(*arg_val
));
566 switch (darg
->let
->val
->func
->type
) {
568 if (iarg
->type
->type
== PSI_T_BOOL
) {
569 arg_val
->cval
= iarg
->val
.zend
.bval
;
571 arg_val
->cval
= zend_is_true(iarg
->_zv
);
575 if (iarg
->type
->type
== PSI_T_INT
) {
576 arg_val
->lval
= iarg
->val
.zend
.lval
;
578 arg_val
->lval
= zval_get_long(iarg
->_zv
);
582 if (iarg
->type
->type
== PSI_T_STRING
) {
583 arg_val
->ptr
= estrdup(iarg
->val
.zend
.str
->val
);
584 darg
->let
->mem
= arg_val
->ptr
;
585 zend_string_release(iarg
->val
.zend
.str
);
587 zend_string
*zs
= zval_get_string(iarg
->_zv
);
588 arg_val
->ptr
= estrdup(zs
->val
);
589 darg
->let
->mem
= arg_val
->ptr
;
590 zend_string_release(zs
);
594 if (iarg
->type
->type
== PSI_T_STRING
) {
595 arg_val
->lval
= iarg
->val
.zend
.str
->len
;
596 zend_string_release(iarg
->val
.zend
.str
);
598 zend_string
*zs
= zval_get_string(iarg
->_zv
);
599 arg_val
->lval
= zs
->len
;
600 zend_string_release(zs
);
604 if (iarg
->type
->type
== PSI_T_ARRAY
) {
605 decl_type
*type
= real_decl_type(darg
->type
);
607 switch (type
->type
) {
609 arg_val
->ptr
= psi_array_to_struct(type
->strct
, HASH_OF(iarg
->_zv
));
610 darg
->let
->mem
= arg_val
->ptr
;
615 EMPTY_SWITCH_DEFAULT_CASE();
619 if (darg
->let
->val
&& darg
->let
->val
->is_reference
) {
620 return &darg
->let
->ptr
;
622 return darg
->let
->ptr
;
626 static inline void psi_do_set(zval
*return_value
, set_value
*set
)
628 ZVAL_DEREF(return_value
);
629 zval_dtor(return_value
);
631 set
->func
->handler(return_value
, set
, set
->vars
->vars
[0]->arg
->let
->ptr
);
634 static inline void psi_do_return(zval
*return_value
, return_stmt
*ret
, impl_val
*ret_val
)
636 ret
->set
->func
->handler(return_value
, ret
->set
, ret_val
);
639 static inline void psi_do_free(free_stmt
*fre
)
644 for (i
= 0; i
< fre
->calls
->count
; ++i
) {
645 free_call
*f
= fre
->calls
->list
[i
];
647 for (j
= 0; j
< f
->vars
->count
; ++j
) {
648 decl_var
*dvar
= f
->vars
->vars
[j
];
649 decl_arg
*darg
= dvar
->arg
;
651 f
->decl
->call
.args
[j
] = &darg
->let
->out
;
654 /* FIXME: check in validate_* that free functions return scalar */
655 PSI_ContextCall(&PSI_G(context
), &dummy
, f
->decl
);
659 static inline void psi_do_clean(impl
*impl
)
663 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
664 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
666 switch (iarg
->type
->type
) {
668 if (iarg
->val
.zend
.str
) {
669 zend_string_release(iarg
->val
.zend
.str
);
675 if (impl
->decl
->args
) for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
676 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
678 if (darg
->let
&& darg
->let
->mem
) {
679 decl_type
*type
= real_decl_type(darg
->type
);
681 if (type
->type
== PSI_T_STRUCT
) {
682 void **ptr
= (void **) ((char *) darg
->let
->mem
+ type
->strct
->size
);
688 efree(darg
->let
->mem
);
689 darg
->let
->mem
= NULL
;
694 void psi_call(zend_execute_data
*execute_data
, zval
*return_value
, impl
*impl
)
699 if (SUCCESS
!= psi_parse_args(execute_data
, impl
)) {
703 if (impl
->decl
->args
) {
704 for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
705 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
707 impl
->decl
->call
.args
[i
] = psi_do_let(darg
);
711 memset(&ret_val
, 0, sizeof(ret_val
));
712 PSI_ContextCall(&PSI_G(context
), &ret_val
, impl
->decl
);
714 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
716 for (i
= 0; i
< impl
->stmts
->set
.count
; ++i
) {
717 set_stmt
*set
= impl
->stmts
->set
.list
[i
];
720 psi_do_set(set
->arg
->_zv
, set
->val
);
724 for (i
= 0; i
< impl
->stmts
->fre
.count
; ++i
) {
725 free_stmt
*fre
= impl
->stmts
->fre
.list
[i
];
733 PHP_MINIT_FUNCTION(psi
)
735 PSI_ContextOps
*ops
= NULL
;
737 REGISTER_INI_ENTRIES();
740 if (!strcasecmp(PSI_G(engine
), "jit")) {
749 php_error(E_WARNING
, "No PSI engine found");
753 PSI_ContextInit(&PSI_G(context
), ops
, psi_error
);
754 PSI_ContextBuild(&PSI_G(context
), PSI_G(directory
));
756 if (getenv("PSI_DUMP")) {
757 PSI_ContextDump(&PSI_G(context
), STDOUT_FILENO
);
762 PHP_MSHUTDOWN_FUNCTION(psi
)
764 PSI_ContextDtor(&PSI_G(context
));
766 UNREGISTER_INI_ENTRIES();
771 /* Remove if there's nothing to do at request start */
772 /* {{{ PHP_RINIT_FUNCTION
774 PHP_RINIT_FUNCTION(psi
)
776 #if defined(COMPILE_DL_PSI) && defined(ZTS)
777 ZEND_TSRMLS_CACHE_UPDATE();
783 /* Remove if there's nothing to do at request end */
784 /* {{{ PHP_RSHUTDOWN_FUNCTION
786 PHP_RSHUTDOWN_FUNCTION(psi
)
792 PHP_MINFO_FUNCTION(psi
)
794 php_info_print_table_start();
795 php_info_print_table_header(2, "psi support", "enabled");
796 php_info_print_table_end();
798 DISPLAY_INI_ENTRIES();
800 const zend_function_entry psi_functions
[] = {
804 zend_module_entry psi_module_entry
= {
805 STANDARD_MODULE_HEADER
,
810 PHP_RINIT(psi
), /* Replace with NULL if there's nothing to do at request start */
811 PHP_RSHUTDOWN(psi
), /* Replace with NULL if there's nothing to do at request end */
814 STANDARD_MODULE_PROPERTIES
817 #ifdef COMPILE_DL_PSI
819 ZEND_TSRMLS_CACHE_DEFINE();
829 * vim600: noet sw=4 ts=4 fdm=marker
830 * vim<600: noet sw=4 ts=4