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
, token_t t
, impl_val
*ret_val
, set_value
*set
, decl_var
*var
)
209 impl_val
*v
= deref_impl_val(ret_val
, var
);
213 RETVAL_DOUBLE((double) v
->fval
);
216 RETVAL_DOUBLE(v
->dval
);
219 RETVAL_LONG(v
->lval
);
222 convert_to_boolean(return_value
);
225 void psi_to_int(zval
*return_value
, token_t t
, impl_val
*ret_val
, set_value
*set
, decl_var
*var
)
227 impl_val
*v
= deref_impl_val(ret_val
, var
);
231 RETVAL_DOUBLE((double) v
->fval
);
234 RETVAL_DOUBLE(v
->dval
);
237 RETVAL_LONG(v
->lval
);
240 convert_to_long(return_value
);
243 void psi_to_double(zval
*return_value
, token_t t
, impl_val
*ret_val
, set_value
*set
, decl_var
*var
)
245 impl_val
*v
= deref_impl_val(ret_val
, var
);
249 RETVAL_DOUBLE((double) v
->fval
);
252 RETVAL_DOUBLE(v
->dval
);
255 RETVAL_DOUBLE((double) v
->lval
);
260 void psi_to_string(zval
*return_value
, token_t t
, impl_val
*ret_val
, set_value
*set
, decl_var
*var
)
265 if (!var
->arg
->var
->pointer_level
) {
266 RETVAL_STRINGL(&ret_val
->cval
, 1);
268 ret_val
= deref_impl_val(ret_val
, var
);
269 if (ret_val
&& ret_val
->ptr
) {
270 RETVAL_STRING(ret_val
->ptr
);
272 RETVAL_EMPTY_STRING();
277 RETVAL_DOUBLE((double) deref_impl_val(ret_val
, var
)->fval
);
280 RETVAL_DOUBLE(deref_impl_val(ret_val
, var
)->dval
);
283 RETVAL_LONG(deref_impl_val(ret_val
, var
)->lval
);
286 convert_to_string(return_value
);
290 static impl_val
*iterate(impl_val
*val
, token_t t
, unsigned i
, impl_val
*tmp
)
292 size_t size
= psi_t_size(t
);
294 memset(tmp
, 0, sizeof(*tmp
));
295 memcpy(tmp
, val
->ptr
+ size
* i
, size
);
299 void psi_from_zval(impl_val
*mem
, decl_arg
*spec
, zval
*zv
, void **tmp
)
301 decl_type
*type
= real_decl_type(spec
->type
);
303 switch (type
->type
) {
305 mem
->fval
= (float) zval_get_double(zv
);
308 mem
->dval
= zval_get_double(zv
);
312 if (spec
->var
->pointer_level
) {
313 zend_string
*zs
= zval_get_string(zv
);
314 *tmp
= mem
->ptr
= estrndup(zs
->val
, zs
->len
);
315 zend_string_release(zs
);
320 mem
->zend
.lval
= zval_get_long(zv
);
325 void *psi_array_to_struct(decl_struct
*s
, HashTable
*arr
)
328 char *mem
= ecalloc(1, s
->size
+ s
->args
->count
* sizeof(void *));
330 if (arr
) for (i
= 0; i
< s
->args
->count
; ++i
) {
331 decl_arg
*darg
= s
->args
->args
[i
];
332 zval
*entry
= zend_hash_str_find_ind(arr
, darg
->var
->name
, strlen(darg
->var
->name
));
338 memset(&tmp
, 0, sizeof(tmp
));
339 psi_from_zval(&val
, darg
, entry
, &tmp
);
340 memcpy(mem
+ darg
->layout
->pos
, &val
, darg
->layout
->len
);
342 ((void **)(mem
+ s
->size
))[j
++] = tmp
;
349 void psi_to_array(zval
*return_value
, token_t t
, impl_val
*ret_val
, set_value
*set
, decl_var
*var
)
355 array_init(return_value
);
357 if (t
== PSI_T_STRUCT
) {
358 decl_struct
*s
= real_decl_type(var
->arg
->type
)->strct
;
359 ret_val
= deref_impl_val(ret_val
, var
);
364 /* explicit member casts */
365 for (i
= 0; i
< set
->count
; ++i
) {
368 set_value
*sub_set
= set
->inner
[i
];
369 decl_var
*sub_var
= sub_set
->vars
->vars
[0];
370 decl_arg
*sub_arg
= sub_var
->arg
;
373 token_t t
= real_decl_type(sub_arg
->type
)->type
;
374 void *ptr
= malloc(sub_arg
->layout
->len
);
376 memcpy(ptr
, (char *) ret_val
->ptr
+ sub_arg
->layout
->pos
,
377 sub_arg
->layout
->len
);
378 tmp_ptr
= enref_impl_val(ptr
, sub_arg
->var
);
379 sub_set
->func
->handler(&ztmp
, t
, tmp_ptr
, sub_set
, sub_var
);
380 add_assoc_zval(return_value
, sub_var
->name
, &ztmp
);
382 if (tmp_ptr
!= ptr
) {
390 // for (i = 0; i < s->args->count; ++i) {
391 // decl_arg *darg = s->args->args[i];
392 // impl_val tmp, tmp_ptr;
394 // char *ptr = (char *) ret_val->ptr + darg->layout->pos;
396 // tmp_ptr.ptr = &tmp;
397 // memset(&tmp, 0, sizeof(tmp));
398 // memcpy(&tmp, ptr, darg->layout->len);
399 // switch (real_decl_type(darg->type)->type) {
401 // case PSI_T_DOUBLE:
402 // psi_to_double(&ztmp, real_decl_type(darg->type)->type, &tmp, darg->var);
406 // if (darg->var->pointer_level) {
407 // psi_to_string(&ztmp, real_decl_type(darg->type)->type, &tmp_ptr, darg->var);
412 // case PSI_T_UINT16:
414 // case PSI_T_UINT32:
416 // case PSI_T_UINT64:
417 // psi_to_int(&ztmp, real_decl_type(darg->type)->type, &tmp, darg->var);
419 // case PSI_T_STRUCT:
420 // psi_to_array(&ztmp, real_decl_type(darg->type)->type, &tmp_ptr, darg->var);
423 // printf("t=%d\n", real_decl_type(darg->type)->type);
426 // add_assoc_zval(return_value, darg->var->name, &ztmp);
430 ret_val
= deref_impl_val(ret_val
, var
);
431 for (i
= 0; i
< var
->arg
->var
->array_size
; ++i
) {
432 impl_val
*ptr
= iterate(ret_val
, t
, i
, &tmp
);
436 ZVAL_DOUBLE(&ele
, (double) ptr
->fval
);
439 ZVAL_DOUBLE(&ele
, ptr
->dval
);
442 ZVAL_LONG(&ele
, ptr
->lval
);
446 add_next_index_zval(return_value
, &ele
);
450 static inline ZEND_RESULT_CODE
psi_parse_args(zend_execute_data
*execute_data
, impl
*impl
)
454 if (!impl
->func
->args
->count
) {
455 return zend_parse_parameters_none();
458 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl
), impl
->func
->args
->count
)
460 iarg
= impl
->func
->args
->args
[_i
];
464 if (PSI_T_BOOL
== iarg
->type
->type
) {
466 iarg
->val
.zend
.bval
= iarg
->def
->type
== PSI_T_TRUE
? 1 : 0;
468 Z_PARAM_BOOL(iarg
->val
.zend
.bval
);
469 } else if (PSI_T_INT
== iarg
->type
->type
) {
471 iarg
->val
.zend
.lval
= zend_atol(iarg
->def
->text
, strlen(iarg
->def
->text
));
473 Z_PARAM_LONG(iarg
->val
.zend
.lval
);
474 } else if (PSI_T_FLOAT
== iarg
->type
->type
|| PSI_T_DOUBLE
== iarg
->type
->type
) {
476 iarg
->val
.dval
= zend_strtod(iarg
->def
->text
, NULL
);
478 Z_PARAM_DOUBLE(iarg
->val
.dval
);
479 } else if (PSI_T_STRING
== iarg
->type
->type
) {
480 struct {char *val
; size_t len
;} str
;
483 str
.len
= strlen(iarg
->def
->text
) - 2;
484 str
.val
= &iarg
->def
->text
[1];
486 Z_PARAM_STR_EX(iarg
->val
.zend
.str
, 1, 0);
487 if (iarg
->val
.zend
.str
) {
488 zend_string_addref(iarg
->val
.zend
.str
);
489 } else if (iarg
->def
) {
490 iarg
->val
.zend
.str
= zend_string_init(str
.val
, str
.len
, 0);
492 } else if (PSI_T_ARRAY
== iarg
->type
->type
) {
493 /* handled as _zv in let or set */
496 error_code
= ZPP_ERROR_FAILURE
;
500 if (_i
< _max_num_args
) {
503 ZEND_PARSE_PARAMETERS_END_EX(return FAILURE
);
508 static inline void *psi_do_calloc(let_calloc
*alloc
)
510 decl_type
*type
= real_decl_type(alloc
->type
);
513 if (type
->type
== PSI_T_STRUCT
) {
514 /* psi_do_clean expects at least one NULL pointer after the struct */
515 size
= type
->strct
->size
+ sizeof(void *);
517 size
= psi_t_size(type
->type
);
520 return ecalloc(alloc
->n
, size
);
523 static inline void *psi_do_let(decl_arg
*darg
)
525 impl_arg
*iarg
= darg
->let
->arg
;
528 darg
->let
->ptr
= &darg
->let
->out
;
529 arg_val
= darg
->let
->ptr
;
532 /* let foo = calloc(1, long);
536 if (darg
->let
->val
->func
&& darg
->let
->val
->func
->type
== PSI_T_CALLOC
) {
537 arg_val
->ptr
= psi_do_calloc(darg
->let
->val
->func
->alloc
);
538 darg
->let
->mem
= arg_val
->ptr
;
539 } else if (darg
->var
->array_size
) {
540 arg_val
->ptr
= ecalloc(darg
->var
->array_size
, sizeof(*arg_val
));
541 darg
->let
->mem
= arg_val
->ptr
;
543 memset(arg_val
, 0, sizeof(*arg_val
));
547 switch (darg
->let
->val
->func
->type
) {
549 if (iarg
->type
->type
== PSI_T_BOOL
) {
550 arg_val
->cval
= iarg
->val
.zend
.bval
;
552 arg_val
->cval
= zend_is_true(iarg
->_zv
);
556 if (iarg
->type
->type
== PSI_T_INT
) {
557 arg_val
->lval
= iarg
->val
.zend
.lval
;
559 arg_val
->lval
= zval_get_long(iarg
->_zv
);
563 if (iarg
->type
->type
== PSI_T_STRING
) {
564 arg_val
->ptr
= estrdup(iarg
->val
.zend
.str
->val
);
565 darg
->let
->mem
= arg_val
->ptr
;
566 zend_string_release(iarg
->val
.zend
.str
);
568 zend_string
*zs
= zval_get_string(iarg
->_zv
);
569 arg_val
->ptr
= estrdup(zs
->val
);
570 darg
->let
->mem
= arg_val
->ptr
;
571 zend_string_release(zs
);
575 if (iarg
->type
->type
== PSI_T_STRING
) {
576 arg_val
->lval
= iarg
->val
.zend
.str
->len
;
577 zend_string_release(iarg
->val
.zend
.str
);
579 zend_string
*zs
= zval_get_string(iarg
->_zv
);
580 arg_val
->lval
= zs
->len
;
581 zend_string_release(zs
);
585 if (iarg
->type
->type
== PSI_T_ARRAY
) {
586 decl_type
*type
= real_decl_type(darg
->type
);
588 switch (type
->type
) {
590 arg_val
->ptr
= psi_array_to_struct(type
->strct
, HASH_OF(iarg
->_zv
));
591 darg
->let
->mem
= arg_val
->ptr
;
596 EMPTY_SWITCH_DEFAULT_CASE();
600 if (darg
->let
->val
&& darg
->let
->val
->is_reference
) {
601 return &darg
->let
->ptr
;
603 return darg
->let
->ptr
;
607 static inline void psi_do_set(zval
*return_value
, set_value
*set
)
609 impl_val
*val
= (impl_val
*) &set
->vars
->vars
[0]->arg
->let
->ptr
;
610 token_t t
= real_decl_type(set
->vars
->vars
[0]->arg
->type
)->type
;
612 ZVAL_DEREF(return_value
);
613 zval_dtor(return_value
);
615 set
->func
->handler(return_value
, t
, val
, set
, set
->vars
->vars
[0]);
618 static inline void psi_do_return(zval
*return_value
, return_stmt
*ret
, impl_val
*ret_val
)
620 token_t t
= real_decl_type(ret
->decl
->type
)->type
;
622 ret
->set
->func
->handler(return_value
, t
, ret_val
, ret
->set
, ret
->decl
->var
);
625 static inline void psi_do_free(free_stmt
*fre
)
630 for (i
= 0; i
< fre
->calls
->count
; ++i
) {
631 free_call
*f
= fre
->calls
->list
[i
];
633 for (j
= 0; j
< f
->vars
->count
; ++j
) {
634 decl_var
*dvar
= f
->vars
->vars
[j
];
635 decl_arg
*darg
= dvar
->arg
;
637 f
->decl
->call
.args
[j
] = &darg
->let
->out
;
640 /* FIXME: check in validate_* that free functions return scalar */
641 PSI_ContextCall(&PSI_G(context
), &dummy
, f
->decl
);
645 static inline void psi_do_clean(impl
*impl
)
649 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
650 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
652 switch (iarg
->type
->type
) {
654 if (iarg
->val
.zend
.str
) {
655 zend_string_release(iarg
->val
.zend
.str
);
661 if (impl
->decl
->args
) for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
662 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
664 if (darg
->let
&& darg
->let
->mem
) {
665 decl_type
*type
= real_decl_type(darg
->type
);
667 if (type
->type
== PSI_T_STRUCT
) {
668 void **ptr
= (void **) ((char *) darg
->let
->mem
+ type
->strct
->size
);
674 efree(darg
->let
->mem
);
675 darg
->let
->mem
= NULL
;
680 void psi_call(zend_execute_data
*execute_data
, zval
*return_value
, impl
*impl
)
685 if (SUCCESS
!= psi_parse_args(execute_data
, impl
)) {
689 if (impl
->decl
->args
) {
690 for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
691 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
693 impl
->decl
->call
.args
[i
] = psi_do_let(darg
);
697 memset(&ret_val
, 0, sizeof(ret_val
));
698 PSI_ContextCall(&PSI_G(context
), &ret_val
, impl
->decl
);
700 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
702 for (i
= 0; i
< impl
->stmts
->set
.count
; ++i
) {
703 set_stmt
*set
= impl
->stmts
->set
.list
[i
];
706 psi_do_set(set
->arg
->_zv
, set
->val
);
710 for (i
= 0; i
< impl
->stmts
->fre
.count
; ++i
) {
711 free_stmt
*fre
= impl
->stmts
->fre
.list
[i
];
719 PHP_MINIT_FUNCTION(psi
)
721 PSI_ContextOps
*ops
= NULL
;
723 REGISTER_INI_ENTRIES();
726 if (!strcasecmp(PSI_G(engine
), "jit")) {
735 php_error(E_WARNING
, "No PSI engine found");
739 PSI_ContextInit(&PSI_G(context
), ops
, psi_error
);
740 PSI_ContextBuild(&PSI_G(context
), PSI_G(directory
));
744 PHP_MSHUTDOWN_FUNCTION(psi
)
746 PSI_ContextDtor(&PSI_G(context
));
748 UNREGISTER_INI_ENTRIES();
753 /* Remove if there's nothing to do at request start */
754 /* {{{ PHP_RINIT_FUNCTION
756 PHP_RINIT_FUNCTION(psi
)
758 #if defined(COMPILE_DL_PSI) && defined(ZTS)
759 ZEND_TSRMLS_CACHE_UPDATE();
765 /* Remove if there's nothing to do at request end */
766 /* {{{ PHP_RSHUTDOWN_FUNCTION
768 PHP_RSHUTDOWN_FUNCTION(psi
)
774 PHP_MINFO_FUNCTION(psi
)
776 php_info_print_table_start();
777 php_info_print_table_header(2, "psi support", "enabled");
778 php_info_print_table_end();
780 DISPLAY_INI_ENTRIES();
782 const zend_function_entry psi_functions
[] = {
786 zend_module_entry psi_module_entry
= {
787 STANDARD_MODULE_HEADER
,
792 PHP_RINIT(psi
), /* Replace with NULL if there's nothing to do at request start */
793 PHP_RSHUTDOWN(psi
), /* Replace with NULL if there's nothing to do at request end */
796 STANDARD_MODULE_PROPERTIES
799 #ifdef COMPILE_DL_PSI
801 ZEND_TSRMLS_CACHE_DEFINE();
811 * vim600: noet sw=4 ts=4 fdm=marker
812 * vim<600: noet sw=4 ts=4