X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=src%2Flibjit.c;h=f99c0390f5932b54f69808aa4f21c5e4ddbe065d;hb=09735ea6055f959a9cac481268754f07d6f6eb9f;hp=b6022c5f92592cad376e14fd42e84deee6ef2e1d;hpb=5359ad5c181e5772f350fe1cba060cbed3a05b91;p=m6w6%2Fext-psi

diff --git a/src/libjit.c b/src/libjit.c
index b6022c5..f99c039 100644
--- a/src/libjit.c
+++ b/src/libjit.c
@@ -1,5 +1,7 @@
 #ifdef HAVE_CONFIG_H
 # include "config.h"
+#else
+# include "php_config.h"
 #endif
 
 #include "php.h"
@@ -272,6 +274,37 @@ static void psi_jit_init(struct psi_context *C)
 	C->context = psi_jit_context_init(NULL);
 }
 
+static inline void psi_jit_destroy_callbacks(struct psi_context *C, let_val *let_val) {
+	let_callback *cb;
+	let_func *fn = NULL;
+
+	switch (let_val->kind) {
+	case PSI_LET_CALLBACK:
+		cb = let_val->data.callback;
+
+		if (cb->decl && cb->decl->call.info) {
+			psi_jit_call_free(cb->decl->call.info);
+		}
+		fn = cb->func;
+		/* no break */
+	case PSI_LET_FUNC:
+		if (!fn) {
+			fn = let_val->data.func;
+		}
+
+		if (fn->inner) {
+			size_t i;
+
+			for (i = 0; i < fn->inner->count; ++i) {
+				psi_jit_destroy_callbacks(C, fn->inner->vals[i]);
+			}
+		}
+		break;
+	default:
+		break;
+	}
+}
+
 static void psi_jit_dtor(struct psi_context *C)
 {
 	if (C->decls) {
@@ -292,19 +325,46 @@ static void psi_jit_dtor(struct psi_context *C)
 			impl *impl = C->impls->list[i];
 
 			for (j = 0; j < impl->stmts->let.count; ++j) {
-				let_stmt *let = impl->stmts->let.list[j];
+				psi_jit_destroy_callbacks(C, impl->stmts->let.list[j]->val);
+			}
+		}
+	}
+	psi_jit_context_free((void *) &C->context);
+}
 
-				if (let->val && let->val->kind == PSI_LET_CALLBACK) {
-					let_callback *cb = let->val->data.callback;
+static inline void psi_jit_compile_callbacks(struct psi_context *C, let_val *let_val) {
+	struct psi_jit_call *call;
+	let_callback *cb;
+	let_func *fn = NULL;
 
-					if (cb->decl && cb->decl->call.info) {
-						psi_jit_call_free(cb->decl->call.info);
-					}
-				}
+	switch (let_val->kind) {
+	case PSI_LET_CALLBACK:
+		cb = let_val->data.callback;
+		if ((call = psi_jit_call_alloc(C, cb->decl))) {
+			if (!psi_jit_call_init_callback_closure(C, call, cb)) {
+				psi_jit_call_free(call);
+				break;
 			}
+
+			cb->decl->call.sym = call->closure;
 		}
+		fn = cb->func;
+		/* no break */
+	case PSI_LET_FUNC:
+		if (!fn) {
+			fn = let_val->data.func;
+		}
+		if (fn->inner) {
+			size_t i;
+
+			for (i = 0; i < fn->inner->count; ++i) {
+				psi_jit_compile_callbacks(C, fn->inner->vals[i]);
+			}
+		}
+		break;
+	default:
+		break;
 	}
-	psi_jit_context_free((void *) &C->context);
 }
 
 static zend_function_entry *psi_jit_compile(struct psi_context *C)
@@ -343,20 +403,7 @@ static zend_function_entry *psi_jit_compile(struct psi_context *C)
 		++j;
 
 		for (c = 0; c < impl->stmts->let.count; ++c) {
-			let_stmt *let = impl->stmts->let.list[c];
-
-			if (let->val && let->val->kind == PSI_LET_CALLBACK) {
-				let_callback *cb = let->val->data.callback;
-
-				if ((call = psi_jit_call_alloc(C, cb->decl))) {
-					if (!psi_jit_call_init_callback_closure(C, call, cb)) {
-						psi_jit_call_free(call);
-						continue;
-					}
-
-					cb->decl->call.sym = call->closure;
-				}
-			}
+			psi_jit_compile_callbacks(C, impl->stmts->let.list[c]->val);
 		}
 	}