X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=src%2Ftypes%2Fimpl.c;h=d2c87b9eff47818ab3fa386b57d7de9cb7b315cc;hb=09529efcde471127419e141807b83b37077003a0;hp=33d3301fa8a7863abe0e63dfbebabc072f8eb575;hpb=9bcb1df0786a8193d65949c857baaba2f4296e84;p=m6w6%2Fext-psi

diff --git a/src/types/impl.c b/src/types/impl.c
index 33d3301..d2c87b9 100644
--- a/src/types/impl.c
+++ b/src/types/impl.c
@@ -39,6 +39,7 @@ struct psi_impl *psi_impl_init(struct psi_impl_func *func,
 	impl->stmts.let = psi_plist_init((psi_plist_dtor) psi_let_stmt_free);
 	impl->stmts.set = psi_plist_init((psi_plist_dtor) psi_set_stmt_free);
 	impl->stmts.fre = psi_plist_init((psi_plist_dtor) psi_free_stmt_free);
+	impl->stmts.ass = psi_plist_init((psi_plist_dtor) psi_assert_stmt_free);
 
 	while (psi_plist_get(stmts, i++, &abstract_stmt)) {
 		switch ((*abstract_stmt)->type) {
@@ -55,6 +56,10 @@ struct psi_impl *psi_impl_init(struct psi_impl_func *func,
 		case PSI_T_FREE:
 			impl->stmts.fre = psi_plist_add(impl->stmts.fre, &abstract_stmt);
 			break;
+		case PSI_T_PRE_ASSERT:
+		case PSI_T_POST_ASSERT:
+			impl->stmts.ass = psi_plist_add(impl->stmts.ass, &abstract_stmt);
+			break;
 		default:
 			assert(0);
 		}
@@ -77,6 +82,7 @@ void psi_impl_free(struct psi_impl **impl_ptr)
 		psi_plist_free(impl->stmts.let);
 		psi_plist_free(impl->stmts.set);
 		psi_plist_free(impl->stmts.fre);
+		psi_plist_free(impl->stmts.ass);
 		free(impl);
 	}
 }
@@ -88,12 +94,16 @@ void psi_impl_dump(int fd, struct psi_impl *impl)
 	struct psi_let_stmt *let;
 	struct psi_set_stmt *set;
 	struct psi_free_stmt *fre;
+	struct psi_assert_stmt *ass;
 
 	psi_impl_func_dump(fd, impl->func);
 	dprintf(fd, " {\n");
 	for (i = 0; psi_plist_get(impl->stmts.let, i, &let); ++i) {
 		psi_let_stmt_dump(fd, let);
 	}
+	for (i = 0; psi_plist_get(impl->stmts.ass, i, &ass); ++i) {
+		psi_assert_stmt_dump(fd, ass);
+	}
 	for (i = 0; psi_plist_get(impl->stmts.ret, i, &ret); ++i) {
 		psi_return_stmt_dump(fd, ret);
 	}
@@ -106,21 +116,27 @@ void psi_impl_dump(int fd, struct psi_impl *impl)
 	dprintf(fd, "}\n");
 }
 
-bool psi_impl_validate(struct psi_data *data, struct psi_impl *impl)
+bool psi_impl_validate(struct psi_data *data, struct psi_impl *impl,
+		struct psi_validate_scope *scope)
 {
-	if (!psi_impl_func_validate(data, impl->func)) {
+	scope->impl = impl;
+
+	if (!psi_impl_func_validate(data, impl->func, scope)) {
+		return false;
+	}
+	if (!psi_return_stmt_validate(data, scope)) {
 		return false;
 	}
-	if (!psi_return_stmt_validate(data, impl)) {
+	if (!psi_let_stmts_validate(data, scope)) {
 		return false;
 	}
-	if (!psi_let_stmts_validate(data, impl)) {
+	if (!psi_set_stmts_validate(data, scope)) {
 		return false;
 	}
-	if (!psi_set_stmts_validate(data, impl)) {
+	if (!psi_assert_stmts_validate(data, scope)) {
 		return false;
 	}
-	if (!psi_free_stmts_validate(data, impl)) {
+	if (!psi_free_stmts_validate(data, scope)) {
 		return false;
 	}
 	return true;
@@ -154,11 +170,36 @@ void psi_impl_stmt_free(struct psi_token ***abstract_stmt)
 	case PSI_T_FREE:
 		psi_free_stmt_free((void *) abstract_stmt);
 		break;
+	case PSI_T_PRE_ASSERT:
+	case PSI_T_POST_ASSERT:
+		psi_assert_stmt_free((void *) abstract_stmt);
+		break;
 	default:
 		assert(0);
 	}
 }
 
+struct psi_decl_arg *psi_impl_get_decl_arg(struct psi_impl *impl,
+		struct psi_decl_var *var)
+{
+	struct psi_return_stmt *ret;
+
+	if (psi_plist_get(impl->stmts.ret, 0, &ret)) {
+		if (ret->exp->args) {
+			size_t i = 0;
+			struct psi_decl_var *arg;
+
+			while (psi_plist_get(ret->exp->args, i++, &arg)) {
+				if (zend_string_equals(var->name, arg->name)) {
+					return var->arg = arg->arg;
+				}
+			}
+		}
+	}
+
+	return psi_decl_get_arg(impl->decl, var);
+}
+
 struct psi_let_stmt *psi_impl_get_let(struct psi_impl *impl,
 		struct psi_decl_var* var)
 {
@@ -180,7 +221,7 @@ struct psi_impl_arg *psi_impl_get_arg(struct psi_impl *impl,
 	struct psi_impl_arg *iarg;
 
 	while (psi_plist_get(impl->func->args, i++, &iarg)) {
-		if (!strcmp(var->name, iarg->var->name)) {
+		if (zend_string_equals(var->name, iarg->var->name)) {
 			return var->arg = iarg;
 		}
 	}
@@ -197,7 +238,7 @@ struct psi_decl_arg *psi_impl_get_temp_let_arg(struct psi_impl *impl,
 		if (let->exp->kind != PSI_LET_TMP) {
 			continue;
 		}
-		if (strcmp(let->exp->var->name, var->name)) {
+		if (!zend_string_equals(let->exp->var->name, var->name)) {
 			continue;
 		}
 		return var->arg = let->exp->var->arg;