X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fengine.c;fp=src%2Fengine.c;h=c3e95c524690b800758ab5f39e85ed2c9101fb76;hp=0000000000000000000000000000000000000000;hb=b4a3c33fc143fd57288fcfe0878e7a22eeaf61bf;hpb=7e4b0ccdd2123647b6fff8065c0abc61be3fb44d

diff --git a/src/engine.c b/src/engine.c
new file mode 100644
index 0000000..c3e95c5
--- /dev/null
+++ b/src/engine.c
@@ -0,0 +1,609 @@
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include "php.h"
+#include "php_psi.h"
+
+#include "zend_exceptions.h"
+
+#include "engine.h"
+#include "calc.h"
+#include "marshal.h"
+
+size_t psi_t_alignment(token_t t)
+{
+#define PSI_ALIGNOF(T) case PSI_T_## T: return ALIGNOF_## T ##_T;
+	switch (t) {
+	PSI_ALIGNOF(INT8);
+	PSI_ALIGNOF(UINT8);
+	PSI_ALIGNOF(INT16);
+	PSI_ALIGNOF(UINT16);
+	PSI_ALIGNOF(INT32);
+	PSI_ALIGNOF(UINT32);
+	PSI_ALIGNOF(INT64);
+	PSI_ALIGNOF(UINT64);
+	case PSI_T_FLOAT:
+		return ALIGNOF_FLOAT;
+	case PSI_T_DOUBLE:
+		return ALIGNOF_DOUBLE;
+	case PSI_T_POINTER:
+		return ALIGNOF_VOID_P;
+	EMPTY_SWITCH_DEFAULT_CASE();
+	}
+	return 0;
+}
+
+size_t psi_t_size(token_t t)
+{
+#define PSI_SIZEOF(T) case PSI_T_## T : return SIZEOF_## T ##_T;
+	switch (t) {
+	PSI_SIZEOF(INT8);
+	PSI_SIZEOF(UINT8);
+	PSI_SIZEOF(INT16);
+	PSI_SIZEOF(UINT16);
+	PSI_SIZEOF(INT32);
+	PSI_SIZEOF(UINT32);
+	PSI_SIZEOF(INT64);
+	PSI_SIZEOF(UINT64);
+	case PSI_T_FLOAT:
+		return SIZEOF_FLOAT;
+	case PSI_T_DOUBLE:
+		return SIZEOF_DOUBLE;
+	case PSI_T_POINTER:
+		return SIZEOF_VOID_P;
+	EMPTY_SWITCH_DEFAULT_CASE();
+	}
+	return 0;
+}
+
+size_t psi_t_align(token_t t, size_t s)
+{
+	size_t a = psi_t_alignment(t);
+	return ((s - 1) | (a - 1)) + 1;
+}
+
+size_t psi_offset_padding(size_t diff, size_t alignment)
+{
+	if (diff && diff <= ((diff - 1) | (alignment -1)) + 1) {
+		diff = 0;
+	}
+
+	return diff;
+}
+
+int psi_internal_type(impl_type *type)
+{
+	switch (type->type) {
+	case PSI_T_BOOL:
+		return _IS_BOOL;
+	case PSI_T_INT:
+		return IS_LONG;
+	case PSI_T_FLOAT:
+	case PSI_T_DOUBLE:
+		return IS_DOUBLE;
+	case PSI_T_STRING:
+		return IS_STRING;
+	case PSI_T_ARRAY:
+		return IS_ARRAY;
+	default:
+		return 0;
+	}
+}
+
+zend_internal_arg_info *psi_internal_arginfo(impl *impl)
+{
+	size_t i;
+	zend_internal_arg_info *aip;
+	zend_internal_function_info *fi;
+
+	aip = calloc(impl->func->args->count + 1 + !!impl->func->args->vararg.name, sizeof(*aip));
+
+	fi = (zend_internal_function_info *) &aip[0];
+	fi->allow_null = 1;
+	fi->required_num_args = psi_num_min_args(impl);
+	fi->return_reference = impl->func->return_reference;
+	fi->type_hint = psi_internal_type(impl->func->return_type);
+
+	if (impl->func->args->vararg.name) {
+		impl_arg *vararg = impl->func->args->vararg.name;
+		zend_internal_arg_info *ai = &aip[impl->func->args->count];
+
+		ai->name = vararg->var->name;
+		ai->allow_null = 1;
+		ai->type_hint = psi_internal_type(vararg->type);
+		if (vararg->var->reference) {
+			ai->pass_by_reference = 1;
+		}
+		ai->is_variadic = 1;
+	}
+
+	for (i = 0; i < impl->func->args->count; ++i) {
+		impl_arg *iarg = impl->func->args->args[i];
+		zend_internal_arg_info *ai = &aip[i+1];
+
+		ai->name = iarg->var->name;
+		ai->type_hint = psi_internal_type(iarg->type);
+		if (iarg->var->reference) {
+			ai->pass_by_reference = 1;
+		}
+		//if (iarg->var->reference || (iarg->def && iarg->def->type == PSI_T_NULL)) {
+			ai->allow_null = 1;
+		//}
+	}
+
+	return aip;
+}
+
+size_t psi_num_min_args(impl *impl)
+{
+	size_t i, n = impl->func->args->count;
+
+	for (i = 0; i < impl->func->args->count; ++i) {
+		if (impl->func->args->args[i]->def) {
+			--n;
+		}
+	}
+	return n;
+}
+
+
+static inline ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl)
+{
+	size_t i;
+	impl_arg *iarg;
+	zend_error_handling zeh;
+
+	zend_replace_error_handling(EH_THROW, zend_exception_get_default(), &zeh);
+
+	if (!impl->func->args->count) {
+		ZEND_RESULT_CODE rv;
+
+		rv = zend_parse_parameters_none();
+		zend_restore_error_handling(&zeh);
+		return rv;
+	}
+
+	ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->vararg.name ? -1 : impl->func->args->count)
+	nextarg:
+		if (impl->func->args->vararg.name && _i >= impl->func->args->count) {
+			impl_arg *varg = impl->func->args->vararg.name;
+			iarg = init_impl_arg(
+					init_impl_type(varg->type->type, varg->type->name),
+					init_impl_var(varg->var->name, varg->var->reference),
+					NULL);
+
+			Z_PARAM_OPTIONAL;
+			if (_i == impl->func->args->count) {
+				impl->func->args->vararg.args = init_impl_args(iarg);
+			} else {
+				add_impl_arg(impl->func->args->vararg.args, iarg);
+			}
+		} else {
+			iarg = impl->func->args->args[_i];
+			if (iarg->def) {
+				Z_PARAM_OPTIONAL;
+			}
+		}
+		if (PSI_T_BOOL == iarg->type->type) {
+			Z_PARAM_BOOL(iarg->val.zend.bval);
+		} else if (PSI_T_INT == iarg->type->type) {
+			Z_PARAM_LONG(iarg->val.zend.lval);
+		} else if (PSI_T_FLOAT == iarg->type->type || PSI_T_DOUBLE == iarg->type->type) {
+			Z_PARAM_DOUBLE(iarg->val.dval);
+		} else if (PSI_T_STRING == iarg->type->type) {
+			Z_PARAM_STR_EX(iarg->val.zend.str, 1, iarg->var->reference);
+			if (iarg->val.zend.str) {
+				zend_string_addref(iarg->val.zend.str);
+			}
+		} else if (PSI_T_ARRAY == iarg->type->type) {
+			Z_PARAM_PROLOGUE(0);
+		} else if (PSI_T_OBJECT == iarg->type->type) {
+			Z_PARAM_PROLOGUE(0);
+		} else if (PSI_T_MIXED == iarg->type->type) {
+			Z_PARAM_PROLOGUE(0);
+		} else {
+			error_code = ZPP_ERROR_FAILURE;
+			break;
+		}
+		iarg->_zv = _arg;
+		ZVAL_DEREF(iarg->_zv);
+		if (_i < _num_args) {
+			goto nextarg;
+		}
+	ZEND_PARSE_PARAMETERS_END_EX(
+		zend_restore_error_handling(&zeh);
+		return FAILURE
+	);
+
+	/* set up defaults */
+	for (i = 0; i < impl->func->args->count; ++i) {
+		if (i >= EX_NUM_ARGS() && iarg->def) {
+			iarg = impl->func->args->args[i];
+
+			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_restore_error_handling(&zeh);
+	return SUCCESS;
+}
+
+static inline void *psi_do_calloc(let_calloc *alloc)
+{
+	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 *));
+#if 0
+	fprintf(stderr, "calloc: %p\n", mem);
+#endif
+	return mem;
+}
+
+static inline impl_val *psi_let_val(token_t let_func, impl_arg *iarg, impl_val *arg_val, decl_struct *strct, void **to_free)
+{
+	switch (let_func) {
+	case PSI_T_BOOLVAL:
+		if (iarg->type->type == PSI_T_BOOL) {
+			arg_val->cval = iarg->val.zend.bval;
+		} else {
+			arg_val->cval = zend_is_true(iarg->_zv);
+		}
+		break;
+	case PSI_T_INTVAL:
+		if (iarg->type->type == PSI_T_INT) {
+			arg_val->lval = iarg->val.zend.lval;
+		} else {
+			arg_val->lval = zval_get_long(iarg->_zv);
+		}
+		break;
+	case PSI_T_FLOATVAL:
+		if (iarg->type->type == PSI_T_FLOAT || iarg->type->type == PSI_T_DOUBLE) {
+			arg_val->dval = iarg->val.dval;
+		} else {
+			arg_val->dval = zval_get_double(iarg->_zv);
+		}
+		break;
+	case PSI_T_PATHVAL:
+	case PSI_T_STRVAL:
+		if (iarg->type->type == PSI_T_STRING) {
+			if (iarg->val.zend.str) {
+				arg_val->ptr = estrndup(iarg->val.zend.str->val, iarg->val.zend.str->len);
+				*to_free = arg_val->ptr;
+			} else {
+				arg_val->ptr = "";
+			}
+		} else {
+			zend_string *zs = zval_get_string(iarg->_zv);
+			arg_val->ptr = estrdup(zs->val);
+			*to_free = arg_val->ptr;
+			zend_string_release(zs);
+		}
+		if (PSI_T_PATHVAL == let_func) {
+			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) {
+			if (iarg->val.zend.str) {
+				arg_val->lval = iarg->val.zend.str->len;
+			} else {
+				arg_val->lval = 0;
+			}
+		} else {
+			zend_string *zs = zval_get_string(iarg->_zv);
+			arg_val->lval = zs->len;
+			zend_string_release(zs);
+		}
+		break;
+	case PSI_T_ARRVAL:
+		if (iarg->type->type == PSI_T_ARRAY) {
+			arg_val = psi_array_to_struct(strct, HASH_OF(iarg->_zv));
+			*to_free = arg_val;
+		}
+		break;
+	case PSI_T_OBJVAL:
+		if (iarg->type->type == PSI_T_OBJECT) {
+			psi_object *obj;
+
+			if (!instanceof_function(Z_OBJCE_P(iarg->_zv), psi_object_get_class_entry())) {
+				return NULL;
+			}
+
+			obj = PSI_OBJ(iarg->_zv, NULL);
+			arg_val->ptr = obj->data;
+		}
+		break;
+	EMPTY_SWITCH_DEFAULT_CASE();
+	}
+	return arg_val;
+}
+
+static inline void *psi_do_let(let_stmt *let)
+{
+	decl_arg *darg = let->var->arg;
+	impl_val *arg_val = darg->ptr;
+	impl_arg *iarg;
+
+	switch (let->val ? let->val->kind : PSI_LET_NULL) {
+	case PSI_LET_TMP:
+		memcpy(arg_val, deref_impl_val(let->val->data.var->arg->let->ptr, let->val->data.var), sizeof(*arg_val));
+#if 0
+		fprintf(stderr, "LET TMP: %p -> %p\n",
+				let->val->data.var->arg->let->ptr,
+				arg_val->ptr);
+#endif
+		break;
+	case PSI_LET_NULL:
+		if (darg->var->array_size) {
+			arg_val->ptr = ecalloc(darg->var->array_size, sizeof(*arg_val));
+			darg->mem = arg_val->ptr;
+		} else {
+			memset(arg_val, 0, sizeof(*arg_val));
+		}
+		break;
+	case PSI_LET_CALLOC:
+		arg_val->ptr = psi_do_calloc(let->val->data.alloc);
+		darg->mem = arg_val->ptr;
+		break;
+	case PSI_LET_NUMEXP:
+		arg_val->zend.lval = psi_long_num_exp(let->val->data.num, NULL);
+		break;
+	case PSI_LET_FUNC:
+		iarg = let->val->data.func->arg;
+
+		if (!(darg->ptr = psi_let_val(let->val->data.func->type, iarg, darg->ptr, real_decl_type(darg->type)->strct, &darg->mem))) {
+			return NULL;
+		}
+	}
+
+	if (let->val && let->val->flags.one.is_reference) {
+		return let->ptr = &darg->ptr;
+	} else {
+		return let->ptr = darg->ptr;
+	}
+}
+
+static inline void psi_do_set(zval *return_value, set_value *set)
+{
+	decl_arg *set_arg = set->vars->vars[0]->arg;
+
+	zval_dtor(return_value);
+	set->func->handler(return_value, set, set_arg->let ? set_arg->let->ptr : set_arg->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_free(free_stmt *fre)
+{
+	size_t i, j;
+
+	for (i = 0; i < fre->calls->count; ++i) {
+		free_call *f = fre->calls->list[i];
+
+		for (j = 0; j < f->vars->count; ++j) {
+			decl_var *dvar = f->vars->vars[j];
+			decl_arg *darg = dvar->arg;
+			impl_val *fval = darg->let ? darg->let->ptr : darg->ptr;
+
+			f->decl->call.args[j] = deref_impl_val(fval, dvar);
+		}
+
+		/* FIXME: check in validate_* that free functions return scalar */
+		PSI_ContextCall(&PSI_G(context), &f->decl->call, NULL);
+	}
+}
+
+static inline void psi_clean_array_struct(decl_arg *darg) {
+	if (darg->let
+	&&	darg->let->val->kind == PSI_LET_FUNC
+	&&	darg->let->val->data.func->type == PSI_T_ARRVAL) {
+		decl_type *type = real_decl_type(darg->type);
+
+		if (type->type == PSI_T_STRUCT) {
+			void **ptr = (void **) ((char *) darg->mem + type->strct->size);
+
+			while (*ptr) {
+				efree(*ptr++);
+			}
+		}
+	}
+}
+
+static inline void psi_do_clean(impl *impl)
+{
+	size_t i;
+
+	if (impl->decl->func->ptr != &impl->decl->func->val) {
+		efree(impl->decl->func->ptr);
+		impl->decl->func->ptr = &impl->decl->func->val;
+	}
+	for (i = 0; i < impl->func->args->count; ++i ) {
+		impl_arg *iarg = impl->func->args->args[i];
+
+		switch (iarg->type->type) {
+		case PSI_T_STRING:
+			if (iarg->val.zend.str) {
+				zend_string_release(iarg->val.zend.str);
+			}
+			break;
+		}
+	}
+
+	if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
+		decl_arg *darg = impl->decl->args->args[i];
+
+		if (darg->mem) {
+			psi_clean_array_struct(darg);
+			efree(darg->mem);
+			darg->mem = NULL;
+		}
+		darg->ptr = &darg->val;
+	}
+
+	if (impl->func->args->vararg.args) {
+		free_impl_args(impl->func->args->vararg.args);
+		impl->func->args->vararg.args = NULL;
+	}
+	if (impl->func->args->vararg.types) {
+		efree(impl->func->args->vararg.types);
+		impl->func->args->vararg.types = NULL;
+	}
+	if (impl->func->args->vararg.values) {
+		efree(impl->func->args->vararg.values);
+		impl->func->args->vararg.values = NULL;
+	}
+	if (impl->func->args->vararg.free_list) {
+		void **list = impl->func->args->vararg.free_list;
+
+		while (*list) {
+			efree(*list++);
+		}
+
+		efree(impl->func->args->vararg.free_list);
+		impl->func->args->vararg.free_list = NULL;
+	}
+}
+
+
+static inline void psi_do_args(impl *impl) {
+	size_t i;
+
+	for (i = 0; i < impl->decl->args->count; ++i) {
+		impl->decl->call.args[i] = impl->decl->args->args[i]->let->ptr;
+	}
+
+	if (!impl->decl->func->var->pointer_level) {
+		decl_type *real = real_decl_type(impl->decl->func->type);
+
+		switch (real->type) {
+		case PSI_T_STRUCT:
+			impl->decl->func->ptr = psi_array_to_struct(real->strct, NULL);
+			break;
+		}
+	}
+}
+
+static inline impl_vararg *psi_do_varargs(impl *impl) {
+	size_t i, j;
+	impl_vararg *va = &impl->func->args->vararg;
+	size_t vacount = va->args->count;
+
+
+	if (!vacount) {
+		return NULL;
+	}
+
+	va->types = ecalloc(vacount, sizeof(*va->types));
+	va->values = ecalloc(vacount, sizeof(*va->values));
+
+	for (i = 0, j = 0; i < vacount; ++i) {
+		impl_arg *vaarg = va->args->args[i];
+		void *to_free = NULL;
+		token_t let_fn, vatype = va->name->type->type;
+
+		if (vatype == PSI_T_MIXED) {
+			switch (Z_TYPE_P(vaarg->_zv)) {
+			case IS_TRUE:
+			case IS_FALSE:	vatype = PSI_T_BOOL;	break;
+			case IS_LONG:	vatype = PSI_T_INT;		break;
+			case IS_DOUBLE:	vatype = PSI_T_FLOAT;	break;
+			default:		vatype = PSI_T_STRING;	break;
+			}
+		}
+
+
+		switch (vatype) {
+		case PSI_T_BOOL:	let_fn = PSI_T_BOOLVAL;	break;
+		case PSI_T_INT:		let_fn = PSI_T_INTVAL;	break;
+		case PSI_T_FLOAT:
+		case PSI_T_DOUBLE:	let_fn = PSI_T_FLOATVAL;break;
+		case PSI_T_STRING:	let_fn = PSI_T_STRVAL;	break;
+		EMPTY_SWITCH_DEFAULT_CASE();
+		}
+
+		va->types[i] = vatype;
+		/* FIXME: varargs with struct-by-value :) */
+		if (!psi_let_val(let_fn, vaarg, &va->values[i], NULL, &to_free)) {
+			return NULL;
+		}
+
+		if (to_free) {
+			if (!va->free_list) {
+				va->free_list = ecalloc(vacount - i + 1, sizeof(*va->free_list));
+			}
+			va->free_list[j++] = to_free;
+		}
+	}
+
+	return va;
+}
+
+void psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl)
+{
+	size_t i;
+	impl_vararg *va = NULL;
+
+	memset(impl->decl->func->ptr, 0, sizeof(impl_val));
+
+	if (SUCCESS != psi_parse_args(execute_data, impl)) {
+		return;
+	}
+
+	for (i = 0; i < impl->stmts->let.count; ++i) {
+		let_stmt *let = impl->stmts->let.list[i];
+
+		if (!psi_do_let(let)) {
+			psi_do_return(return_value, impl->stmts->ret.list[0]);
+			psi_do_clean(impl);
+			return;
+		}
+	}
+
+	if (impl->decl->args) {
+		psi_do_args(impl);
+
+		if (impl->func->args->vararg.args) {
+			va = psi_do_varargs(impl);
+		}
+	}
+
+	PSI_ContextCall(&PSI_G(context), &impl->decl->call, va);
+	psi_do_return(return_value, impl->stmts->ret.list[0]);
+
+	for (i = 0; i < impl->stmts->set.count; ++i) {
+		set_stmt *set = impl->stmts->set.list[i];
+
+		if (set->arg->_zv) {
+			psi_do_set(set->arg->_zv, set->val);
+		}
+	}
+
+	for (i = 0; i < impl->stmts->fre.count; ++i) {
+		free_stmt *fre = impl->stmts->fre.list[i];
+
+		psi_do_free(fre);
+	}
+	psi_do_clean(impl);
+}