X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fmodule.c;h=53079d0a1293771ea8b74001595224a7f6ef1350;hp=0994630984786b3205d33ce08a77cece05c313e7;hb=481515f92d2baa83dea51d2effcbde711aa99e3b;hpb=5d54598329a4d04dd0080bc1252d1de85a7befef diff --git a/src/module.c b/src/module.c index 0994630..53079d0 100644 --- a/src/module.c +++ b/src/module.c @@ -6,9 +6,13 @@ #include "php.h" #include "php_ini.h" #include "ext/standard/info.h" +#include "zend_exceptions.h" +#include "zend_constants.h" +#include "zend_operators.h" #include "php_psi.h" #include "parser.h" +#include "context.h" #if HAVE_LIBJIT # include "libjit.h" @@ -28,6 +32,9 @@ PHP_INI_BEGIN() STD_PHP_INI_ENTRY("psi.directory", "psi.d", PHP_INI_SYSTEM, OnUpdateString, directory, zend_psi_globals, psi_globals) PHP_INI_END(); +static zend_object_handlers psi_object_handlers; +static zend_class_entry *psi_class_entry; + void psi_error(int type, const char *msg, ...) { char buf[0x1000]; @@ -60,6 +67,7 @@ size_t psi_t_alignment(token_t t) return ALIGNOF_VOID_P; EMPTY_SWITCH_DEFAULT_CASE(); } + return 0; } size_t psi_t_size(token_t t) @@ -82,6 +90,7 @@ size_t psi_t_size(token_t t) return SIZEOF_VOID_P; EMPTY_SWITCH_DEFAULT_CASE(); } + return 0; } size_t psi_t_align(token_t t, size_t s) @@ -151,6 +160,11 @@ size_t psi_num_min_args(impl *impl) return n; } +void psi_to_void(zval *return_value, set_value *set, impl_val *ret_val) +{ + RETVAL_NULL(); +} + void psi_to_bool(zval *return_value, set_value *set, impl_val *ret_val) { psi_to_int(return_value, set, ret_val); @@ -263,6 +277,7 @@ void psi_to_string(zval *return_value, set_value *set, impl_val *ret_val) token_t t = real_decl_type(var->arg->type)->type; switch (t) { + case PSI_T_VOID: case PSI_T_INT8: case PSI_T_UINT8: if (!var->arg->var->pointer_level) { @@ -270,7 +285,11 @@ void psi_to_string(zval *return_value, set_value *set, impl_val *ret_val) } else { ret_val = deref_impl_val(ret_val, var); if (ret_val && ret_val->ptr) { - RETVAL_STRING(ret_val->ptr); + if (set->num) { + RETVAL_STRINGL(ret_val->ptr, psi_long_num_exp(set->num, NULL)); + } else { + RETVAL_STRING(ret_val->ptr); + } } else { RETVAL_EMPTY_STRING(); } @@ -347,16 +366,6 @@ void *psi_array_to_struct(decl_struct *s, HashTable *arr) return mem; } -static inline impl_val *struct_member_ref(decl_arg *set_arg, impl_val *struct_ptr, impl_val **to_free) { - void *ptr = (char *) struct_ptr->ptr + set_arg->layout->pos; - impl_val *val = enref_impl_val(ptr, set_arg->var); - - if (val != ptr) { - *to_free = val; - } - - return val; -} void psi_to_array(zval *return_value, set_value *set, impl_val *r_val) { size_t i; @@ -419,12 +428,12 @@ void psi_to_array(zval *return_value, set_value *set, impl_val *r_val) } else if (set->vars->count > 1) { /* to_array(arr_var, cnt_var[, cnt_var...], to_int(*arr_var)) * check for length in second var - */ + * / size_t count = 0; zval ele; if (set->outer.set) { - /* struct */ + / * struct *//* for (i = 1; i < set->vars->count; ++i) { impl_val *tmp = NULL, *cnt_val; decl_var *cnt_var = set->vars->vars[i]; @@ -444,21 +453,91 @@ void psi_to_array(zval *return_value, set_value *set, impl_val *r_val) size_t size = psi_t_size(var->arg->var->pointer_level ? PSI_T_POINTER : t); impl_val *ptr = iterate(ret_val, size, i, &tmp); + set->inner[0]->func->handler(&ele, set->inner[0], ptr); + add_next_index_zval(return_value, &ele); + } + */ + } else if (set->num) { + /* to_array(arr_var, num_expr, to_int(*arr_var)) + */ + zval ele; + zend_long i, n = psi_long_num_exp(set->num, set->outer.val); + + for (i = 0; i < n; ++i) { + size_t size = psi_t_size(var->arg->var->pointer_level ? PSI_T_POINTER : t); + impl_val *ptr = iterate(ret_val, size, i, &tmp); + set->inner[0]->func->handler(&ele, set->inner[0], ptr); add_next_index_zval(return_value, &ele); } } else { ZEND_ASSERT(0); } +} +void psi_to_object(zval *return_value, set_value *set, impl_val *r_val) +{ + decl_var *var = set->vars->vars[0]; + impl_val *ret_val = deref_impl_val(r_val, var); + psi_object *obj; + + if (ret_val->ptr) { + object_init_ex(return_value, psi_class_entry); + obj = PSI_OBJ(return_value, NULL); + obj->data = ret_val->ptr; + } else { + RETVAL_NULL(); + } } static inline ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl) { + size_t i; + zval *zarg = ZEND_CALL_ARG(execute_data, 0); impl_arg *iarg; + zend_error_handling zeh; + + zend_replace_error_handling(EH_THROW, zend_exception_get_default(), &zeh); if (!impl->func->args->count) { - return zend_parse_parameters_none(); + ZEND_RESULT_CODE rv; + + rv = zend_parse_parameters_none(); + zend_restore_error_handling(&zeh); + return rv; + } + + /* set up defaults */ + for (i = 0; i < impl->func->args->count; ++i) { + iarg = impl->func->args->args[i]; + + if (i < EX_NUM_ARGS()) { + iarg->_zv = ++zarg; + ZVAL_DEREF(iarg->_zv); + if (iarg->var->reference) { + zval_dtor(iarg->_zv); + ZVAL_NULL(iarg->_zv); + } + } + + if (iarg->def) { + switch (iarg->type->type) { + case PSI_T_BOOL: + iarg->val.zend.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0; + break; + case PSI_T_INT: + iarg->val.zend.lval = zend_atol(iarg->def->text, strlen(iarg->def->text)); + break; + case PSI_T_FLOAT: + case PSI_T_DOUBLE: + iarg->val.dval = zend_strtod(iarg->def->text, NULL); + break; + case PSI_T_STRING: + /* FIXME */ + iarg->val.zend.str = zend_string_init(&iarg->def->text[1], strlen(iarg->def->text) - 2, 0); + break; + } + } } ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->count) @@ -483,56 +562,44 @@ static inline ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, i } Z_PARAM_DOUBLE(iarg->val.dval); } else if (PSI_T_STRING == iarg->type->type) { - struct {char *val; size_t len;} str; - if (iarg->def) { - /* FIXME */ - str.len = strlen(iarg->def->text) - 2; - str.val = &iarg->def->text[1]; - } - Z_PARAM_STR_EX(iarg->val.zend.str, 1, 0); - if (iarg->val.zend.str) { - zend_string_addref(iarg->val.zend.str); - } else if (iarg->def) { - iarg->val.zend.str = zend_string_init(str.val, str.len, 0); + zend_string *str = iarg->val.zend.str; + Z_PARAM_STR_EX(iarg->val.zend.str, 1, iarg->var->reference); + if (str && str != iarg->val.zend.str) { + zend_string_release(str); } } else if (PSI_T_ARRAY == iarg->type->type) { - /* handled as _zv in let or set */ + Z_PARAM_PROLOGUE(0); + } else if (PSI_T_OBJECT == iarg->type->type) { Z_PARAM_PROLOGUE(0); } else { error_code = ZPP_ERROR_FAILURE; break; } - iarg->_zv = _arg; - if (_i < _max_num_args) { + if (_i < _num_args) { goto nextarg; } - ZEND_PARSE_PARAMETERS_END_EX(return FAILURE); + ZEND_PARSE_PARAMETERS_END_EX( + zend_restore_error_handling(&zeh); + return FAILURE + ); + zend_restore_error_handling(&zeh); return SUCCESS; } static inline void *psi_do_calloc(let_calloc *alloc) { - decl_type *type = real_decl_type(alloc->type); - size_t size; - - if (type->type == PSI_T_STRUCT) { - /* psi_do_clean expects at least one NULL pointer after the struct */ - size = type->strct->size + sizeof(void *); - } else { - size = psi_t_size(type->type); - } - - return ecalloc(alloc->n, size); + zend_long n = psi_long_num_exp(alloc->nmemb, NULL), s = psi_long_num_exp(alloc->size, NULL); + void *mem = safe_emalloc(n, s, sizeof(void *)); + memset(mem, 0, n * s + sizeof(void *)); + return mem; } -static inline void *psi_do_let(decl_arg *darg) +static inline void *psi_do_let(let_stmt *let) { + decl_arg *darg = let->var->arg; impl_arg *iarg = darg->let->arg; - impl_val *arg_val; - - darg->let->ptr = &darg->let->out; - arg_val = darg->let->ptr; + impl_val *arg_val = darg->ptr; if (!iarg) { /* let foo = calloc(1, long); @@ -541,10 +608,10 @@ static inline void *psi_do_let(decl_arg *darg) */ if (darg->let->val->func && darg->let->val->func->type == PSI_T_CALLOC) { arg_val->ptr = psi_do_calloc(darg->let->val->func->alloc); - darg->let->mem = arg_val->ptr; + darg->mem = arg_val->ptr; } else if (darg->var->array_size) { arg_val->ptr = ecalloc(darg->var->array_size, sizeof(*arg_val)); - darg->let->mem = arg_val->ptr; + darg->mem = arg_val->ptr; } else { memset(arg_val, 0, sizeof(*arg_val)); } @@ -565,17 +632,24 @@ static inline void *psi_do_let(decl_arg *darg) arg_val->lval = zval_get_long(iarg->_zv); } break; + case PSI_T_PATHVAL: case PSI_T_STRVAL: if (iarg->type->type == PSI_T_STRING) { arg_val->ptr = estrdup(iarg->val.zend.str->val); - darg->let->mem = arg_val->ptr; + darg->mem = arg_val->ptr; zend_string_release(iarg->val.zend.str); } else { zend_string *zs = zval_get_string(iarg->_zv); arg_val->ptr = estrdup(zs->val); - darg->let->mem = arg_val->ptr; + darg->mem = arg_val->ptr; zend_string_release(zs); } + if (PSI_T_PATHVAL == darg->let->val->func->type) { + if (SUCCESS != php_check_open_basedir(arg_val->ptr)) { + efree(arg_val->ptr); + return NULL; + } + } break; case PSI_T_STRLEN: if (iarg->type->type == PSI_T_STRING) { @@ -594,31 +668,45 @@ static inline void *psi_do_let(decl_arg *darg) switch (type->type) { case PSI_T_STRUCT: arg_val->ptr = psi_array_to_struct(type->strct, HASH_OF(iarg->_zv)); - darg->let->mem = arg_val->ptr; + darg->mem = arg_val->ptr; break; } } break; + case PSI_T_OBJVAL: + if (iarg->type->type == PSI_T_OBJECT) { + psi_object *obj; + + if (!instanceof_function(Z_OBJCE_P(iarg->_zv), psi_class_entry)) { + return NULL; + } + + obj = PSI_OBJ(iarg->_zv, NULL); + arg_val->ptr = obj->data; + } + break; EMPTY_SWITCH_DEFAULT_CASE(); } } if (darg->let->val && darg->let->val->is_reference) { - return &darg->let->ptr; + return darg->let->ptr = &darg->ptr; } else { - return darg->let->ptr; + return darg->let->ptr = darg->ptr; } } static inline void psi_do_set(zval *return_value, set_value *set) { - ZVAL_DEREF(return_value); - zval_dtor(return_value); + set->func->handler(return_value, set, set->vars->vars[0]->arg->ptr); +} - set->func->handler(return_value, set, set->vars->vars[0]->arg->let->ptr); +static inline void psi_do_return(zval *return_value, return_stmt *ret) +{ + ret->set->func->handler(return_value, ret->set, ret->set->vars->vars[0]->arg->ptr); } -static inline void psi_do_return(zval *return_value, return_stmt *ret, impl_val *ret_val) +static inline void psi_do_return2(zval *return_value, return_stmt *ret, impl_val *ret_val) { ret->set->func->handler(return_value, ret->set, ret_val); } @@ -635,7 +723,7 @@ static inline void psi_do_free(free_stmt *fre) decl_var *dvar = f->vars->vars[j]; decl_arg *darg = dvar->arg; - f->decl->call.args[j] = &darg->let->out; + f->decl->call.args[j] = &darg->val; } /* FIXME: check in validate_* that free functions return scalar */ @@ -662,43 +750,276 @@ static inline void psi_do_clean(impl *impl) if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) { decl_arg *darg = impl->decl->args->args[i]; - if (darg->let && darg->let->mem) { + if (darg->mem) { decl_type *type = real_decl_type(darg->type); if (type->type == PSI_T_STRUCT) { - void **ptr = (void **) ((char *) darg->let->mem + type->strct->size); + void **ptr = (void **) ((char *) darg->mem + type->strct->size); while (*ptr) { efree(*ptr++); } } - efree(darg->let->mem); - darg->let->mem = NULL; + efree(darg->mem); + darg->mem = NULL; + } + } +} + +static inline int psi_calc_num_exp_value(num_exp *exp, impl_val *ref, impl_val *res) { + impl_val *tmp = NULL; + + switch (exp->t) { + case PSI_T_NUMBER: + switch (is_numeric_string(exp->u.numb, strlen(exp->u.numb), (zend_long *) res, (double *) res, 0)) { + case IS_LONG: + return PSI_T_INT64; + case IS_DOUBLE: + return PSI_T_DOUBLE; + } + break; + + case PSI_T_NSNAME: + switch (exp->u.cnst->type->type) { + case PSI_T_INT: + res->i64 = zend_get_constant_str(exp->u.cnst->name, strlen(exp->u.cnst->name))->value.lval; + return PSI_T_INT64; + case PSI_T_FLOAT: + res->dval = zend_get_constant_str(exp->u.cnst->name, strlen(exp->u.cnst->name))->value.dval; + return PSI_T_DOUBLE; + default: + return 0; + } + break; + + case PSI_T_NAME: + if (1) { + ref = exp->u.dvar->arg->ptr; + } else { + ref = struct_member_ref(exp->u.dvar->arg, ref, &tmp); + } + switch (real_decl_type(exp->u.dvar->arg->type)->type) { + case PSI_T_INT8: + case PSI_T_UINT8: + case PSI_T_INT16: + case PSI_T_UINT16: + case PSI_T_INT32: + case PSI_T_UINT32: + case PSI_T_INT64: + case PSI_T_UINT64: + memcpy(res, deref_impl_val(ref, exp->u.dvar), sizeof(*res)); + if (tmp) { + free(tmp); + } + return real_decl_type(exp->u.dvar->arg->type)->type; + + case PSI_T_FLOAT: + case PSI_T_DOUBLE: + memcpy(res, deref_impl_val(ref, exp->u.dvar), sizeof(*res)); + if (tmp) { + free(tmp); + } + return real_decl_type(exp->u.dvar->arg->type)->type; + + EMPTY_SWITCH_DEFAULT_CASE(); } + break; + + EMPTY_SWITCH_DEFAULT_CASE(); + } + return 0; +} + +int psi_calc_num_exp(num_exp *exp, impl_val *ref, impl_val *res) { + impl_val num = {0}; + int num_type = psi_calc_num_exp_value(exp, ref, &num); + + if (exp->operand) { + impl_val tmp = {0}; + int tmp_type = psi_calc_num_exp(exp->operand, ref, &tmp); + + return exp->calculator(num_type, &num, tmp_type, &tmp, res); } + + memcpy(res, &num, sizeof(*res)); + return num_type; } +#define PRIfval "f" +#define PRIdval "lf" + +#define PSI_CALC_OP(var) do { \ + const char *fmt = "calc %" PRI##var ", %" PRI##var ": %" PRI##var "\n"; \ + res->var = PSI_CALC(v1->var, v2->var); \ + if (!res->var) fprintf(stderr, fmt, v1->var, v2->var, res->var); \ +} while (0) +#define PSI_CALC_OP2(vres, var1, var2) do { \ + const char *fmt = "calc %" PRI##var1 ", %" PRI##var2 ": %" PRI##vres "\n"; \ + res->vres = PSI_CALC(v1->var1, v2->var2); \ + if (!res->vres) fprintf(stderr, fmt, v1->var1, v2->var2, res->vres); \ +} while(0) +#define PSI_CALC_FN(op) int psi_calc_##op(int t1, impl_val *v1, int t2, impl_val *v2, impl_val *res) \ +{ \ + if (t1 == t2) { \ + switch (t1) { \ + case PSI_T_FLOAT: PSI_CALC_OP(fval); break; \ + case PSI_T_DOUBLE: PSI_CALC_OP(dval); break; \ + case PSI_T_INT8: PSI_CALC_OP(i8); break; \ + case PSI_T_UINT8: PSI_CALC_OP(u8); break; \ + case PSI_T_INT16: PSI_CALC_OP(i16); break; \ + case PSI_T_UINT16: PSI_CALC_OP(u16); break; \ + case PSI_T_INT32: PSI_CALC_OP(i32); break; \ + case PSI_T_UINT32: PSI_CALC_OP(u32); break; \ + case PSI_T_INT64: PSI_CALC_OP(i64); break; \ + case PSI_T_UINT64: PSI_CALC_OP(u64); break; \ + EMPTY_SWITCH_DEFAULT_CASE(); \ + } \ + return t1; \ + } else if (t1 == PSI_T_DOUBLE) { \ + switch (t2) { \ + case PSI_T_FLOAT: PSI_CALC_OP2(dval, dval, fval); break; \ + case PSI_T_INT8: PSI_CALC_OP2(dval, dval, i8); break; \ + case PSI_T_UINT8: PSI_CALC_OP2(dval, dval, u8); break; \ + case PSI_T_INT16: PSI_CALC_OP2(dval, dval, i16); break; \ + case PSI_T_UINT16: PSI_CALC_OP2(dval, dval, u16); break; \ + case PSI_T_INT32: PSI_CALC_OP2(dval, dval, i32); break; \ + case PSI_T_UINT32: PSI_CALC_OP2(dval, dval, u32); break; \ + case PSI_T_INT64: PSI_CALC_OP2(dval, dval, i64); break; \ + case PSI_T_UINT64: PSI_CALC_OP2(dval, dval, u64); break; \ + EMPTY_SWITCH_DEFAULT_CASE(); \ + } \ + return t1; \ + } else if (t2 == PSI_T_DOUBLE) { \ + switch (t1) { \ + case PSI_T_FLOAT: PSI_CALC_OP2(dval, fval, dval); break; \ + case PSI_T_INT8: PSI_CALC_OP2(dval, i8, dval); break; \ + case PSI_T_UINT8: PSI_CALC_OP2(dval, u8, dval); break; \ + case PSI_T_INT16: PSI_CALC_OP2(dval, i16, dval); break; \ + case PSI_T_UINT16: PSI_CALC_OP2(dval, u16, dval); break; \ + case PSI_T_INT32: PSI_CALC_OP2(dval, i32, dval); break; \ + case PSI_T_UINT32: PSI_CALC_OP2(dval, u32, dval); break; \ + case PSI_T_INT64: PSI_CALC_OP2(dval, i64, dval); break; \ + case PSI_T_UINT64: PSI_CALC_OP2(dval, u64, dval); break; \ + EMPTY_SWITCH_DEFAULT_CASE(); \ + } \ + return t2; \ + } else if (t1 == PSI_T_FLOAT) { \ + switch (t2) { \ + case PSI_T_DOUBLE: PSI_CALC_OP2(dval, fval, dval); return t2; \ + case PSI_T_INT8: PSI_CALC_OP2(fval, fval, i8); break; \ + case PSI_T_UINT8: PSI_CALC_OP2(fval, fval, u8); break; \ + case PSI_T_INT16: PSI_CALC_OP2(fval, fval, i16); break; \ + case PSI_T_UINT16: PSI_CALC_OP2(fval, fval, u16); break; \ + case PSI_T_INT32: PSI_CALC_OP2(fval, fval, i32); break; \ + case PSI_T_UINT32: PSI_CALC_OP2(fval, fval, u32); break; \ + case PSI_T_INT64: PSI_CALC_OP2(fval, fval, i64); break; \ + case PSI_T_UINT64: PSI_CALC_OP2(fval, fval, u64); break; \ + EMPTY_SWITCH_DEFAULT_CASE(); \ + } \ + return t1; \ + } else if (t2 == PSI_T_FLOAT) { \ + switch (t1) { \ + case PSI_T_DOUBLE: PSI_CALC_OP2(dval, dval, fval); return t1; \ + case PSI_T_INT8: PSI_CALC_OP2(fval, i8, fval); break; \ + case PSI_T_UINT8: PSI_CALC_OP2(fval, u8, fval); break; \ + case PSI_T_INT16: PSI_CALC_OP2(fval, i16, fval); break; \ + case PSI_T_UINT16: PSI_CALC_OP2(fval, u16, fval); break; \ + case PSI_T_INT32: PSI_CALC_OP2(fval, i32, fval); break; \ + case PSI_T_UINT32: PSI_CALC_OP2(fval, u32, fval); break; \ + case PSI_T_INT64: PSI_CALC_OP2(fval, i64, fval); break; \ + case PSI_T_UINT64: PSI_CALC_OP2(fval, u64, fval); break; \ + EMPTY_SWITCH_DEFAULT_CASE(); \ + } \ + return t2; \ + } else { \ + int64_t sval1 = v1->i64, sval2 = v2->i64; \ + uint64_t uval1 = v1->u64, uval2 = v2->u64; \ + switch (t1) { \ + case PSI_T_INT8: sval1 >>= 8; \ + case PSI_T_INT16: sval1 >>= 8; \ + case PSI_T_INT32: sval1 >>= 8; \ + case PSI_T_INT64: \ + switch (t2) { \ + case PSI_T_INT8: sval2 >>= 8; \ + case PSI_T_INT16: sval2 >>= 8; \ + case PSI_T_INT32: sval2 >>= 8; \ + case PSI_T_INT64: \ + res->i64 = PSI_CALC(sval1 , sval2); \ + return PSI_T_INT64; \ + case PSI_T_UINT8: uval2 >>= 8; \ + case PSI_T_UINT16: uval2 >>= 8; \ + case PSI_T_UINT32: uval2 >>= 8; \ + case PSI_T_UINT64: \ + res->i64 = PSI_CALC(sval1, uval2); \ + return PSI_T_INT64; \ + } \ + break; \ + case PSI_T_UINT8: uval1 >>= 8; \ + case PSI_T_UINT16: uval1 >>= 8; \ + case PSI_T_UINT32: uval1 >>= 8; \ + case PSI_T_UINT64: \ + switch (t2) { \ + case PSI_T_INT8: sval2 >>= 8; \ + case PSI_T_INT16: sval2 >>= 8; \ + case PSI_T_INT32: sval2 >>= 8; \ + case PSI_T_INT64: \ + res->i64 = PSI_CALC(uval1, sval2); \ + return PSI_T_INT64; \ + case PSI_T_UINT8: uval2 >>= 8; \ + case PSI_T_UINT16: uval2 >>= 8; \ + case PSI_T_UINT32: uval2 >>= 8; \ + case PSI_T_UINT64: \ + res->u64 = PSI_CALC(uval1, uval2); \ + return PSI_T_UINT64; \ + } \ + break; \ + } \ + } \ + ZEND_ASSERT(0); \ + return 0; \ +} + +#undef PSI_CALC +#define PSI_CALC(var1, var2) (var1) + (var2) +PSI_CALC_FN(add) +#undef PSI_CALC +#define PSI_CALC(var1, var2) (var1) * (var2) +PSI_CALC_FN(mul) +#undef PSI_CALC +#define PSI_CALC(var1, var2) (var1) - (var2) +PSI_CALC_FN(sub) +#undef PSI_CALC +#define PSI_CALC(var1, var2) (var1) / (var2) +PSI_CALC_FN(div) + void psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl) { - impl_val ret_val; size_t i; + return_stmt *ret = impl->stmts->ret.list[0]; + decl_var *var = ret->set->vars->vars[0]; + + memset(var->arg->ptr, 0, sizeof(impl_val)); if (SUCCESS != psi_parse_args(execute_data, impl)) { return; } - if (impl->decl->args) { - for (i = 0; i < impl->decl->args->count; ++i) { - decl_arg *darg = impl->decl->args->args[i]; + for (i = 0; i < impl->stmts->let.count; ++i) { + let_stmt *let = impl->stmts->let.list[i]; - impl->decl->call.args[i] = psi_do_let(darg); + if (!psi_do_let(let)) { + psi_do_return(return_value, ret); + psi_do_clean(impl); + return; } } - memset(&ret_val, 0, sizeof(ret_val)); - PSI_ContextCall(&PSI_G(context), &ret_val, impl->decl); + if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) { + impl->decl->call.args[i] = impl->decl->args->args[i]->let->ptr; + } - psi_do_return(return_value, impl->stmts->ret.list[0], &ret_val); + PSI_ContextCall(&PSI_G(context), var->arg->ptr, impl->decl); + psi_do_return(return_value, ret); for (i = 0; i < impl->stmts->set.count; ++i) { set_stmt *set = impl->stmts->set.list[i]; @@ -713,16 +1034,46 @@ void psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl) psi_do_free(fre); } - psi_do_clean(impl); } +static void psi_object_free(zend_object *o) +{ + psi_object *obj = PSI_OBJ(NULL, o); + + if (obj->data) { + // free(obj->data); + obj->data = NULL; + } + zend_object_std_dtor(o); +} + +static zend_object *psi_object_init(zend_class_entry *ce) +{ + psi_object *o = ecalloc(1, sizeof(*o) + zend_object_properties_size(ce)); + + zend_object_std_init(&o->std, ce); + object_properties_init(&o->std, ce); + o->std.handlers = &psi_object_handlers; + return &o->std; +} + PHP_MINIT_FUNCTION(psi) { PSI_ContextOps *ops = NULL; + zend_class_entry ce = {0}; REGISTER_INI_ENTRIES(); + INIT_NS_CLASS_ENTRY(ce, "psi", "object", NULL); + psi_class_entry = zend_register_internal_class_ex(&ce, NULL); + psi_class_entry->create_object = psi_object_init; + + memcpy(&psi_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers)); + psi_object_handlers.offset = XtOffsetOf(psi_object, std); + psi_object_handlers.free_obj = psi_object_free; + psi_object_handlers.clone_obj = NULL; + #ifdef HAVE_LIBJIT if (!strcasecmp(PSI_G(engine), "jit")) { ops = PSI_Libjit(); @@ -740,12 +1091,13 @@ PHP_MINIT_FUNCTION(psi) PSI_ContextInit(&PSI_G(context), ops, psi_error); PSI_ContextBuild(&PSI_G(context), PSI_G(directory)); - if (getenv("PSI_DUMP")) { + if (psi_check_env("PSI_DUMP")) { PSI_ContextDump(&PSI_G(context), STDOUT_FILENO); } return SUCCESS; } + PHP_MSHUTDOWN_FUNCTION(psi) { PSI_ContextDtor(&PSI_G(context)); @@ -755,26 +1107,13 @@ PHP_MSHUTDOWN_FUNCTION(psi) return SUCCESS; } -/* Remove if there's nothing to do at request start */ -/* {{{ PHP_RINIT_FUNCTION - */ +#if defined(COMPILE_DL_PSI) && defined(ZTS) PHP_RINIT_FUNCTION(psi) { -#if defined(COMPILE_DL_PSI) && defined(ZTS) ZEND_TSRMLS_CACHE_UPDATE(); -#endif return SUCCESS; } -/* }}} */ - -/* Remove if there's nothing to do at request end */ -/* {{{ PHP_RSHUTDOWN_FUNCTION - */ -PHP_RSHUTDOWN_FUNCTION(psi) -{ - return SUCCESS; -} -/* }}} */ +#endif PHP_MINFO_FUNCTION(psi) { @@ -794,8 +1133,12 @@ zend_module_entry psi_module_entry = { psi_functions, PHP_MINIT(psi), PHP_MSHUTDOWN(psi), +#if defined(COMPILE_DL_PSI) && defined(ZTS) PHP_RINIT(psi), /* Replace with NULL if there's nothing to do at request start */ - PHP_RSHUTDOWN(psi), /* Replace with NULL if there's nothing to do at request end */ +#else + NULL, +#endif + NULL, PHP_MINFO(psi), PHP_PSI_VERSION, STANDARD_MODULE_PROPERTIES