X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fvalidator.c;h=7bc2f701989c698c63a0e491f4031ab38b30dd19;hp=4e38929e1d42f9411468af53676f7b8b49233e9b;hb=d60f4bbdd315ddf42dbafeff0fd3d87d2e7a51f7;hpb=9c72f6584bd95dd6e9c7dcb314a6583f7130362f

diff --git a/src/validator.c b/src/validator.c
index 4e38929..7bc2f70 100644
--- a/src/validator.c
+++ b/src/validator.c
@@ -6,6 +6,8 @@
 
 #include <jit/jit.h>
 
+#include "php.h"
+#include "php_psi.h"
 #include "validator.h"
 
 PSI_Validator *PSI_ValidatorInit(PSI_Validator *V, PSI_Parser *P)
@@ -61,6 +63,7 @@ static int validate_lib(PSI_Validator *V) {
 	}
 	return 1;
 }
+
 static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
 	size_t i;
 
@@ -75,14 +78,32 @@ static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
 	}
 	return 0;
 }
-static inline int validate_decl_type(PSI_Validator *V, decl_arg *arg, decl_type *type) {
-	if (type->type == PSI_T_NAME) {
+static inline int locate_decl_type_struct(decl_structs *structs, decl_type *type) {
+	size_t i;
+
+	if (type->strct) {
+		return 1;
+	}
+	for (i = 0; i < structs->count; ++i) {
+		if (!strcmp(structs->list[i]->name, type->name)) {
+			type->strct = structs->list[i];
+			return 1;
+		}
+	}
+	return 0;
+}
+static inline int validate_decl_type(PSI_Validator *V, decl_type *type) {
+	switch (type->type) {
+	case PSI_T_NAME:
 		if (!V->defs || !locate_decl_type_alias(V->defs, type)) {
-			V->error(PSI_WARNING, "Cannot use '%s' as type for '%s';"
-				" Use 'typedef <type> <basic_type>;' statement",
-				type->name, arg->var->name);
 			return 0;
 		}
+		return validate_decl_type(V, type->real);
+	case PSI_T_STRUCT:
+		if (!V->structs || !locate_decl_type_struct(V->structs, type)) {
+			return 0;
+		}
+		break;
 	}
 	return 1;
 }
@@ -106,6 +127,42 @@ static inline int validate_typedefs(PSI_Validator *V) {
 
 	return 1;
 }
+static const char * const abi_ccs[] = {
+		"default", /* \                 */
+		"extern",  /*  > - all the same */
+		"cdecl",   /* /                 */
+		"stdcall",
+		"fastcall",
+};
+static inline int validate_decl_abi(PSI_Validator *V, decl_abi *abi) {
+	size_t i;
+
+	for (i = 0; i < sizeof(abi_ccs)/sizeof(char*); ++i) {
+		if (strcasecmp(abi->convention, abi_ccs[i])) {
+			return 1;
+		}
+	}
+	V->error(PSI_WARNING, "Invalid calling convention: '%s'", abi->convention);
+	return 0;
+}
+static inline int validate_decl_arg(PSI_Validator *V, decl_arg *arg) {
+	if (!validate_decl_type(V, arg->type)) {
+		V->error(PSI_WARNING, "Cannot use '%s' as type for '%s'",
+			arg->type->name, arg->var->name);
+		return 0;
+	}
+	return 1;
+}
+static inline int validate_decl_args(PSI_Validator *V, decl_args *args) {
+	size_t i;
+
+	for (i = 0; i < args->count; ++i) {
+		if (!validate_decl_arg(V, args->args[i])) {
+			return 0;
+		}
+	}
+	return 1;
+}
 static inline int validate_decl_func(PSI_Validator *V, decl *decl, decl_arg *func)
 {
 	if (!strcmp(func->var->name, "dlsym")) {
@@ -113,58 +170,69 @@ static inline int validate_decl_func(PSI_Validator *V, decl *decl, decl_arg *fun
 		return 0;
 	}
 
-	if (!validate_decl_type(V, func, func->type)) {
+	if (!validate_decl_arg(V, func)) {
 		return 0;
 	}
-
-	decl->dlptr = dlsym(V->dlopened, func->var->name);
+#ifndef RTLD_NEXT
+# define RTLD_NEXT ((void *) -1l)
+#endif
+	decl->dlptr = dlsym(V->dlopened ?: RTLD_NEXT, func->var->name);
 	if (!decl->dlptr) {
 		V->error(PSI_WARNING, "Failed to located symbol '%s': %s",
 			func->var->name, dlerror());
 	}
 	return 1;
 }
-static inline int validate_decl_abi(PSI_Validator *V, decl_abi *abi) {
-	if (strcasecmp(abi->convention, "default")) {
-		V->error(PSI_WARNING, "Invalid calling convention: '%s'", abi->convention);
+static inline int validate_decl(PSI_Validator *V, decl *decl) {
+	if (!validate_decl_abi(V, decl->abi)) {
 		return 0;
 	}
-	/* FIXME */
-	return 1;
-}
-static inline int validate_decl_arg(PSI_Validator *V, decl *decl, decl_arg *arg) {
-	if (!validate_decl_type(V, arg, arg->type)) {
+	if (!validate_decl_func(V, decl, decl->func)) {
+		return 0;
+	}
+	if (decl->args && !validate_decl_args(V, decl->args)) {
 		return 0;
 	}
 	return 1;
 }
-static inline int validate_decl_args(PSI_Validator *V, decl *decl, decl_args *args) {
+static inline int validate_decls(PSI_Validator *V) {
 	size_t i;
 
-	for (i = 0; i < args->count; ++i) {
-		if (!validate_decl_arg(V, decl, args->args[i])) {
+	for (i = 0; i < V->decls->count; ++i) {
+		if (!validate_decl(V, V->decls->list[i])) {
 			return 0;
 		}
 	}
 	return 1;
 }
-static inline int validate_decl(PSI_Validator *V, decl *decl) {
-	if (!validate_decl_abi(V, decl->abi)) {
-		return 0;
-	}
-	if (!validate_decl_func(V, decl, decl->func)) {
+static inline int validate_struct(PSI_Validator *V, decl_struct *s) {
+	size_t i;
+
+	if (!validate_decl_args(V, s->args)) {
 		return 0;
 	}
-	if (decl->args && !validate_decl_args(V, decl, decl->args)) {
-		return 0;
+
+	s->layout = calloc(s->args->count, sizeof(*s->layout));
+	for (i = 0; i < s->args->count; ++i) {
+		decl_arg *darg = s->args->args[i];
+		decl_type *type = real_decl_type(darg->type);
+		token_t t = darg->var->pointer_level ? PSI_T_POINTER : type->type;
+
+		if (i) {
+			decl_struct_layout *l = &s->layout[i-1];
+			s->layout[i].pos = psi_t_align(t, l->pos + l->len);
+		} else {
+			s->layout[i].pos = 0;
+		}
+		s->layout[i].len = psi_t_size(t);
 	}
 	return 1;
 }
-static inline int validate_decls(PSI_Validator *V) {
+static inline int validate_structs(PSI_Validator *V) {
 	size_t i;
 
-	for (i = 0; i < V->decls->count; ++i) {
-		if (!validate_decl(V, V->decls->list[i])) {
+	for (i = 0; i < V->structs->count; ++i) {
+		if (!validate_struct(V, V->structs->list[i])) {
 			return 0;
 		}
 	}
@@ -198,7 +266,7 @@ static inline int validate_impl_func(PSI_Validator *V, impl *impl, impl_func *fu
 	}
 	return 1;
 }
-static inline decl *locate_impl_decl(decls *decls, ret_stmt *ret) {
+static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
 	size_t i;
 
 	for (i = 0; i < decls->count; ++i) {
@@ -213,10 +281,11 @@ static inline int validate_impl_stmts(PSI_Validator *V, impl *impl, impl_stmts *
 	/* okay,
 	 * - we must have exactly one ret stmt delcaring the native func to call and which type cast to apply
 	 * - we can have multiple let stmts; every arg of the ret stmts var (the function to call) must have one
-	 * - we can have any count of set stmts; processing out vars, etc.
+	 * - we can have any count of set stmts; processing out vars
+	 * - we can have any count of free stmts; freeing any out vars
 	 */
-	size_t i, j;
-	ret_stmt *ret;
+	size_t i, j, k;
+	return_stmt *ret;
 	decl *decl;
 
 	 if (!stmts) {
@@ -227,7 +296,7 @@ static inline int validate_impl_stmts(PSI_Validator *V, impl *impl, impl_stmts *
 	if (stmts->ret.count != 1) {
 		if (stmts->ret.count > 1) {
 			V->error(PSI_WARNING, "Too many `ret` statements for implmentation %s;"
-					"found %zu, exactly one is needed",
+					" found %zu, exactly one is needed",
 					impl->func->name, stmts->ret.count);
 		} else {
 			V->error(PSI_WARNING, "Missing `ret` statement for implementation %s",
@@ -245,7 +314,7 @@ static inline int validate_impl_stmts(PSI_Validator *V, impl *impl, impl_stmts *
 	}
 
 	/* check that we have a let stmt for every decl arg */
-	for (i = 0; i < decl->args->count; ++i) {
+	if (decl->args) for (i = 0; i < decl->args->count; ++i) {
 		decl_arg *darg = decl->args->args[i];
 		int check = 0;
 
@@ -260,7 +329,7 @@ static inline int validate_impl_stmts(PSI_Validator *V, impl *impl, impl_stmts *
 		}
 		if (!check) {
 			V->error(PSI_WARNING, "Missing `let` statement for arg '%s %.*s%s'"
-					"of declaration '%s' for implementation '%s'",
+					" of declaration '%s' for implementation '%s'",
 					darg->type->name, (int) darg->var->pointer_level, "*****",
 					darg->var->name, decl->func->var->name, impl->func->name);
 			return 0;
@@ -271,8 +340,15 @@ static inline int validate_impl_stmts(PSI_Validator *V, impl *impl, impl_stmts *
 		let_stmt *let = stmts->let.list[i];
 		int check = 0;
 
-		if (let->val->var) {
-			for (j = 0; j < impl->func->args->count; ++j) {
+		if (let->val && let->val->func && let->val->func->alloc) {
+			if (!validate_decl_type(V, let->val->func->alloc->type)) {
+				V->error(PSI_WARNING, "Cannot use '%s' as type for calloc in `let` statement",
+					let->val->func->alloc->type->name);
+				return 0;
+			}
+		}
+		if (let->val && let->val->var) {
+			if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
 				impl_arg *iarg = impl->func->args->args[j];
 
 				if (!strcmp(let->val->var->name, iarg->var->name)) {
@@ -289,6 +365,78 @@ static inline int validate_impl_stmts(PSI_Validator *V, impl *impl, impl_stmts *
 			}
 		}
 	}
+	/* check that set stmts reference known variables */
+	for (i = 0; i < stmts->set.count; ++i) {
+		set_stmt *set = stmts->set.list[i];
+		int check = 0;
+
+		if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
+			impl_arg *iarg = impl->func->args->args[j];
+
+			if (!strcmp(set->var->name, iarg->var->name)) {
+				set->arg = iarg;
+				check = 1;
+				break;
+			}
+		}
+		if (!check) {
+			V->error(PSI_WARNING, "Unknown variable '$%s' of `set` statement"
+					" of implementation '%s'",
+					set->var->name, impl->func->name);
+			return 0;
+		}
+
+		for (j = 0; j < set->val->vars->count; ++j) {
+			decl_var *set_var = set->val->vars->vars[j];
+
+			check = 0;
+			if (decl->args) for (k = 0; k < decl->args->count; ++k) {
+				decl_arg *set_arg = decl->args->args[k];
+
+				if (!strcmp(set_var->name, set_arg->var->name)) {
+					check = 1;
+					set_var->arg = set_arg;
+					break;
+				}
+			}
+
+			if (!check) {
+				V->error(PSI_WARNING, "Unknown value '%s' of `set` statement"
+						" for variable '$%s' of implementation '%s'",
+						set_var->name, set->arg->var->name, impl->func->name);
+				return 0;
+			}
+		}
+	}
+	/* check free stmts */
+	for (i = 0; i < stmts->fre.count; ++i) {
+		free_stmt *fre = stmts->fre.list[i];
+
+		for (j = 0; j < fre->vars->count; ++j) {
+			decl_var *free_var = fre->vars->vars[j];
+			int check = 0;
+
+			if (!strcmp(free_var->name, decl->func->var->name)) {
+				continue;
+			}
+			if (decl->args) for (k = 0; k < decl->args->count; ++k) {
+				decl_arg *free_arg = decl->args->args[k];
+
+				if (!strcmp(free_var->name, free_arg->var->name)) {
+					check = 1;
+					free_var->arg = free_arg;
+					break;
+				}
+			}
+
+			if (!check) {
+				V->error(PSI_WARNING, "Unknown variable '%s' of `free` statement"
+						" of implementation '%s'",
+						free_var->name, impl->func->name);
+				return 0;
+			}
+		}
+	}
 
 	impl->decl = decl;
 
@@ -323,10 +471,13 @@ int PSI_ValidatorValidate(PSI_Validator *V)
 	if (V->defs && !validate_typedefs(V)) {
 		return 0;
 	}
+	if (V->structs && !validate_structs(V)) {
+		return 0;
+	}
 	if (V->decls && !validate_decls(V)) {
 		return 0;
 	}
-	if (V->impls && !validate_impls(V)) {
+	if (!V->impls || !validate_impls(V)) {
 		return 0;
 	}
 	return 1;