X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fcontext.c;h=727388bcca095f9e521903ac5810bba2eddbd6bb;hp=987fb9abc0a45a660ba8353d25047a8e75c6feb4;hb=3137da963d298c73ce3bc858874fdfc87fa941b7;hpb=6a8a77b52b636041de54d1022016b4f2aa510c05

diff --git a/src/context.c b/src/context.c
index 987fb9a..727388b 100644
--- a/src/context.c
+++ b/src/context.c
@@ -1,19 +1,116 @@
-#include <sys/param.h>
-#include <dlfcn.h>
-#include <dirent.h>
-#include <fnmatch.h>
-#include <errno.h>
-
 #ifdef HAVE_CONFIG_H
 # include "config.h"
 #endif
 
+#include "php.h"
+
+#include <stdio.h>
+#ifdef HAVE_SYS_TYPES_H
+# include <sys/types.h>
+#endif
+#ifdef HAVE_SYS_STAT_H
+# include <sys/stat.h>
+#endif
+#ifdef STDC_HEADERS
+# include <stdlib.h>
+# include <stddef.h>
+#else
+# ifdef HAVE_STDLIB_H
+#  include <stdlib.h>
+# endif
+#endif
+#ifdef HAVE_STRING_H
+# if !defined STDC_HEADERS && defined HAVE_MEMORY_H
+#  include <memory.h>
+# endif
+# include <string.h>
+#endif
+#ifdef HAVE_STRINGS_H
+# include <strings.h>
+#endif
+#ifdef HAVE_INTTYPES_H
+# include <inttypes.h>
+#endif
+#ifdef HAVE_STDINT_H
+# include <stdint.h>
+#endif
+#ifdef HAVE_UNISTD_H
+# include <unistd.h>
+#endif
+
+#ifdef HAVE_ERRNO_H
+# include <errno.h>
+#endif
+#ifdef HAVE_GLOB_H
+# include <glob.h>
+#endif
+#ifdef HAVE_NETINET_IN_H
+# include <netinet/in.h>
+#endif
+#ifdef HAVE_ARPA_NAMESER_H
+# include <arpa/nameser.h>
+#endif
+#ifdef HAVE_NETDB_H
+# include <netdb.h>
+#endif
+#ifdef HAVE_RESOLV_H
+# include <resolv.h>
+#endif
+#ifdef HAVE_SYS_SELECT_H
+# include <sys/select.h>
+#endif
+#ifdef HAVE_SYS_SOCKET_H
+# include <sys/socket.h>
+#endif
+#ifdef HAVE_SYS_TIME_H
+# include <sys/time.h>
+#endif
+#ifdef HAVE_SYS_TIMES_H
+# include <sys/times.h>
+#endif
+#ifdef HAVE_SYS_UIO_H
+# include <sys/uio.h>
+#endif
+#ifdef HAVE_SYS_UTSNAME_H
+# include <sys/utsname.h>
+#endif
+#ifdef HAVE_TIME_H
+# include <time.h>
+#endif
+#ifdef HAVE_WCHAR_H
+# include <wchar.h>
+#endif
+
+#ifdef HAVE_DIRENT_H
+# include <dirent.h>
+# define NAMLEN(dirent) strlen ((dirent)->d_name)
+#else
+# define dirent direct
+# define NAMLEN(dirent) ((dirent)->d_namlen)
+# ifdef HAVE_SYS_NDIR_H
+#  include <sys/ndir.h>
+# endif
+# ifdef HAVE_SYS_DIR_H
+#  include <sys/dir.h>
+# endif
+# ifdef HAVE_NDIR_H
+#  include <ndir.h>
+# endif
+#endif
+
+#include <sys/param.h>
+#include <dlfcn.h>
+#include <fnmatch.h>
+
 #include "php.h"
 #include "php_scandir.h"
 #include "php_psi.h"
 #include "context.h"
 #include "parser.h"
 
+#include "libjit.h"
+#include "libffi.h"
+
 #define psi_predef_count(of) ((sizeof(psi_predef ##of## s)/sizeof(psi_predef ##of))-1)
 typedef struct psi_predef_type {
 	token_t type_tag;
@@ -124,7 +221,7 @@ static inline int locate_decl_type_struct(decl_structs *structs, decl_type *type
 static inline int validate_decl_type(PSI_Data *data, decl_type *type) {
 	switch (type->type) {
 	case PSI_T_NAME:
-		if (!data->defs || !locate_decl_type_alias(data->defs, type)) {
+		if (!data->defs |!locate_decl_type_alias(data->defs, type)) {
 			return 0;
 		}
 		return validate_decl_type(data, type->real);
@@ -153,8 +250,8 @@ static inline int validate_constant(PSI_Data *data, constant *c) {
 
 static inline int validate_decl_arg(PSI_Data *data, decl_arg *arg) {
 	if (!validate_decl_type(data, arg->type)) {
-		data->error(PSI_WARNING, "Cannot use '%s' as type for '%s'",
-			arg->type->name, arg->var->name);
+		data->error(PSI_WARNING, "Cannot use '%s'(%d) as type for '%s'",
+			arg->type->name, arg->type->type, arg->var->name);
 		return 0;
 	}
 	return 1;
@@ -224,6 +321,8 @@ static inline int validate_decl_abi(PSI_Data *data, decl_abi *abi) {
 }
 static inline int validate_decl_func(PSI_Data *data, void *dl, decl *decl, decl_arg *func)
 {
+	size_t i;
+
 	if (!strcmp(func->var->name, "dlsym")) {
 		data->error(PSI_WARNING, "Cannot dlsym dlsym (sic!)");
 		return 0;
@@ -232,21 +331,19 @@ static inline int validate_decl_func(PSI_Data *data, void *dl, decl *decl, decl_
 	if (!validate_decl_arg(data, func)) {
 		return 0;
 	}
+	for (i = 0; i < psi_predef_func_count(); ++i) {
+		psi_predef_func *pre = &psi_predef_funcs[i];
+
+		if (!strcmp(func->var->name, pre->name)) {
+			decl->call.sym = pre->func;
+			break;
+		}
+	}
+	if (!decl->call.sym) {
 #ifndef RTLD_NEXT
 # define RTLD_NEXT ((void *) -1l)
 #endif
-	decl->call.sym = dlsym(dl ?: RTLD_NEXT, func->var->name);
-	if (!decl->call.sym) {
-		size_t i;
-
-		for (i = 0; i < psi_predef_func_count(); ++i) {
-			psi_predef_func *pre = &psi_predef_funcs[i];
-
-			if (!strcmp(func->var->name, pre->name)) {
-				decl->call.sym = pre->func;
-				break;
-			}
-		}
+		decl->call.sym = dlsym(dl ?: RTLD_NEXT, func->var->name);
 		if (!decl->call.sym) {
 			data->error(PSI_WARNING, "Failed to locate symbol '%s': %s",
 				func->var->name, dlerror());
@@ -273,22 +370,27 @@ static inline int validate_decl(PSI_Data *data, void *dl, decl *decl) {
 	}
 	return 1;
 }
-
-static inline decl_arg *locate_struct_member(decl_struct *s, decl_var *var) {
+static inline decl_arg *locate_decl_var_arg(decl_var *var, decl_args *args) {
 	size_t i;
 
-	ZEND_ASSERT(s);
-	for (i = 0; i < s->args->count; ++i) {
-		decl_arg *darg = s->args->args[i];
+	for (i = 0; i < args->count; ++i) {
+		decl_arg *arg = args->args[i];
 
-		if (!strcmp(var->name, darg->var->name)) {
-			return var->arg = darg;
+		if (!strcmp(var->name, arg->var->name)) {
+			return var->arg = arg;
 		}
 	}
 
 	return NULL;
 }
-static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *ref) {
+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 NULL;
+}
+static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
 	size_t i;
 	decl_type *ref_type = real_decl_type(ref->type);
 	decl_var *set_var = set->vars->vars[0];
@@ -312,36 +414,65 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
 	EMPTY_SWITCH_DEFAULT_CASE();
 	}
 
-	if (strcmp(set_var->name, ref->var->name)) {
-		return 0;
+	for (i = 1; i < set->vars->count; ++i) {
+		if (!locate_decl_var_arg(set->vars->vars[i], ref_list)) {
+			return 0;
+		}
 	}
+	set_var->arg = ref;
 
-	if (set->count && (set->func->type != PSI_T_TO_ARRAY || ref_type->type != PSI_T_STRUCT)) {
-		data->error(E_WARNING, "Inner `set` statement casts only work with to_array() casts on structs");
+	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");
 		return 0;
 	}
-	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);
+
+	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);
+
+			if (sub_ref) {
+				if (!validate_set_value(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);
 
 		if (sub_ref) {
-			if (!validate_set_value(data, set->inner[i], 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)) {
+				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");
+		return 0;
 	}
 
 	return 1;
 }
 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
-	size_t i;
+	if (decls) {
+		size_t i;
 
-	for (i = 0; i < decls->count; ++i) {
-		if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
-			ret->decl = decls->list[i]->func;
-			return decls->list[i];
+		for (i = 0; i < decls->count; ++i) {
+			if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
+				ret->decl = decls->list[i]->func;
+				return decls->list[i];
+			}
 		}
 	}
+
 	return NULL;
 }
 static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
@@ -369,7 +500,7 @@ static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 		return 0;
 	}
 
-	if (!validate_set_value(data, ret->set, ret->decl)) {
+	if (!validate_set_value(data, ret->set, ret->decl, impl->decl->args)) {
 		return 0;
 	}
 
@@ -467,7 +598,7 @@ 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)) {
+					if (!validate_set_value(data, set->val, set_arg, impl->decl->args)) {
 						return 0;
 					}
 					set_var->arg = set_arg;
@@ -486,14 +617,17 @@ static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
 	return 1;
 }
 static inline decl *locate_free_decl(decls *decls, free_call *f) {
-	size_t i;
+	if (decls)  {
+		size_t i;
 
-	for (i = 0; i < decls->count; ++i) {
-		if (!strcmp(decls->list[i]->func->var->name, f->func)) {
-			f->decl = decls->list[i];
-			return decls->list[i];
+		for (i = 0; i < decls->count; ++i) {
+			if (!strcmp(decls->list[i]->func->var->name, f->func)) {
+				f->decl = decls->list[i];
+				return decls->list[i];
+			}
 		}
 	}
+
 	return NULL;
 }
 static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
@@ -739,7 +873,7 @@ static int psi_select_dirent(const struct dirent *entry)
 
 void PSI_ContextBuild(PSI_Context *C, const char *paths)
 {
-	int i, n;
+	int i, n, flags = getenv("PSI_DEBUG") ? PSI_PARSER_DEBUG : 0;
 	char *sep = NULL, *cpy = strdup(paths), *ptr = cpy;
 	struct dirent **entries = NULL;
 
@@ -762,7 +896,7 @@ void PSI_ContextBuild(PSI_Context *C, const char *paths)
 					C->error(PSI_WARNING, "Path to PSI file too long: %s/%s",
 						ptr, entries[i]->d_name);
 				}
-				if (!PSI_ParserInit(&P, psi, C->error, 0)) {
+				if (!PSI_ParserInit(&P, psi, C->error, flags)) {
 					C->error(PSI_WARNING, "Failed to init PSI parser (%s): %s",
 						psi, strerror(errno));
 					continue;
@@ -836,6 +970,68 @@ void PSI_ContextCall(PSI_Context *C, impl_val *ret_val, decl *decl)
 	C->ops->call(C, ret_val, decl);
 }
 
+static inline void dump_decl_type(int fd, decl_type *t) {
+	const char *pre;
+
+	switch (t->type) {
+	case PSI_T_STRUCT:
+		pre = "struct ";
+		break;
+	default:
+		pre = "";
+	}
+	dprintf(fd, "%s%s", pre, t->name);
+}
+static inline void dump_decl_var(int fd, decl_var *v) {
+	dprintf(fd, "%.*s%s", v->pointer_level-!!v->array_size, "**********", v->name);
+	if (v->array_size) {
+		dprintf(fd, "[%u]", v->array_size);
+	}
+}
+static inline void dump_decl_arg(int fd, decl_arg *a) {
+	dump_decl_type(fd, a->type);
+	dprintf(fd, " ");
+	dump_decl_var(fd, a->var);
+}
+void PSI_ContextDump(PSI_Context *C, int fd)
+{
+	size_t i, j;
+
+#ifdef HAVE_LIBJIT
+	if (C->ops == PSI_Libjit()) {
+		dprintf(fd, "#PSI(libjit)\n");
+	}
+#endif
+#ifdef HAVE_LIBFFI
+	if (C->ops == PSI_Libffi()) {
+		dprintf(fd, "#PSI(libffi)\n");
+	}
+#endif
+
+	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);
+	}
+	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);
+		}
+		dprintf(fd, "}\n");
+	}
+
+	dprintf(fd, "\n");
+}
+
 void PSI_ContextDtor(PSI_Context *C)
 {
 	size_t i;
@@ -847,15 +1043,19 @@ void PSI_ContextDtor(PSI_Context *C)
 
 	free_decl_libs(&C->psi.libs);
 
-	for (i = 0; i < C->count; ++i) {
-		PSI_DataDtor(&C->data[i]);
+	if (C->data) {
+		for (i = 0; i < C->count; ++i) {
+			PSI_DataDtor(&C->data[i]);
+		}
+		free(C->data);
 	}
-	free(C->data);
 
-	for (zfe = C->closures; zfe->fname; ++zfe) {
-		free((void *) zfe->arg_info);
+	if (C->closures) {
+		for (zfe = C->closures; zfe->fname; ++zfe) {
+			free((void *) zfe->arg_info);
+		}
+		free(C->closures);
 	}
-	free(C->closures);
 
 	if (C->consts) {
 		if (C->consts->list) {