libjit backend callback support

[m6w6/ext-psi] / src / libffi.c

diff --git a/src/libffi.c b/src/libffi.c
index 0e948a85257739e376f10ad28e3f0bb32d48ff1c..19afda09baa803e8edef148f8742cc5265905194 100644 (file)
--- a/src/libffi.c
+++ b/src/libffi.c
@@ -78,44 +78,7 @@ static void psi_ffi_handler(ffi_cif *_sig, void *_result, void **_args, void *_d
 
 static void psi_ffi_callback(ffi_cif *_sig, void *_result, void **_args, void *_data)
 {
-       size_t i;
-       unsigned argc = _sig->nargs;
-       void **argv = _args;
-       let_callback *cb = _data;
-       decl *decl_cb = cb->decl;
-       impl_arg *iarg = cb->func->var->arg;
-       zval return_value, *zargv = calloc(argc, sizeof(*zargv));
-       void *result, *to_free = NULL;
-
-       ZEND_ASSERT(argc == cb->decl->args->count);
-
-       /* prepare args for the userland call */
-       for (i = 0; i < argc; ++i) {
-               cb->decl->args->args[i]->ptr = argv[i];
-       }
-       for (i = 0; i < cb->args->count; ++i) {
-               psi_do_set(&zargv[i], cb->args->vals[i]);
-       }
-       zend_fcall_info_argp(&iarg->val.zend.cb->fci, cb->args->count, zargv);
-
-       /* callback into userland */
-       ZVAL_UNDEF(&return_value);
-       iarg->_zv = &return_value;
-       zend_fcall_info_call(&iarg->val.zend.cb->fci, &iarg->val.zend.cb->fcc, iarg->_zv, NULL);
-
-       /* marshal return value of the userland call */
-       switch (iarg->type->type) {
-       case PSI_T_BOOL:        zend_parse_arg_bool(iarg->_zv, &iarg->val.zend.bval, NULL, 0);          break;
-       case PSI_T_LONG:        zend_parse_arg_long(iarg->_zv, &iarg->val.zend.lval, NULL, 0, 1);       break;
-       case PSI_T_FLOAT:
-       case PSI_T_DOUBLE:      zend_parse_arg_double(iarg->_zv, &iarg->val.dval, NULL, 0);                     break;
-       case PSI_T_STRING:      zend_parse_arg_str(iarg->_zv, &iarg->val.zend.str, 0);                          break;
-       }
-       result = cb->func->handler(_result, decl_cb->func->type, iarg, &to_free);
-
-       if (result != _result) {
-               *(void **)_result = result;
-       }
+       psi_callback(_data, _result, _sig->nargs, _args);
 }
 
 static inline ffi_type *psi_ffi_decl_arg_type(decl_arg *darg);
@@ -307,9 +270,6 @@ static inline ffi_type *psi_ffi_decl_type(decl_type *type) {
        decl_type *real = real_decl_type(type);
 
        switch (real->type) {
-       case PSI_T_FUNCTION:
-               return &ffi_type_pointer;
-
        case PSI_T_STRUCT:
                if (!real->strct->engine.type) {
                        ffi_type *strct = calloc(1, sizeof(ffi_type));
@@ -356,6 +316,13 @@ static inline PSI_LibffiContext *PSI_LibffiContextInit(PSI_LibffiContext *L) {
        return L;
 }
 
+static inline void PSI_LibffiContextFree(PSI_LibffiContext **L) {
+       if (*L) {
+               free(*L);
+               *L = NULL;
+       }
+}
+
 static void psi_ffi_init(PSI_Context *C)
 {
        C->context = PSI_LibffiContextInit(NULL);
@@ -373,8 +340,28 @@ static void psi_ffi_dtor(PSI_Context *C)
                                PSI_LibffiCallFree(decl->call.info);
                        }
                }
+
+       }
+       if (C->impls) {
+               size_t i, j;
+
+               for (i = 0; i < C->impls->count; ++i) {
+                       impl *impl = C->impls->list[i];
+
+                       for (j = 0; j < impl->stmts->let.count; ++j) {
+                               let_stmt *let = impl->stmts->let.list[j];
+
+                               if (let->val && let->val->kind == PSI_LET_CALLBACK) {
+                                       let_callback *cb = let->val->data.callback;
+
+                                       if (cb->decl && cb->decl->call.info) {
+                                               PSI_LibffiCallFree(cb->decl->call.info);
+                                       }
+                               }
+                       }
+               }
        }
-       free(C->context);
+       PSI_LibffiContextFree((void *) &C->context);
 }
 
 static zend_function_entry *psi_ffi_compile(PSI_Context *C)
@@ -396,10 +383,11 @@ static zend_function_entry *psi_ffi_compile(PSI_Context *C)
                        continue;
                }
 
-               call = PSI_LibffiCallAlloc(C, impl->decl);
-               if (FFI_OK != PSI_LibffiCallInitClosure(C, call, impl)) {
-                       PSI_LibffiCallFree(call);
-                       continue;
+               if ((call = PSI_LibffiCallAlloc(C, impl->decl))) {
+                       if (FFI_OK != PSI_LibffiCallInitClosure(C, call, impl)) {
+                               PSI_LibffiCallFree(call);
+                               continue;
+                       }
                }
 
                zf->fname = impl->func->name + (impl->func->name[0] == '\\');
@@ -411,16 +399,17 @@ static zend_function_entry *psi_ffi_compile(PSI_Context *C)
                for (c = 0; c < impl->stmts->let.count; ++c) {
                        let_stmt *let = impl->stmts->let.list[c];
 
-                       if (let->val->kind == PSI_LET_CALLBACK) {
+                       if (let->val && let->val->kind == PSI_LET_CALLBACK) {
                                let_callback *cb = let->val->data.callback;
 
-                               call = PSI_LibffiCallAlloc(C, cb->decl);
-                               if (FFI_OK != PSI_LibffiCallInitCallbackClosure(C, call, cb)) {
-                                       PSI_LibffiCallFree(call);
-                                       continue;
-                               }
+                               if ((call = PSI_LibffiCallAlloc(C, cb->decl))) {
+                                       if (FFI_OK != PSI_LibffiCallInitCallbackClosure(C, call, cb)) {
+                                               PSI_LibffiCallFree(call);
+                                               continue;
+                                       }
 
-                               cb->decl->call.sym = call->code;
+                                       cb->decl->call.sym = call->code;
+                               }
                        }
                }
        }
@@ -428,9 +417,6 @@ static zend_function_entry *psi_ffi_compile(PSI_Context *C)
        for (i = 0; i < C->decls->count; ++i) {
                decl *decl = C->decls->list[i];
 
-//             if (decl->impl) {
-//                     continue;
-//             }
                if (decl->call.info) {
                        continue;
                }