X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fcontext.c;h=9b5730ed64736fe121866a0f04c2d807f34c233a;hp=49391305879892bfc33399d34ab0a4c0be77f4ec;hb=b52dfcdc26059b167eca9baa1e92b3ba9611c3c7;hpb=7e5e4c6d2b654cfd3737c37e9e1894be43642721

diff --git a/src/context.c b/src/context.c
index 4939130..9b5730e 100644
--- a/src/context.c
+++ b/src/context.c
@@ -1,24 +1,133 @@
-#include <sys/param.h>
-#include <dirent.h>
-#include <fnmatch.h>
-#include <errno.h>
+/*******************************************************************************
+ Copyright (c) 2016, Michael Wallner <mike@php.net>.
+ All rights reserved.
+
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+
+     * Redistributions of source code must retain the above copyright notice,
+       this list of conditions and the following disclaimer.
+     * Redistributions in binary form must reproduce the above copyright
+       notice, this list of conditions and the following disclaimer in the
+       documentation and/or other materials provided with the distribution.
+
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
+ FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+*******************************************************************************/
+
+#include "php_psi_stdinc.h"
 
 #include "php.h"
+
+#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 <fnmatch.h>
+
 #include "php_scandir.h"
-#include "context.h"
+#include "php_psi.h"
+#include "calc.h"
+#include "call.h"
+#include "libjit.h"
+#include "libffi.h"
+
+#include "token.h"
 #include "parser.h"
-#include "validator.h"
 
-PSI_Context *PSI_ContextInit(PSI_Context *C, PSI_ContextOps *ops, PSI_ContextErrorFunc error)
+#include "php_psi_posix.h"
+
+PHP_MINIT_FUNCTION(psi_context)
+{
+	unsigned flags = 0;
+	struct psi_context_ops *ops = NULL;
+
+#ifdef HAVE_LIBJIT
+	if (!strcasecmp(PSI_G(engine), "jit")) {
+		ops = psi_libjit_ops();
+	} else
+#endif
+#ifdef HAVE_LIBFFI
+		ops = psi_libffi_ops();
+#endif
+
+	if (!ops) {
+		php_error(E_WARNING, "No PSI engine found");
+		return FAILURE;
+	}
+
+	PSI_G(ops) = ops;
+	if (ops->load && SUCCESS != ops->load()) {
+		return FAILURE;
+	}
+
+	if (psi_check_env("PSI_DEBUG")) {
+		flags |= PSI_DEBUG;
+	}
+	if (psi_check_env("PSI_SILENT")) {
+		flags |= PSI_SILENT;
+	}
+
+	PSI_G(context) = psi_context_init(NULL, PSI_G(ops), psi_error_wrapper, flags);
+	psi_context_build(PSI_G(context), PSI_G(directory));
+
+	return SUCCESS;
+}
+
+PHP_MSHUTDOWN_FUNCTION(psi_context)
+{
+	if (psi_check_env("PSI_DUMP")) {
+		struct psi_dump dump = {{.hn = stdout}, (psi_dump_cb) fprintf};
+
+		psi_context_dump(&dump, PSI_G(context));
+	}
+
+	psi_context_free(&PSI_G(context));
+
+	if (PSI_G(ops)->free) {
+		PSI_G(ops)->free();
+	}
+
+	return SUCCESS;
+}
+
+struct psi_context *psi_context_init(struct psi_context *C, struct psi_context_ops *ops, psi_error_cb error, unsigned flags)
 {
 	if (!C) {
-		C = malloc(sizeof(*C));
+		C = pemalloc(sizeof(*C), 1);
 	}
 	memset(C, 0, sizeof(*C));
 
-	C->error = error;
+	psi_data_ctor(PSI_DATA(C), error, flags);
 	C->ops = ops;
-	ops->init(C);
+
+	if (ops->init) {
+		ops->init(C);
+	}
+
+	assert(ops->call != NULL);
+	assert(ops->compile != NULL);
 
 	return C;
 }
@@ -26,133 +135,266 @@ PSI_Context *PSI_ContextInit(PSI_Context *C, PSI_ContextOps *ops, PSI_ContextErr
 static int psi_select_dirent(const struct dirent *entry)
 {
 #ifndef FNM_CASEFOLD
-#define FNM_CASEFOLD 0
+# define FNM_CASEFOLD 0
 #endif
 	return 0 == fnmatch("*.psi", entry->d_name, FNM_CASEFOLD);
 }
 
+static bool psi_context_add(struct psi_context *C, struct psi_parser *P)
+{
+	bool valid;
+	struct psi_data *D;
+	struct psi_validate_scope scope = {0};
+
+	C->data = safe_perealloc(C->data, (C->count + 1), sizeof(*C->data), 0, 1);
+	D = psi_data_exchange(&C->data[C->count++], PSI_DATA(P));
+
+	psi_validate_scope_ctor(&scope);
+	scope.cpp = P->preproc;
+	valid = psi_validate(&scope, PSI_DATA(C), D);
+	psi_validate_scope_dtor(&scope);
+
+	return valid;
+}
 
-void PSI_ContextBuild(PSI_Context *C, const char *path)
+void psi_context_build(struct psi_context *C, const char *paths)
 {
 	int i, n;
-	struct dirent **entries = NULL;
+	char *sep = NULL, *cpy = strdup(paths), *ptr = cpy;
+	struct dirent **entries;
+
+	do {
+		sep = strchr(ptr, ':');
+
+		if (sep) {
+			*sep = 0;
+		}
 
-	n = php_scandir(path, &entries, psi_select_dirent, alphasort);
+		entries = NULL;
+		n = php_scandir(ptr, &entries, psi_select_dirent, alphasort);
 
-	if (n < 0) {
-		return;
-	} else for (i = 0; i < n; ++i) {
-		char psi[MAXPATHLEN];
-		PSI_Parser P;
-		PSI_Validator V;
+		if (n > 0) {
+			for (i = 0; i < n; ++i) {
+				char psi[MAXPATHLEN];
+				struct psi_parser P;
+				struct psi_parser_input *I;
 
-		if (MAXPATHLEN <= slprintf(psi, MAXPATHLEN, "%s/%s", path, entries[i]->d_name)) {
-			C->error(PSI_WARNING, "Path to PSI file too long: %s/%s",
-				path, entries[i]->d_name);
+				if (MAXPATHLEN <= slprintf(psi, MAXPATHLEN, "%s/%s", ptr, entries[i]->d_name)) {
+					C->error(PSI_DATA(C), NULL, PSI_WARNING, "Path to PSI file too long: %s/%s",
+						ptr, entries[i]->d_name);
+				}
+				if (!psi_parser_init(&P, C->error, C->flags)) {
+					C->error(PSI_DATA(C), NULL, PSI_WARNING, "Failed to init PSI parser (%s): %s",
+						psi, strerror(errno));
+					continue;
+				}
+				if (!(I = psi_parser_open_file(&P, psi, true))) {
+					C->error(PSI_DATA(C), NULL, PSI_WARNING, "Failed to open PSI file (%s): %s",
+						psi, strerror(errno));
+					continue;
+				}
+				psi_parser_parse(&P, I);
+				psi_context_add(C, &P);
+				psi_parser_dtor(&P);
+				psi_parser_input_free(&I);
+			}
 		}
-		if (!PSI_ParserInit(&P, psi, C->error, 0)) {
-			C->error(PSI_WARNING, "Failed to init PSI parser (%s): %s",
-				psi, strerror(errno));
-			continue;
+
+		if (entries) {
+			for (i = 0; i < n; ++i) {
+				free(entries[i]);
+			}
+			free(entries);
 		}
 
-		while (-1 != PSI_ParserScan(&P)) {
-			PSI_ParserParse(&P, PSI_TokenAlloc(&P));
-		};
-		PSI_ParserParse(&P, NULL);
+		ptr = sep + 1;
+	} while (sep);
 
-		if (!PSI_ValidatorInit(&V, &P)) {
-			C->error(PSI_WARNING, "Failed to init PSI validator");
+
+	if (psi_context_compile(C)) {
+		/* zend_register_functions depends on EG(current_module) pointing into module */
+		EG(current_module) = zend_hash_str_find_ptr(&module_registry, "psi", sizeof("psi") - 1);
+		if (SUCCESS != zend_register_functions(NULL, C->closures, NULL, MODULE_PERSISTENT)) {
+			C->error(PSI_DATA(C), NULL, PSI_WARNING, "Failed to register functions!");
+		}
+		EG(current_module) = NULL;
+	}
+
+	free(cpy);
+}
+
+#include <ctype.h>
+static inline bool prefix_match(zend_string *a, zend_string *b)
+{
+	size_t i;
+
+	for (i = 0; i < a->len && i < b->len; ++i) {
+		if (tolower(a->val[i]) != tolower(b->val[i])) {
+			return false;
+		}
+		if (i && a->val[i] == '_') {
 			break;
 		}
-		PSI_ParserDtor(&P);
+	}
+
+	return true;
+}
+
+zend_function_entry *psi_context_compile(struct psi_context *C)
+{
+	zend_constant zc;
+
+	ZEND_CONSTANT_SET_FLAGS(&zc, CONST_CS|CONST_PERSISTENT, EG(current_module)->module_number);
 
-		if (PSI_ValidatorValidate(&V)) {
-			zend_function_entry *closures;
+	if (C->consts) {
+		size_t i = 0;
+		struct psi_const *c;
 
-			closures = PSI_ContextCompile(C, (PSI_Data *) &V);
-			if (closures && SUCCESS != zend_register_functions(NULL, closures, NULL, MODULE_PERSISTENT)) {
-				C->error(PSI_WARNING, "Failed to register functions!");
+		while (psi_plist_get(C->consts, i++, &c)) {
+
+			if (zend_get_constant(c->name)) {
+				continue;
 			}
+
+			zc.name = zend_string_copy(c->name);
+			psi_impl_def_val_get_zval(c->val, c->type ? c->type->type : PSI_T_MIXED, &zc.value);
+
+			zend_register_constant(&zc);
 		}
-		PSI_ValidatorDtor(&V);
 	}
-	if (entries) {
-		for (i = 0; i < n; ++i) {
-			free(entries[i]);
+
+	if (C->enums) {
+		size_t i = 0;
+		struct psi_decl_enum *e;
+
+		while (psi_plist_get(C->enums, i++, &e)) {
+			size_t j = 0;
+			struct psi_decl_enum_item *item;
+
+			while (psi_plist_get(e->items, j++, &item)) {
+				zend_string *name;
+
+				if (psi_decl_type_is_anon(e->name, "enum")
+						|| prefix_match(e->name, item->name)) {
+					name = strpprintf(0, "psi\\%s", item->name->val);
+				} else {
+
+					name = strpprintf(0, "psi\\%s\\%s", e->name->val, item->name->val);
+				}
+
+				if (zend_get_constant(name)) {
+					zend_string_release(name);
+					continue;
+				}
+
+				zc.name = zend_string_dup(name, 1);
+				ZVAL_LONG(&zc.value, psi_num_exp_get_long(item->num, NULL, NULL));
+				zend_register_constant(&zc);
+				zend_string_release(name);
+			}
 		}
-		free(entries);
 	}
 
+	return C->closures = C->ops->compile(C);
 }
 
-zend_function_entry *PSI_ContextCompile(PSI_Context *C, PSI_Data *D)
+
+ZEND_RESULT_CODE psi_context_call(struct psi_context *C, zend_execute_data *execute_data, zval *return_value, struct psi_impl *impl)
 {
-	size_t i, count = C->count++;
-	zend_function_entry *zfe;
+	struct psi_call_frame *frame;
 
-	if (D->consts) {
-		zend_constant zc;
+	frame = psi_call_frame_init(C, impl->decl, impl);
 
-		zc.flags = CONST_PERSISTENT|CONST_CS;
-		zc.module_number = EG(current_module)->module_number;
+	if (SUCCESS != psi_call_frame_parse_args(frame, execute_data)) {
+		psi_call_frame_free(frame);
 
-		for (i = 0; i < D->consts->count; ++i) {
-			constant *c = D->consts->list[i];
+		return FAILURE;
+	}
 
-			zc.name = zend_string_init(c->name + (c->name[0] == '\\'), strlen(c->name) - (c->name[0] == '\\'), 1);
-			ZVAL_NEW_STR(&zc.value, zend_string_init(c->val->text, strlen(c->val->text), 1));
+	psi_call_frame_enter(frame);
 
-			switch (c->type->type) {
-			case PSI_T_BOOL:
-				convert_to_boolean(&zc.value);
-				break;
-			case PSI_T_INT:
-				convert_to_long(&zc.value);
-				break;
-			case PSI_T_FLOAT:
-				convert_to_double(&zc.value);
-				break;
-			}
-			zend_register_constant(&zc);
-		}
+	if (SUCCESS != psi_call_frame_do_let(frame)) {
+		psi_call_frame_do_return(frame, return_value);
+		psi_call_frame_free(frame);
+
+		return FAILURE;
 	}
 
-	C->data = realloc(C->data, C->count * sizeof(*C->data));
-	PSI_DataExchange(&C->data[count], D);
+	if (SUCCESS != psi_call_frame_do_assert(frame, PSI_ASSERT_PRE)) {
+		psi_call_frame_do_return(frame, return_value);
+		psi_call_frame_free(frame);
 
-	zfe = C->ops->compile(C, &C->data[count]);
+		return FAILURE;
+	}
 
-	C->closures = realloc(C->closures, C->count * sizeof(*C->closures));
-	C->closures[count] = zfe;
+	C->ops->call(frame);
 
-	return zfe;
+	if (SUCCESS != psi_call_frame_do_assert(frame, PSI_ASSERT_POST)) {
+		psi_call_frame_do_return(frame, return_value);
+		psi_call_frame_free(frame);
+
+		return FAILURE;
+	}
+
+	psi_call_frame_do_return(frame, return_value);
+	psi_call_frame_do_set(frame);
+	psi_call_frame_do_free(frame);
+	psi_call_frame_free(frame);
+
+	return SUCCESS;
 }
 
-void PSI_ContextDtor(PSI_Context *C)
+
+void psi_context_dtor(struct psi_context *C)
 {
 	size_t i;
+	zend_function_entry *zfe;
 
-	C->ops->dtor(C);
+	if (C->ops->dtor) {
+		C->ops->dtor(C);
+	}
+
+	psi_data_dtor(PSI_DATA(C));
 
-	for (i = 0; i < C->count; ++i) {
-		PSI_DataDtor(&C->data[i]);
-		if (C->closures[i]){
-			free(C->closures[i]);
+	if (C->data) {
+		for (i = 0; i < C->count; ++i) {
+			psi_data_dtor(&C->data[i]);
 		}
+		free(C->data);
 	}
-	free(C->data);
-	free(C->closures);
 
-	memset(C, 0, sizeof(*C));
+	if (C->closures) {
+		for (zfe = C->closures; zfe->fname; ++zfe) {
+			free((void *) zfe->arg_info);
+		}
+		free(C->closures);
+	}
 }
 
-void PSI_ContextFree(PSI_Context **C)
+void psi_context_free(struct psi_context **C)
 {
 	if (*C) {
-		PSI_ContextDtor(*C);
+		psi_context_dtor(*C);
 		free(*C);
 		*C = NULL;
 	}
 }
 
+void psi_context_dump(struct psi_dump *dump, struct psi_context *C)
+{
+	PSI_DUMP(dump, "// psi.engine=%s\n// %lu files\n",
+			(char *) C->ops->query(C, PSI_CONTEXT_QUERY_SELF, NULL),
+			C->count);
+
+	psi_data_dump(dump, PSI_DATA(C));
+
+#if 0
+	if (C->flags & PSI_DEBUG) {
+		size_t i;
+
+		for (i = 0; i < C->count; ++i) {
+			psi_data_dump(dump, &C->data[i]);
+		}
+	}
+#endif
+}