X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fcontext.c;h=1a20af9decd4b57e97757670ea9b881af33b4c62;hp=5e66a45f0d4a41b155f5b780ccdb8f832c0b6410;hb=481515f92d2baa83dea51d2effcbde711aa99e3b;hpb=b4508f6b917660970f887894a6aaed6c220a2c72

diff --git a/src/context.c b/src/context.c
index 5e66a45..1a20af9 100644
--- a/src/context.c
+++ b/src/context.c
@@ -150,10 +150,24 @@ static struct psi_predef_const {
 
 PSI_MACROS
 
+size_t psi_fread(void *ptr, size_t size, size_t nmemb, FILE *stream)
+{
+	size_t rv = fread(ptr, size, nmemb, stream);
+	return rv;
+}
+
+FILE *psi_fopen(const char *path, const char *mode)
+{
+	FILE *f = fopen(path, mode);
+	return f;
+}
+
 static struct psi_func_redir {
 	const char *name;
 	void (*func)(void);
 } psi_func_redirs[] = {
+		{"fopen", (void (*)(void)) psi_fopen},
+		{"fread", (void (*)(void)) psi_fread},
 	PSI_REDIRS
 	{0}
 };
@@ -224,8 +238,8 @@ static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
 	for (stdtyp = &psi_std_types[0]; stdtyp->type_tag; ++stdtyp) {
 		if (!strcmp(type->name, stdtyp->type_name)) {
 			type->type = stdtyp->type_tag;
+			return 1;
 		}
-		return 1;
 	}
 
 	return 0;
@@ -397,31 +411,87 @@ static inline int validate_decl(PSI_Data *data, void *dl, decl *decl) {
 	}
 	return 1;
 }
-static inline decl_arg *locate_decl_var_arg(decl_var *var, decl_args *args) {
+static inline decl_arg *locate_decl_var_arg(decl_var *var, decl_args *args, decl_arg *func) {
 	size_t i;
 
 	for (i = 0; i < args->count; ++i) {
 		decl_arg *arg = args->args[i];
 
 		if (!strcmp(var->name, arg->var->name)) {
+			ZEND_ASSERT(!var->arg || var->arg == arg);
 			return var->arg = arg;
 		}
 	}
+	if (func && !strcmp(var->name, func->var->name)) {
+		return var->arg = func;
+	}
 
 	return NULL;
 }
 static inline decl_arg *locate_struct_member(decl_struct *s, decl_var *var) {
 	if (s->args) {
-		return locate_decl_var_arg(var, s->args);
+		return locate_decl_var_arg(var, s->args, NULL);
 	}
 
 	return NULL;
 }
-static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
+static inline constant *locate_num_exp_constant(num_exp *exp, constants *consts) {
 	size_t i;
-	decl_type *ref_type = real_decl_type(ref->type);
-	decl_var *set_var = set->vars->vars[0];
 
+	for (i = 0; i < consts->count; ++i) {
+		constant *cnst = consts->list[i];
+
+		if (!strcmp(cnst->name, exp->u.numb)) {
+			free(exp->u.numb);
+			return exp->u.cnst = cnst;
+		}
+	}
+
+	return NULL;
+}
+static inline int validate_num_exp(PSI_Data *data, decl_args *dargs, decl_arg *func, num_exp *exp) {
+	if (exp->operand) {
+		switch (exp->operator) {
+		case PSI_T_PLUS:
+			exp->calculator = psi_calc_add;
+			break;
+		case PSI_T_MINUS:
+			exp->calculator = psi_calc_sub;
+			break;
+		case PSI_T_ASTERISK:
+			exp->calculator = psi_calc_mul;
+			break;
+		case PSI_T_SLASH:
+			exp->calculator = psi_calc_div;
+			break;
+		EMPTY_SWITCH_DEFAULT_CASE();
+		}
+		if (!validate_num_exp(data, dargs, func, exp->operand)) {
+			return 0;
+		}
+	}
+	switch (exp->t) {
+	case PSI_T_NAME:
+		if (!locate_decl_var_arg(exp->u.dvar, dargs, func)) {
+			data->error(PSI_WARNING, "Unknown variable '%s' in numeric expression",
+					exp->u.dvar->name);
+			return 0;
+		}
+		return 1;
+	case PSI_T_NSNAME:
+		if (!locate_num_exp_constant(exp, data->consts)) {
+			data->error(PSI_WARNING, "Unknown constant '%s' in numeric expression",
+					exp->u.numb);
+			return 0;
+		}
+		return 1;
+	case PSI_T_NUMBER:
+		return 1;
+	default:
+		return 0;
+	}
+}
+static inline int validate_set_value_handler(set_value *set) {
 	switch (set->func->type) {
 	case PSI_T_TO_BOOL:
 		set->func->handler = psi_to_bool;
@@ -438,48 +508,95 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
 	case PSI_T_TO_ARRAY:
 		set->func->handler = psi_to_array;
 		break;
-	EMPTY_SWITCH_DEFAULT_CASE();
+	case PSI_T_TO_OBJECT:
+		set->func->handler = psi_to_object;
+		break;
+	case PSI_T_VOID:
+		set->func->handler = psi_to_void;
+		break;
+	default:
+		return 0;
 	}
+	return 1;
+}
+static inline void decl_var_arg_v(decl_args *args, va_list argp) {
+	int argc;
+	decl_arg **argv;
 
-	for (i = 1; i < set->vars->count; ++i) {
-		if (!locate_decl_var_arg(set->vars->vars[i], ref_list)) {
-			return 0;
+	memset(args, 0, sizeof(*args));
+
+	while ((argc = va_arg(argp, int))) {
+		argv = va_arg(argp, decl_arg **);
+		while (argc--) {
+			add_decl_arg(args, *argv++);
 		}
 	}
-	set_var->arg = ref;
+}
+static inline int validate_set_value_ex(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
+	size_t i;
+	decl_type *ref_type;
+	decl_var *set_var = set->vars->vars[0];
 
-	if (set->count && (set->func->type != PSI_T_TO_ARRAY || (ref_type->type != PSI_T_STRUCT && !ref->var->arg->var->pointer_level))) {
-		data->error(E_WARNING, "Inner `set` statement casts only work with to_array() casts on structs or pointers");
+	if (!validate_set_value_handler(set)) {
+		data->error(PSI_WARNING, "Invalid cast '%s'", set->func->name);
 		return 0;
 	}
 
+	for (i = 0; i < set->vars->count; ++i) {
+		if (!locate_decl_var_arg(set->vars->vars[i], ref_list, NULL)) {
+			data->error(PSI_WARNING, "Unknown variable '%s'", set->vars->vars[i]->name);
+			return 0;
+		}
+	}
+
+	if (!ref) {
+		ref = set_var->arg;
+	}
+	ref_type = real_decl_type(ref->type);
+
+	if (set->count) {
+		int is_to_array = (set->func->type == PSI_T_TO_ARRAY);
+		int is_pointer_to_struct = (ref_type->type == PSI_T_STRUCT && ref->var->pointer_level);
+
+		if (!is_to_array && !is_pointer_to_struct) {
+			data->error(E_WARNING, "Inner `set` statement casts only work with "
+					"to_array() casts on structs or pointers: %s(%s...", set->func->name, set->vars->vars[0]->name);
+			return 0;
+		}
+	}
+	if (set->num) {
+		if (!validate_num_exp(data, ref_list, ref, set->num)) {
+			return 0;
+		}
+	}
+
 	if (ref_type->type == PSI_T_STRUCT) {
 		/* to_array(struct, to_...) */
 		for (i = 0; i < set->count; ++i) {
 			decl_var *sub_var = set->inner[i]->vars->vars[0];
 			decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
 
+			set->inner[i]->outer.set = set;
 			if (sub_ref) {
-				if (!validate_set_value(data, set->inner[i], sub_ref, ref_type->strct->args)) {
+				if (!validate_set_value_ex(data, set->inner[i], sub_ref, ref_type->strct->args)) {
 					return 0;
 				}
-				set->inner[i]->outer.set = set;
 			}
 		}
 	} else if (set->count == 1) {
 		/* to_array(ptr, to_string(*ptr)) */
 		decl_var *sub_var = set->inner[0]->vars->vars[0];
-		decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list);
+		decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list, ref);
 
+		set->inner[0]->outer.set = set;
 		if (sub_ref) {
 			if (strcmp(sub_var->name, set_var->name)) {
 				data->error(E_WARNING, "Inner `set` statement casts on pointers must reference the same variable");
 				return 0;
 			}
-			if (!validate_set_value(data, set->inner[0], sub_ref, ref_list)) {
+			if (!validate_set_value_ex(data, set->inner[0], sub_ref, ref_list)) {
 				return 0;
 			}
-			set->inner[0]->outer.set = set;
 		}
 	} else if (set->count > 1) {
 		data->error(E_WARNING, "Inner `set` statement casts on pointers may only occur once");
@@ -488,6 +605,21 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
 
 	return 1;
 }
+static inline int validate_set_value(PSI_Data *data, set_value *set, ...) {
+	va_list argp;
+	decl_args args = {0};
+	int check;
+
+	va_start(argp, set);
+	decl_var_arg_v(&args, argp);
+	va_end(argp);
+
+	check = validate_set_value_ex(data, set, NULL, &args);
+	if (args.args) {
+		free(args.args);
+	}
+	return check;
+}
 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
 	if (decls) {
 		size_t i;
@@ -527,7 +659,7 @@ static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 		return 0;
 	}
 
-	if (!validate_set_value(data, ret->set, ret->decl, impl->decl->args)) {
+	if (!validate_set_value(data, ret->set, 1, &ret->decl, impl->decl->args ? (int) impl->decl->args->count : 0, impl->decl->args ? impl->decl->args->args : NULL, 0)) {
 		return 0;
 	}
 
@@ -535,9 +667,21 @@ static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 
 	return 1;
 }
+
 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
 	size_t i, j;
 	/* we can have multiple let stmts */
+	/* check that we have a decl arg for every let stmt */
+	for (i = 0; i < impl->stmts->let.count; ++i) {
+		let_stmt *let = impl->stmts->let.list[i];
+
+		if (!locate_decl_var_arg(let->var, impl->decl->args, impl->decl->func)) {
+			data->error(PSI_WARNING, "Unknown variable '%s' in `let` statement"
+					" of implementation '%s'", let->var->name, impl->func->name);
+			return 0;
+		}
+	}
+
 	/* check that we have a let stmt for every decl arg */
 	if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
 		decl_arg *darg = impl->decl->args->args[i];
@@ -566,9 +710,10 @@ static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
 		int check = 0;
 
 		if (let->val && let->val->func && let->val->func->alloc) {
-			if (!validate_decl_type(data, let->val->func->alloc->type)) {
-				data->error(PSI_WARNING, "Cannot use '%s' as type for calloc in `let` statement",
-					let->val->func->alloc->type->name);
+			if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->func->alloc->nmemb)) {
+				return 0;
+			}
+			if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->func->alloc->size)) {
 				return 0;
 			}
 		}
@@ -625,10 +770,10 @@ static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
 
 				if (!strcmp(set_var->name, set_arg->var->name)) {
 					check = 1;
-					if (!validate_set_value(data, set->val, set_arg, impl->decl->args)) {
+					set_var->arg = set_arg;
+					if (!validate_set_value(data, set->val, 1, &set_arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
 						return 0;
 					}
-					set_var->arg = set_arg;
 					break;
 				}
 			}
@@ -953,9 +1098,13 @@ void PSI_ContextBuild(PSI_Context *C, const char *paths)
 					continue;
 				}
 
-				while (-1 != PSI_ParserScan(&P)) {
+				while (0 < PSI_ParserScan(&P)) {
 					PSI_ParserParse(&P, PSI_TokenAlloc(&P));
-				};
+					if (P.num == PSI_T_EOF) {
+						break;
+					}
+				}
+
 				PSI_ParserParse(&P, NULL);
 				PSI_ContextValidate(C, &P);
 				PSI_ParserDtor(&P);
@@ -1044,43 +1193,223 @@ static inline void dump_decl_arg(int fd, decl_arg *a) {
 	dprintf(fd, " ");
 	dump_decl_var(fd, a->var);
 }
+static inline void dump_level(int fd, unsigned level) {
+	dprintf(fd, "%.*s", level, "\t\t\t\t\t\t\t\t\t");
+}
+static inline void dump_num_exp(int fd, num_exp *exp) {
+	while (exp) {
+		switch (exp->t) {
+		case PSI_T_NUMBER:
+			dprintf(fd, "%s", exp->u.numb);
+			break;
+		case PSI_T_NSNAME:
+			dprintf(fd, "%s", exp->u.cnst->name);
+			break;
+		case PSI_T_NAME:
+			dump_decl_var(fd, exp->u.dvar);
+			break;
+		EMPTY_SWITCH_DEFAULT_CASE();
+		}
+		if (exp->operand) {
+			char op;
+
+			switch (exp->operator) {
+			case PSI_T_PLUS:	op = '+'; break;
+			case PSI_T_MINUS:	op = '-'; break;
+			case PSI_T_ASTERISK:op = '*'; break;
+			case PSI_T_SLASH:	op = '/'; break;
+			EMPTY_SWITCH_DEFAULT_CASE();
+			}
+			dprintf(fd, " %c ", op);
+		}
+		exp = exp->operand;
+	}
+}
+static inline void dump_impl_set_value(int fd, set_value *set, unsigned level) {
+	size_t i;
+
+	if (level > 1) {
+		/* only if not directly after `set ...` */
+		dump_level(fd, level);
+	}
+	dprintf(fd, "%s(", set->func->name);
+
+	for (i = 0; i < set->vars->count; ++i) {
+		decl_var *svar = set->vars->vars[i];
+		if (i) {
+			dprintf(fd, ", ");
+		}
+		dump_decl_var(fd, svar);
+	}
+	if (set->num) {
+		dprintf(fd, ", ");
+		dump_num_exp(fd, set->num);
+	}
+	if (set->inner) {
+		dprintf(fd, ",\n");
+		for (i = 0; i < set->count; ++i) {
+			dump_impl_set_value(fd, set->inner[i], level+1);
+		}
+		/* only if inner stmts, i.e. with new lines, were dumped */
+		dump_level(fd, level);
+	}
+	if (level > 1) {
+		dprintf(fd, "),\n");
+	} else {
+		dprintf(fd, ");\n");
+	}
+}
 void PSI_ContextDump(PSI_Context *C, int fd)
 {
-	size_t i, j;
+	size_t i, j, k, l;
 
 #ifdef HAVE_LIBJIT
 	if (C->ops == PSI_Libjit()) {
-		dprintf(fd, "#PSI(libjit)\n");
+		dprintf(fd, "// psi.engine=jit\n");
 	}
 #endif
 #ifdef HAVE_LIBFFI
 	if (C->ops == PSI_Libffi()) {
-		dprintf(fd, "#PSI(libffi)\n");
+		dprintf(fd, "// psi.engine=ffi\n");
 	}
 #endif
+	dprintf(fd, "\n");
 
-	if (C->defs) for (i = 0; i < C->defs->count; ++i) {
-		decl_typedef *tdef = C->defs->list[i];
+	if (C->defs) {
+		for (i = 0; i < C->defs->count; ++i) {
+			decl_typedef *tdef = C->defs->list[i];
 
-		dprintf(fd, "typedef ");
-		dump_decl_type(fd, tdef->type);
-		dprintf(fd, " %s;\n", tdef->alias);
+			dprintf(fd, "typedef ");
+			dump_decl_type(fd, tdef->type);
+			dprintf(fd, " %s;\n", tdef->alias);
+		}
+		dprintf(fd, "\n");
 	}
-	if (C->structs) for (i = 0; i < C->structs->count; ++i) {
-		decl_struct *strct = C->structs->list[i];
-		decl_arg *sarg = NULL;
 
-		dprintf(fd, "struct %s::(%zu) {\n", strct->name, strct->size);
-		for (j = 0; j < strct->args->count; ++j) {
-			sarg = strct->args->args[j];
-			dprintf(fd, "\t");
-			dump_decl_arg(fd, sarg);
-			dprintf(fd, "::(%zu, %zu);\n", sarg->layout->pos, sarg->layout->len);
+	if (C->structs) {
+		for (i = 0; i < C->structs->count; ++i) {
+			decl_struct *strct = C->structs->list[i];
+
+			dprintf(fd, "struct %s::(%zu) {\n", strct->name, strct->size);
+			if (strct->args) for (j = 0; j < strct->args->count; ++j) {
+				decl_arg *sarg = strct->args->args[j];
+
+				dprintf(fd, "\t");
+				dump_decl_arg(fd, sarg);
+				dprintf(fd, "::(%zu, %zu);\n", sarg->layout->pos, sarg->layout->len);
+			}
+			dprintf(fd, "}\n");
 		}
-		dprintf(fd, "}\n");
+		dprintf(fd, "\n");
 	}
+	if (C->consts) {
+		for (i = 0; i < C->consts->count; ++i) {
+			constant *cnst = C->consts->list[i];
 
-	dprintf(fd, "\n");
+			dprintf(fd, "const %s %s = ", cnst->type->name, cnst->name);
+			if (cnst->val->type == PSI_T_QUOTED_STRING) {
+				dprintf(fd, "\"%s\";\n", cnst->val->text);
+			} else {
+				dprintf(fd, "%s;\n", cnst->val->text);
+			}
+		}
+		dprintf(fd, "\n");
+	}
+	if (C->decls) {
+		for (i = 0; i < C->decls->count; ++i) {
+			decl *decl = C->decls->list[i];
+
+			dprintf(fd, "%s ", decl->abi->convention);
+			dump_decl_arg(fd, decl->func);
+			dprintf(fd, "(");
+			if (decl->args) for (j = 0; j < decl->args->count; ++j) {
+				if (j) {
+					dprintf(fd, ", ");
+				}
+				dump_decl_arg(fd, decl->args->args[j]);
+			}
+			dprintf(fd, ");\n");
+		}
+		dprintf(fd, "\n");
+	}
+	if (C->impls) {
+		for (i = 0; i < C->impls->count; ++i) {
+			impl *impl = C->impls->list[i];
+
+			dprintf(fd, "function %s(", impl->func->name);
+			if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
+				impl_arg *iarg = impl->func->args->args[j];
+
+				dprintf(fd, "%s%s %s$%s",
+						j ? ", " : "",
+						iarg->type->name,
+						iarg->var->reference ? "&" : "",
+						iarg->var->name);
+				if (iarg->def) {
+					dprintf(fd, " = %s", iarg->def->text);
+				}
+			}
+			dprintf(fd, ") : %s%s {\n",
+					impl->func->return_reference ? "&":"",
+					impl->func->return_type->name);
+			if (impl->stmts) {
+				for (j = 0; j < impl->stmts->let.count; ++j) {
+					let_stmt *let = impl->stmts->let.list[j];
+
+					dprintf(fd, "\tlet %s", let->var->name);
+					if (let->val) {
+						dprintf(fd, " = %s", let->val->is_reference ? "&" : "");
+						if (let->val->func) {
+							dprintf(fd, "%s(", let->val->func->name);
+							if (let->val->func->alloc) {
+								dump_num_exp(fd, let->val->func->alloc->nmemb);
+								dprintf(fd, ", ");
+								dump_num_exp(fd, let->val->func->alloc->size);
+							} else {
+								dprintf(fd, "$%s", let->val->var->name);
+							}
+							dprintf(fd, ");\n");
+						} else {
+							dprintf(fd, "NULL;\n");
+						}
+					}
+				}
+				for (j = 0; j < impl->stmts->ret.count; ++j) {
+					return_stmt *ret = impl->stmts->ret.list[j];
+
+					dprintf(fd, "\treturn ");
+					dump_impl_set_value(fd, ret->set, 1);
+				}
+				for (j = 0; j < impl->stmts->set.count; ++j) {
+					set_stmt *set = impl->stmts->set.list[j];
+
+					dprintf(fd, "\tset $%s = ", set->var->name);
+					dump_impl_set_value(fd, set->val, 1);
+				}
+				for (j = 0; j < impl->stmts->fre.count; ++j) {
+					free_stmt *fre = impl->stmts->fre.list[j];
+
+					dprintf(fd, "\tfree ");
+					for (k = 0; k < fre->calls->count; ++k) {
+						free_call *call = fre->calls->list[k];
+
+						if (k) {
+							dprintf(fd, ", ");
+						}
+						dprintf(fd, "%s(", call->func);
+						for (l = 0; l < call->vars->count; ++l) {
+							decl_var *fvar = call->vars->vars[l];
+
+							dump_decl_var(fd, fvar);
+						}
+						dprintf(fd, ");\n");
+					}
+				}
+			}
+			dprintf(fd, "}\n");
+		}
+		dprintf(fd, "\n");
+	}
 }
 
 void PSI_ContextDtor(PSI_Context *C)