X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Flibffi.c;h=e30966aeb3a4e635eb6fedb3a314a328338cbc17;hp=b0100f61d96b47763700d1cbf0d06609119899b4;hb=b029005e56a8913fbb3d17ab497b4a37a00a211c;hpb=7474b3e23338ea03bcd65551959e2582cc455f9f diff --git a/src/libffi.c b/src/libffi.c index b0100f6..e30966a 100644 --- a/src/libffi.c +++ b/src/libffi.c @@ -1,6 +1,34 @@ +/******************************************************************************* + Copyright (c) 2016, Michael Wallner . + 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 "context.h" +#include "call.h" #include "php.h" -#include "php_psi.h" -#include "libffi.h" + +#ifdef HAVE_LIBFFI #undef PACKAGE #undef PACKAGE_BUGREPORT @@ -35,7 +63,21 @@ static void *psi_ffi_closure_alloc(size_t s, void **code) } return *code; #else - return NULL; +# error "Neither ffi_closure_alloc() nor mmap() available" +#endif +} + +static ffi_status psi_ffi_prep_closure(ffi_closure **closure, void **code, ffi_cif *sig, void (*handler)(ffi_cif*,void*,void**,void*), void *data) { + *closure = psi_ffi_closure_alloc(sizeof(ffi_closure), code); + assert(*closure != NULL); + +#if PSI_HAVE_FFI_PREP_CLOSURE_LOC + return ffi_prep_closure_loc(*closure, sig, handler, data, *code); + +#elif PSI_HAVE_FFI_PREP_CLOSURE + return ffi_prep_closure(*code, sig, handler, data); +#else +# error "Neither ffi_prep_closure() nor ffi_prep_closure_loc() is available" #endif } @@ -48,54 +90,214 @@ static void psi_ffi_closure_free(void *c) #endif } -static void psi_ffi_handler(ffi_cif *signature, void *_result, void **_args, void *_data); +static void psi_ffi_prep_va(ffi_cif *base, ffi_cif *signature, size_t argc, size_t va_count, + ffi_type **param_types) { + ffi_status rc; -static inline ffi_abi psi_ffi_abi(const char *convention) { - return FFI_DEFAULT_ABI; +#ifdef PSI_HAVE_FFI_PREP_CIF_VAR + rc = ffi_prep_cif_var(signature, base->abi, argc, argc + va_count, + base->rtype, param_types); +#else + /* FIXME: test in config.m4; assume we can just call anyway */ + rc = ffi_prep_cif(signature, base->abi, argc + va_count, base->rtype, param_types); +#endif + + assert(FFI_OK == rc); } -static inline ffi_type *psi_ffi_type(token_t t) { + +#if HAVE_INT128 +static ffi_type *ffi_type_sint128; +static ffi_type *ffi_type_uint128; +#endif + +static inline ffi_type *psi_ffi_decl_arg_type(struct psi_decl_arg *darg); + +static inline ffi_type *psi_ffi_token_type(token_t t) { switch (t) { default: - ZEND_ASSERT(0); + assert(0); /* no break */ case PSI_T_VOID: return &ffi_type_void; - case PSI_T_SINT8: + case PSI_T_INT8: return &ffi_type_sint8; case PSI_T_UINT8: return &ffi_type_uint8; - case PSI_T_SINT16: + case PSI_T_INT16: return &ffi_type_sint16; case PSI_T_UINT16: return &ffi_type_uint16; - case PSI_T_SINT32: + case PSI_T_INT32: return &ffi_type_sint32; case PSI_T_UINT32: return &ffi_type_uint32; - case PSI_T_SINT64: + case PSI_T_INT64: return &ffi_type_sint64; case PSI_T_UINT64: return &ffi_type_uint64; +#if HAVE_INT128 + case PSI_T_INT128: + return ffi_type_sint128; + case PSI_T_UINT128: + return ffi_type_uint128; +#endif case PSI_T_BOOL: return &ffi_type_uchar; - case PSI_T_CHAR: - return &ffi_type_schar; - case PSI_T_SHORT: - return &ffi_type_sshort; - case PSI_T_INT: + case PSI_T_ENUM: return &ffi_type_sint; - case PSI_T_LONG: - return &ffi_type_slong; case PSI_T_FLOAT: return &ffi_type_float; case PSI_T_DOUBLE: return &ffi_type_double; +#ifdef HAVE_LONG_DOUBLE + case PSI_T_LONG_DOUBLE: + return &ffi_type_longdouble; +#endif + case PSI_T_POINTER: + case PSI_T_FUNCTION: + return &ffi_type_pointer; + } +} +static inline ffi_type *psi_ffi_impl_type(token_t impl_type) { + switch (impl_type) { + case PSI_T_BOOL: + return &ffi_type_sint8; + case PSI_T_INT: + return &ffi_type_sint64; + case PSI_T_STRING: + return &ffi_type_pointer; + case PSI_T_FLOAT: + case PSI_T_DOUBLE: + return &ffi_type_double; + EMPTY_SWITCH_DEFAULT_CASE(); + } + return NULL; +} +static void psi_ffi_struct_type_dtor(void *type) { + ffi_type *strct = type; + + if (strct->elements) { + ffi_type **ptr; + + for (ptr = strct->elements; *ptr; ++ptr) { + free(*ptr); + } + free(strct->elements); + } + free(strct); +} + +static size_t psi_ffi_struct_type_pad(ffi_type **els, size_t padding) { + size_t i; + + for (i = 0; i < padding; ++i) { + ffi_type *pad = malloc(sizeof(*pad)); + + memcpy(pad, &ffi_type_schar, sizeof(*pad)); + *els++ = pad; + } + + return padding; +} + +static ffi_type **psi_ffi_struct_type_elements(struct psi_decl_struct *strct) { + size_t i = 0, argc, nels = 0, offset = 0, maxalign = 0, last_arg_pos = -1; + ffi_type **tmp, **els; + struct psi_decl_arg *darg; + + argc = psi_plist_count(strct->args); + els = calloc(argc + 1, sizeof(*els)); + + while (psi_plist_get(strct->args, i++, &darg)) { + ffi_type *type; + size_t padding; + + if (darg->layout->pos == last_arg_pos) { + /* skip bit fields */ + continue; + } + last_arg_pos = darg->layout->pos; + + type = malloc(sizeof(*type)); + *type = *psi_ffi_decl_arg_type(darg); + + if (type->alignment > maxalign) { + maxalign = type->alignment; + } + + assert(type->size <= darg->layout->len); + if ((padding = psi_offset_padding(darg->layout->pos - offset, type->alignment))) { + if (nels + padding + 1 > argc) { + argc += padding; + tmp = realloc(els, (argc + 1) * sizeof(*els)); + if (tmp) { + els = tmp; + } else { + free(els); + return NULL; + } + els[argc] = NULL; + } + psi_ffi_struct_type_pad(&els[nels], padding); + nels += padding; + offset += padding; + } + assert(offset == darg->layout->pos); + + offset = (offset + darg->layout->len + type->alignment - 1) & ~(type->alignment - 1); + els[nels++] = type; + } + + /* apply struct alignment padding */ + offset = (offset + maxalign - 1) & ~(maxalign - 1); + + assert(offset <= strct->size); + if (offset < strct->size) { + size_t padding = strct->size - offset; + + tmp = realloc(els, (padding + argc + 1) * sizeof(*els)); + if (tmp) { + els = tmp; + } else { + free(els); + return NULL; + } + psi_ffi_struct_type_pad(&els[nels], padding); + els[argc + padding] = NULL; } + + return els; } -static inline ffi_type *psi_ffi_decl_type(decl_type *type) { - return psi_ffi_type(real_decl_type(type)->type); +static inline ffi_type *psi_ffi_decl_type(struct psi_decl_type *type) { + struct psi_decl_type *real = psi_decl_type_get_real(type); + + switch (real->type) { + case PSI_T_STRUCT: + if (!real->real.strct->engine.type) { + ffi_type *strct = calloc(1, sizeof(ffi_type)); + + strct->type = FFI_TYPE_STRUCT; + strct->size = 0; + strct->elements = psi_ffi_struct_type_elements(real->real.strct); + + real->real.strct->engine.type = strct; + real->real.strct->engine.dtor = psi_ffi_struct_type_dtor; + } + + return real->real.strct->engine.type; + + case PSI_T_UNION: + { + struct psi_decl_arg *arg; + psi_plist_get(real->real.unn->args, 0, &arg); + return psi_ffi_decl_arg_type(arg); + } + + default: + return psi_ffi_token_type(real->type); + } } -static inline ffi_type *psi_ffi_decl_arg_type(decl_arg *darg) { +static inline ffi_type *psi_ffi_decl_arg_type(struct psi_decl_arg *darg) { if (darg->var->pointer_level) { return &ffi_type_pointer; } else { @@ -103,76 +305,307 @@ static inline ffi_type *psi_ffi_decl_arg_type(decl_arg *darg) { } } -typedef struct PSI_LibffiContext { +static inline ffi_abi psi_ffi_abi(const char *convention) { + if (FFI_LAST_ABI - 2 != FFI_FIRST_ABI) { +#ifdef HAVE_FFI_STDCALL + if (!strcasecmp(convention, "stdcall")) { + return FFI_STDCALL; + } +#endif +#ifdef HAVE_FFI_FASTCALL + if (!strcasecmp(convention, "fastcall")) { + return FFI_FASTCALL; + } +#endif + } + return FFI_DEFAULT_ABI; +} + +struct psi_ffi_context { ffi_cif signature; ffi_type *params[2]; - struct { - struct PSI_LibffiData **list; - size_t count; - } data; -} PSI_LibffiContext; - -typedef struct PSI_LibffiData { - PSI_LibffiContext *context; - impl *impl; - zend_internal_arg_info *arginfo; +}; + +struct psi_ffi_impl_info { + struct psi_context *context; + struct psi_call_frame *frame; + void *code; ffi_closure *closure; +}; + +struct psi_ffi_callback_info { + struct psi_ffi_impl_info *impl_info; + struct psi_let_exp *let_exp; + + void *code; + ffi_closure *closure; +}; + +struct psi_ffi_decl_info { ffi_cif signature; ffi_type *params[1]; -} PSI_LibffiData; +}; + +static inline struct psi_ffi_decl_info *psi_ffi_decl_init(struct psi_decl *decl) { + if (!decl->info) { + int rc; + size_t i, c = psi_plist_count(decl->args); + struct psi_decl_arg *arg; + struct psi_ffi_decl_info *info = calloc(1, sizeof(*info) + 2 * c * sizeof(void *)); + + for (i = 0; psi_plist_get(decl->args, i, &arg); ++i) { + info->params[i] = psi_ffi_decl_arg_type(arg); + } + info->params[c] = NULL; + + rc = ffi_prep_cif(&info->signature, psi_ffi_abi(decl->abi->convention), + c, psi_ffi_decl_arg_type(decl->func), info->params); + + if (FFI_OK != rc) { + free(info); + } else { + decl->info = info; + } + } + + return decl->info; +} + +static inline void psi_ffi_decl_dtor(struct psi_decl *decl) { + if (decl->info) { + free(decl->info); + decl->info = NULL; + } +} + +static void psi_ffi_handler(ffi_cif *sig, void *result, void **args, void *data) +{ + struct psi_impl *impl = data; + struct psi_ffi_impl_info *info = impl->info; + + psi_context_call(info->context, *(zend_execute_data **)args[0], *(zval **)args[1], impl); +} + +static void psi_ffi_callback(ffi_cif *sig, void *result, void **args, void *data) +{ + struct psi_ffi_callback_info *cb_info = data; + struct psi_call_frame_callback cb_data; + + assert(cb_info->impl_info->frame); + + cb_data.cb = cb_info->let_exp; + cb_data.argc = sig->nargs; + cb_data.argv = args; + cb_data.rval = result; + + psi_call_frame_do_callback(cb_info->impl_info->frame, &cb_data); +} -static inline PSI_LibffiData *PSI_LibffiDataAlloc(PSI_LibffiContext *context, impl *impl) { +static inline void psi_ffi_callback_init(struct psi_ffi_impl_info *impl_info, + struct psi_let_exp *let_exp) { + struct psi_ffi_callback_info *cb_info; + struct psi_ffi_decl_info *decl_info; + struct psi_let_callback *cb; + struct psi_let_func *fn = NULL; ffi_status rc; - size_t i, c = impl->decl->args ? impl->decl->args->count : 0; - PSI_LibffiData *data = malloc(sizeof(*data) + c * sizeof(ffi_type *)); - - data->context = context; - data->impl = impl; - data->arginfo = psi_internal_arginfo(impl); - for (i = 0; i < c; ++i) { - data->params[i] = psi_ffi_decl_arg_type(impl->decl->args->args[i]); - } - data->params[c] = NULL; - - rc = ffi_prep_cif( - &data->signature, - psi_ffi_abi(data->impl->decl->abi->convention), - c, - psi_ffi_decl_arg_type(data->impl->decl->func), - data->params); - ZEND_ASSERT(FFI_OK == rc); - - data->closure = psi_ffi_closure_alloc(sizeof(ffi_closure), &data->code); - ZEND_ASSERT(data->closure != NULL); -#if PSI_HAVE_FFI_PREP_CLOSURE_LOC - rc = ffi_prep_closure_loc( - data->closure, - &context->signature, - psi_ffi_handler, - data, - data->code); - ZEND_ASSERT(FFI_OK == rc); -#elif PSI_HAVE_FFI_PREP_CLOSURE - rc = ffi_prep_closure(data->code, &context->signature, psi_ffi_handler, data); - ZEND_ASSERT(FFI_OK == rc); -#else -# error "Neither ffi_prep_closure() nor ffi_prep_closure_loc() available" -#endif - context->data.list = realloc(context->data.list, ++context->data.count * sizeof(*context->data.list)); - context->data.list[context->data.count-1] = data; + switch (let_exp->kind) { + case PSI_LET_CALLBACK: + cb = let_exp->data.callback; + if (cb->decl->info) { + decl_info = cb->decl->info; + } else { + decl_info = psi_ffi_decl_init(cb->decl); + } + + cb_info = calloc(1, sizeof(*cb_info)); + cb_info->impl_info = impl_info; + cb_info->let_exp = let_exp; + rc = psi_ffi_prep_closure(&cb_info->closure, &cb_info->code, + &decl_info->signature, psi_ffi_callback, cb_info); + + if (FFI_OK != rc) { + free(cb_info); + break; + } + cb->info = cb_info; + + assert(!cb->decl->sym); + cb->decl->sym = cb_info->code; + fn = cb->func; + /* no break */ + + case PSI_LET_FUNC: + if (!fn) { + fn = let_exp->data.func; + } + if (fn->inner) { + size_t i = 0; + struct psi_let_exp *inner_let; + + while (psi_plist_get(fn->inner, i++, &inner_let)) { + psi_ffi_callback_init(impl_info, inner_let); + } + } + break; + default: + break; + } +} + +static inline void psi_ffi_callback_dtor(struct psi_let_exp *let_exp) { + struct psi_let_callback *cb; + struct psi_let_func *fn = NULL; + + switch (let_exp->kind) { + case PSI_LET_CALLBACK: + cb = let_exp->data.callback; + + psi_ffi_decl_dtor(cb->decl); + + if (cb->info) { + struct psi_ffi_callback_info *info = cb->info; + + if (info->closure) { + psi_ffi_closure_free(info->closure); + } + free(info); + cb->info = NULL; + } + fn = cb->func; + /* no break */ + case PSI_LET_FUNC: + if (!fn) { + fn = let_exp->data.func; + } + + if (fn->inner) { + size_t i = 0; + struct psi_let_exp *cb; + + while (psi_plist_get(fn->inner, i++, &cb)) { + psi_ffi_callback_dtor(cb); + } + } + break; + default: + break; + } +} + +static inline struct psi_ffi_impl_info *psi_ffi_impl_init(struct psi_impl *impl, + struct psi_context *C) { + struct psi_ffi_context *context = C->context; + struct psi_ffi_impl_info *info = calloc(1, sizeof(*info)); + struct psi_let_stmt *let; + ffi_status rc; + size_t l = 0; + + info->context = C; + + rc = psi_ffi_prep_closure(&info->closure, &info->code, + &context->signature, psi_ffi_handler, impl); + + if (FFI_OK != rc) { + free(info); + return NULL; + } + + while (psi_plist_get(impl->stmts.let, l++, &let)) { + psi_ffi_callback_init(info, let->exp); + } + + return impl->info = info; +} + +static inline void psi_ffi_impl_dtor(struct psi_impl *impl) { + struct psi_ffi_impl_info *info = impl->info; + struct psi_let_stmt *let; + size_t j = 0; + + while (psi_plist_get(impl->stmts.let, j++, &let)) { + psi_ffi_callback_dtor(let->exp); + } + + if (info) { + if (info->closure) { + psi_ffi_closure_free(info->closure); + } + free(info); + impl->info = NULL; + } +} + +static void psi_ffi_extvar_get(ffi_cif *sig, void *result, void **args, void *data) { + struct psi_decl_extvar *evar = data; - return data; + psi_decl_extvar_get(evar, result); } -static inline void PSI_LibffiDataFree(PSI_LibffiData *data) { - psi_ffi_closure_free(data->closure); - free(data->arginfo); - free(data); +static void psi_ffi_extvar_set(ffi_cif *sig, void *result, void **args, void *data) { + struct psi_decl_extvar *evar = data; + + psi_decl_extvar_set(evar, args[0]); } -static inline PSI_LibffiContext *PSI_LibffiContextInit(PSI_LibffiContext *L) { +struct psi_ffi_extvar_info { + struct { + ffi_cif signature; + void *code; + ffi_closure *closure; + } get; + struct { + ffi_cif signature; + ffi_type *params[1]; + void *code; + ffi_closure *closure; + } set; +}; + +static inline ffi_status psi_ffi_extvar_init(struct psi_decl_extvar *evar, + ffi_type *type) { + struct psi_ffi_extvar_info *info = calloc(1, sizeof(*info)); + ffi_status rc; + + rc = ffi_prep_cif(&info->get.signature, FFI_DEFAULT_ABI, 0, + type, NULL); + if (FFI_OK != rc) { + return rc; + } + rc = psi_ffi_prep_closure(&info->get.closure, &info->get.code, + &info->get.signature, psi_ffi_extvar_get, evar); + if (FFI_OK != rc) { + return rc; + } + + info->set.params[0] = type; + rc = ffi_prep_cif(&info->set.signature, FFI_DEFAULT_ABI, 1, + &ffi_type_void, info->set.params); + if (FFI_OK != rc) { + return rc; + } + rc = psi_ffi_prep_closure(&info->set.closure, &info->set.code, + &info->set.signature, psi_ffi_extvar_set, evar); + if (FFI_OK != rc) { + return rc; + } + + evar->info = info; + evar->getter->sym = info->get.code; + evar->setter->sym = info->set.code; + + return FFI_OK; +} + +static inline void psi_ffi_extvar_dtor(struct psi_decl_extvar *evar) { + if (evar->info) { + free(evar->info); + evar->info = NULL; + } +} + +static inline struct psi_ffi_context *psi_ffi_context_init(struct psi_ffi_context *L) { ffi_status rc; if (!L) { @@ -183,118 +616,228 @@ static inline PSI_LibffiContext *PSI_LibffiContextInit(PSI_LibffiContext *L) { L->params[0] = &ffi_type_pointer; L->params[1] = &ffi_type_pointer; rc = ffi_prep_cif(&L->signature, FFI_DEFAULT_ABI, 2, &ffi_type_void, L->params); - ZEND_ASSERT(rc == FFI_OK); + assert(rc == FFI_OK); return L; } -static inline void PSI_LibffiContextDtor(PSI_LibffiContext *L) { - size_t i; - - for (i = 0; i < L->data.count; ++i) { - PSI_LibffiDataFree(L->data.list[i]); - } - if (L->data.list) { - free(L->data.list); - } -} - -static inline void PSI_LibffiContextFree(PSI_LibffiContext **L) { +static inline void psi_ffi_context_free(struct psi_ffi_context **L) { if (*L) { - PSI_LibffiContextDtor(*L); free(*L); *L = NULL; } } -static void psi_ffi_handler(ffi_cif *_sig, void *_result, void **_args, void *_data) +static void psi_ffi_init(struct psi_context *C) { - PSI_LibffiData *data = _data; - size_t i; - void **arg_prm = NULL; - impl_val ret_val; + C->context = psi_ffi_context_init(NULL); +} - if (SUCCESS != psi_parse_args(*(zend_execute_data **)_args[0], data->impl)) { - return; - } +static void psi_ffi_dtor(struct psi_context *C) +{ + if (C->decls) { + size_t i = 0; + struct psi_decl *decl; - if (data->impl->decl->args) { - arg_prm = malloc(data->impl->decl->args->count * sizeof(*arg_prm)); + while (psi_plist_get(C->decls, i++, &decl)) { + psi_ffi_decl_dtor(decl); + } - for (i = 0; i < data->impl->decl->args->count; ++i) { - decl_arg *darg = data->impl->decl->args->args[i]; + } + if (C->vars) { + size_t i = 0; + struct psi_decl_extvar *evar; - arg_prm[i] = psi_do_let(darg); + while (psi_plist_get(C->vars, i++, &evar)) { + psi_ffi_extvar_dtor(evar); } } + if (C->impls) { + size_t i = 0; + struct psi_impl *impl; + + while (psi_plist_get(C->impls, i++, &impl)) { + psi_ffi_impl_dtor(impl); + } + } + psi_ffi_context_free((void *) &C->context); +} - ffi_call(&data->signature, FFI_FN(data->impl->decl->dlptr), &ret_val, arg_prm); - psi_do_return(data->impl, &ret_val, *(zval **)_args[1]); +static zend_function_entry *psi_ffi_compile(struct psi_context *C) +{ + size_t i = 0, d = 0, v = 0, nf = 0; + struct psi_impl *impl; + struct psi_decl *decl; + struct psi_decl_extvar *evar; + zend_function_entry *zfe = NULL; - for (i = 0; i < data->impl->stmts->set.count; ++i) { - set_stmt *set = data->impl->stmts->set.list[i]; + while (psi_plist_get(C->vars, v++, &evar)) { + ffi_type *type = psi_ffi_decl_arg_type(evar->arg); - if (set->arg->_zv) { - psi_do_set(set->arg->_zv, set->val->func, set->val->vars); + if (FFI_OK == psi_ffi_extvar_init(evar, type)) { + /* */ } } - for (i = 0; i < data->impl->stmts->fre.count; ++i) { - free_stmt *fre = data->impl->stmts->fre.list[i]; - - psi_do_free(fre); + if (C->impls) { + zfe = calloc(psi_plist_count(C->impls) + 1, sizeof(*zfe)); + + while (psi_plist_get(C->impls, i++, &impl)) { + zend_function_entry *zf = &zfe[nf]; + + if (!impl->decl) { + continue; + } + if (!psi_ffi_decl_init(impl->decl)) { + continue; + } + if (!psi_ffi_impl_init(impl, C)) { + continue; + } + + zf->fname = impl->func->name + (impl->func->name[0] == '\\'); + zf->handler = ((struct psi_ffi_impl_info *) impl->info)->code; + zf->num_args = psi_plist_count(impl->func->args); + zf->arg_info = psi_internal_arginfo(impl); + ++nf; + } } - psi_do_clean(data->impl); + while (psi_plist_get(C->decls, d++, &decl)) { + if (decl->info) { + continue; + } - if (arg_prm) { - free(arg_prm); + psi_ffi_decl_init(decl); } -} -static void psi_ffi_init(PSI_Context *C) -{ - C->context = PSI_LibffiContextInit(NULL); + return zfe; } -static void psi_ffi_dtor(PSI_Context *C) -{ - PSI_LibffiContextFree((void *) &C->context); +static inline void psi_ffi_call_ex(struct psi_call_frame *frame) { + struct psi_decl *decl = psi_call_frame_get_decl(frame); + struct psi_impl *impl = psi_call_frame_get_impl(frame); + struct psi_ffi_decl_info *decl_info = decl->info; + struct psi_ffi_impl_info *impl_info; + struct psi_call_frame *prev; + + if (impl) { + impl_info = impl->info; + prev = impl_info->frame; + impl_info->frame = frame; + } + ffi_call(&decl_info->signature, FFI_FN(decl->sym), + psi_call_frame_get_rpointer(frame), + psi_call_frame_get_arg_pointers(frame)); + if (impl) { + impl_info->frame = prev; + } } -static zend_function_entry *psi_ffi_compile(PSI_Context *C, PSI_Data *D) -{ - size_t i, j = 0; - zend_function_entry *zfe = calloc(D->impls->count + 1, sizeof(*zfe)); - PSI_LibffiContext *ctx = C->context; +static inline void psi_ffi_call_va(struct psi_call_frame *frame) { + ffi_cif signature; + struct psi_call_frame *prev; + struct psi_decl *decl = psi_call_frame_get_decl(frame); + struct psi_impl *impl = psi_call_frame_get_impl(frame); + struct psi_ffi_decl_info *decl_info = decl->info; + struct psi_ffi_impl_info *impl_info; + size_t i, va_count, argc; + ffi_type **param_types; + + argc = psi_plist_count(decl->args); + va_count = psi_call_frame_num_var_args(frame); + param_types = ecalloc(argc + va_count + 1, sizeof(ffi_type *)); + memcpy(param_types, decl_info->params, argc * sizeof(ffi_type *)); + for (i = 0; i < va_count; ++i) { + struct psi_call_frame_argument *frame_arg; + + frame_arg = psi_call_frame_get_var_argument(frame, i); + param_types[argc + i] = psi_ffi_impl_type(frame_arg->va_type); + } - for (i = 0; i < D->impls->count; ++i) { - zend_function_entry *zf = &zfe[j]; - PSI_LibffiData *data; + psi_ffi_prep_va(&decl_info->signature, &signature, argc, va_count, param_types); - if (!D->impls->list[i]->decl) { - continue; - } + if (impl) { + impl_info = impl->info; + prev = impl_info->frame; + impl_info->frame = frame; + } + ffi_call(&signature, FFI_FN(decl->sym), + psi_call_frame_get_rpointer(frame), + psi_call_frame_get_arg_pointers(frame)); + if (impl) { + impl_info->frame = prev; + } + + efree(param_types); +} - data = PSI_LibffiDataAlloc(ctx, D->impls->list[i]); - zf->fname = D->impls->list[i]->func->name + (D->impls->list[i]->func->name[0] == '\\'); - zf->num_args = D->impls->list[i]->func->args->count; - zf->handler = data->code; - zf->arg_info = data->arginfo; - ++j; +static void psi_ffi_call(struct psi_call_frame *frame) { + if (psi_call_frame_num_var_args(frame)) { + psi_ffi_call_va(frame); + } else { + psi_ffi_call_ex(frame); } +} - return zfe; +static void *psi_ffi_query(struct psi_context *C, enum psi_context_query q, void *arg) { + switch (q) { + case PSI_CONTEXT_QUERY_SELF: + return "ffi"; + case PSI_CONTEXT_QUERY_TYPE: + return psi_ffi_impl_type(*(token_t *) arg); + } + return NULL; +} + +static ZEND_RESULT_CODE psi_ffi_load() +{ +#if HAVE_INT128 + ffi_type *i128, *u128; + + i128 = calloc(1, 3*sizeof(ffi_type)); + i128->type = FFI_TYPE_STRUCT; + i128->size = 0; + i128->elements = (ffi_type **) (i128 + 1); + i128->elements[0] = &ffi_type_sint64; + i128->elements[1] = &ffi_type_sint64; + + ffi_type_sint128 = i128; + + u128 = calloc(1, 3*sizeof(ffi_type)); + u128->type = FFI_TYPE_STRUCT; + u128->size = 0; + u128->elements = (ffi_type **) (u128 + 1); + u128->elements[0] = &ffi_type_uint64; + u128->elements[1] = &ffi_type_uint64; + + ffi_type_uint128 = u128; +#endif + return SUCCESS; } -static PSI_ContextOps ops = { +static void psi_ffi_free() +{ +#if HAVE_INT128 + free(ffi_type_sint128); + free(ffi_type_uint128); +#endif +} + +static struct psi_context_ops ops = { + psi_ffi_load, + psi_ffi_free, psi_ffi_init, psi_ffi_dtor, psi_ffi_compile, + psi_ffi_call, + psi_ffi_query, }; -PSI_ContextOps *PSI_Libffi(void) +struct psi_context_ops *psi_libffi_ops(void) { return &ops; } + +#endif /* HAVE_LIBFFI */