flush
[m6w6/ext-psi] / src / context.c
index ac663afa0eaf5b2bfaee6ec05b9c2a8fd4175b50..ba9fabb9cc8e0c0d5ed4542e9f92d96cf1a33f1e 100644 (file)
 #include "php.h"
 #include "php_scandir.h"
 #include "php_psi.h"
-#include "context.h"
-#include "parser.h"
 
 #include "libjit.h"
 #include "libffi.h"
@@ -150,10 +148,24 @@ static struct psi_predef_const {
 
 PSI_MACROS
 
+int psi_glob(const char *pattern, int flags,
+               int (*errfunc) (const char *epath, int eerrno),
+               glob_t *pglob) {
+       size_t offs = flags & GLOB_DOOFFS ? pglob->gl_offs : 0;
+       int rv = glob(pattern, flags, errfunc, pglob);
+       if (pglob->gl_pathv) {
+               while (offs--) {
+                       pglob->gl_pathv[offs] = NULL;
+               }
+       }
+       return rv;
+}
+
 static struct psi_func_redir {
        const char *name;
        void (*func)(void);
 } psi_func_redirs[] = {
+       {"glob", (void (*)(void)) psi_glob},
        PSI_REDIRS
        {0}
 };
@@ -280,8 +292,10 @@ 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'(%d) as type for '%s'",
-                       arg->type->name, arg->type->type, arg->var->name);
+               data->error(PSI_WARNING, "Cannot use '%s'(%d) as type for decl var '%s'"
+                               " in %s on line %zu at col %zu",
+                       arg->type->name, arg->type->type, arg->var->name,
+                       arg->type->token->file, arg->type->token->line, arg->type->token->col);
                return 0;
        }
        return 1;
@@ -477,11 +491,7 @@ static inline int validate_num_exp(PSI_Data *data, decl_args *dargs, decl_arg *f
                return 0;
        }
 }
-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];
-
+static inline int validate_set_value_handler(set_value *set) {
        switch (set->func->type) {
        case PSI_T_TO_BOOL:
                set->func->handler = psi_to_bool;
@@ -504,15 +514,46 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
        case PSI_T_VOID:
                set->func->handler = psi_to_void;
                break;
-       EMPTY_SWITCH_DEFAULT_CASE();
+       default:
+               return 0;
        }
+       return 1;
+}
+static inline void decl_var_arg_v(decl_args *args, va_list argp) {
+       int argc;
+       decl_arg **argv;
+
+       memset(args, 0, sizeof(*args));
 
-       for (i = 1; i < set->vars->count; ++i) {
-               if (!locate_decl_var_arg(set->vars->vars[i], ref_list, ref)) {
+       while ((argc = va_arg(argp, int))) {
+               argv = va_arg(argp, decl_arg **);
+               while (argc--) {
+                       add_decl_arg(args, *argv++);
+               }
+       }
+}
+static inline int validate_set_value_ex(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
+       size_t i;
+       decl_type *ref_type;
+       decl_var *set_var = set->vars->vars[0];
+
+       if (!validate_set_value_handler(set)) {
+               data->error(PSI_WARNING, "Invalid cast '%s'", set->func->name);
+               return 0;
+       }
+
+       for (i = 0; i < set->vars->count; ++i) {
+               decl_var *svar = set->vars->vars[i];
+               if (!svar->arg && !locate_decl_var_arg(svar, ref_list, NULL)) {
+                       data->error(PSI_WARNING, "Unknown variable '%s'", svar->name);
                        return 0;
                }
        }
-       set_var->arg = ref;
+
+       if (!ref) {
+               ref = set_var->arg;
+       }
+       ref_type = real_decl_type(ref->type);
 
        if (set->count) {
                int is_to_array = (set->func->type == PSI_T_TO_ARRAY);
@@ -538,7 +579,7 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
 
                        set->inner[i]->outer.set = set;
                        if (sub_ref) {
-                               if (!validate_set_value(data, set->inner[i], sub_ref, ref_type->strct->args)) {
+                               if (!validate_set_value_ex(data, set->inner[i], sub_ref, ref_type->strct->args)) {
                                        return 0;
                                }
                        }
@@ -554,7 +595,7 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
                                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)) {
+                       if (!validate_set_value_ex(data, set->inner[0], sub_ref, ref_list)) {
                                return 0;
                        }
                }
@@ -565,6 +606,21 @@ static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *r
 
        return 1;
 }
+static inline int validate_set_value(PSI_Data *data, set_value *set, ...) {
+       va_list argp;
+       decl_args args = {0};
+       int check;
+
+       va_start(argp, set);
+       decl_var_arg_v(&args, argp);
+       va_end(argp);
+
+       check = validate_set_value_ex(data, set, NULL, &args);
+       if (args.args) {
+               free(args.args);
+       }
+       return check;
+}
 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
        if (decls) {
                size_t i;
@@ -604,7 +660,7 @@ static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
                return 0;
        }
 
-       if (!validate_set_value(data, ret->set, ret->decl, impl->decl->args)) {
+       if (!validate_set_value(data, ret->set, 1, &ret->decl, impl->decl->args ? (int) impl->decl->args->count : 0, impl->decl->args ? impl->decl->args->args : NULL, 0)) {
                return 0;
        }
 
@@ -616,6 +672,76 @@ static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
        size_t i, j;
        /* we can have multiple let stmts */
+
+       /* check that we have a decl arg for every let stmt */
+       for (i = 0; i < impl->stmts->let.count; ++i) {
+               let_stmt *let = impl->stmts->let.list[i];
+               decl_var *let_var;
+               int check = 0;
+
+               if (let->val && let->val->kind == PSI_LET_TMP) {
+                       let_var = let->val->data.var;
+               } else {
+                       let_var = let->var;
+               }
+
+               if (!locate_decl_var_arg(let_var, impl->decl->args, impl->decl->func)) {
+                       data->error(PSI_WARNING, "Unknown variable '%s' in `let` statement"
+                                       " of implementation '%s'", let_var->name, impl->func->name);
+                       return 0;
+               }
+
+               switch (let->val->kind) {
+               case PSI_LET_NULL:
+                       break;
+               case PSI_LET_TMP:
+                       /* e.g. let bar = &strval($bar); // decl_arg(char **bar) */
+                       /* e.g. let foo = *bar;  */
+                       let->var->pointer_level = let->val->data.var->pointer_level;
+                       let->var->arg = init_decl_arg(
+                                       init_decl_type(
+                                                       real_decl_type(let->val->data.var->arg->type)->type,
+                                                       real_decl_type(let->val->data.var->arg->type)->name),
+                                       init_decl_var(
+                                                       let->var->name,
+                                                       let->var->pointer_level,
+                                                       let->var->array_size));
+                       break;
+               case PSI_LET_NUMEXP:
+                       if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.num)) {
+                               return 0;
+                       }
+                       break;
+               case PSI_LET_CALLOC:
+                       if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.alloc->nmemb)) {
+                               return 0;
+                       }
+                       if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.alloc->size)) {
+                               return 0;
+                       }
+                       break;
+               case PSI_LET_FUNC:
+                       if (impl->func->args) {
+                               for (j = 0; j < impl->func->args->count; ++j) {
+                                       impl_arg *iarg = impl->func->args->args[j];
+
+                                       if (!strcmp(let->val->data.func->var->name, iarg->var->name)) {
+                                               let->val->data.func->arg = iarg;
+                                               check = 1;
+                                               break;
+                                       }
+                               }
+                       }
+                       if (!check) {
+                               data->error(PSI_WARNING, "Unknown value '$%s' of `let` statement"
+                                               " for variable '%s' of implementation '%s'",
+                                               let->val->data.func->var->name, let->var->name, impl->func->name);
+                               return 0;
+                       }
+                       break;
+               }
+       }
+
        /* check that we have a let stmt for every decl arg */
        if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
                decl_arg *darg = impl->decl->args->args[i];
@@ -638,37 +764,7 @@ static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
                        return 0;
                }
        }
-       /* check that the let_value references a known variable or NULL */
-       for (i = 0; i < impl->stmts->let.count; ++i) {
-               let_stmt *let = impl->stmts->let.list[i];
-               int check = 0;
-
-               if (let->val && let->val->func && let->val->func->alloc) {
-                       if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->func->alloc->nmemb)) {
-                               return 0;
-                       }
-                       if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->func->alloc->size)) {
-                               return 0;
-                       }
-               }
-               if (let->val && let->val->var) {
-                       if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
-                               impl_arg *iarg = impl->func->args->args[j];
 
-                               if (!strcmp(let->val->var->name, iarg->var->name)) {
-                                       let->arg = iarg;
-                                       check = 1;
-                                       break;
-                               }
-                       }
-                       if (!check) {
-                               data->error(PSI_WARNING, "Unknown value '$%s' of `let` statement"
-                                               " for variable '%s' of implementation '%s'",
-                                               let->val->var->name, let->var->name, impl->func->name);
-                               return 0;
-                       }
-               }
-       }
        return 1;
 }
 static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
@@ -699,16 +795,35 @@ static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
                        decl_var *set_var = set->val->vars->vars[j];
 
                        check = 0;
-                       if (impl->decl->args) for (k = 0; k < impl->decl->args->count; ++k) {
-                               decl_arg *set_arg = impl->decl->args->args[k];
-
-                               if (!strcmp(set_var->name, set_arg->var->name)) {
-                                       check = 1;
-                                       set_var->arg = set_arg;
-                                       if (!validate_set_value(data, set->val, set_arg, impl->decl->args)) {
-                                               return 0;
+                       if (impl->decl->args) {
+                               for (k = 0; k < impl->decl->args->count; ++k) {
+                                       decl_arg *set_arg = impl->decl->args->args[k];
+
+                                       if (!strcmp(set_var->name, set_arg->var->name)) {
+                                               check = 1;
+                                               set_var->arg = set_arg;
+                                               if (!validate_set_value(data, set->val, 1, &set_arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
+                                                       return 0;
+                                               }
+                                               break;
+                                       }
+                               }
+                       }
+                       if (!check) {
+                               for (k = 0; k < impl->stmts->let.count; ++k) {
+                                       let_stmt *let = impl->stmts->let.list[k];
+
+                                       /* check temp vars */
+                                       if (let->val && let->val->kind == PSI_LET_TMP) {
+                                               if (!strcmp(set_var->name, let->var->name)) {
+                                                       check = 1;
+                                                       set_var->arg = let->var->arg;
+                                                       if (!validate_set_value(data, set->val, 1, &set_var->arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
+                                                               return 0;
+                                                       }
+                                                       break;
+                                               }
                                        }
-                                       break;
                                }
                        }
 
@@ -1101,7 +1216,6 @@ zend_function_entry *PSI_ContextCompile(PSI_Context *C)
 
 void PSI_ContextCall(PSI_Context *C, impl_val *ret_val, decl *decl)
 {
-       errno = 0;
        C->ops->call(C, ret_val, decl);
 }
 
@@ -1131,6 +1245,35 @@ static inline void dump_decl_arg(int fd, decl_arg *a) {
 static inline void dump_level(int fd, unsigned level) {
        dprintf(fd, "%.*s", level, "\t\t\t\t\t\t\t\t\t");
 }
+static inline void dump_num_exp(int fd, num_exp *exp) {
+       while (exp) {
+               switch (exp->t) {
+               case PSI_T_NUMBER:
+                       dprintf(fd, "%s", exp->u.numb);
+                       break;
+               case PSI_T_NSNAME:
+                       dprintf(fd, "%s", exp->u.cnst->name);
+                       break;
+               case PSI_T_NAME:
+                       dump_decl_var(fd, exp->u.dvar);
+                       break;
+               EMPTY_SWITCH_DEFAULT_CASE();
+               }
+               if (exp->operand) {
+                       char op;
+
+                       switch (exp->operator) {
+                       case PSI_T_PLUS:        op = '+'; break;
+                       case PSI_T_MINUS:       op = '-'; break;
+                       case PSI_T_ASTERISK:op = '*'; break;
+                       case PSI_T_SLASH:       op = '/'; break;
+                       EMPTY_SWITCH_DEFAULT_CASE();
+                       }
+                       dprintf(fd, " %c ", op);
+               }
+               exp = exp->operand;
+       }
+}
 static inline void dump_impl_set_value(int fd, set_value *set, unsigned level) {
        size_t i;
 
@@ -1147,6 +1290,10 @@ static inline void dump_impl_set_value(int fd, set_value *set, unsigned level) {
                }
                dump_decl_var(fd, svar);
        }
+       if (set->num) {
+               dprintf(fd, ", ");
+               dump_num_exp(fd, set->num);
+       }
        if (set->inner) {
                dprintf(fd, ",\n");
                for (i = 0; i < set->count; ++i) {
@@ -1161,20 +1308,6 @@ static inline void dump_impl_set_value(int fd, set_value *set, unsigned level) {
                dprintf(fd, ");\n");
        }
 }
-static inline void dump_num_exp(int fd, num_exp *exp) {
-       switch (exp->t) {
-       case PSI_T_NUMBER:
-               dprintf(fd, "%s", exp->u.numb);
-               break;
-       case PSI_T_NAME:
-               dump_decl_var(fd, exp->u.dvar);
-               break;
-       case PSI_T_NSNAME:
-               dprintf(fd, "%s", exp->u.cnst->name);
-               break;
-       EMPTY_SWITCH_DEFAULT_CASE();
-       }
-}
 void PSI_ContextDump(PSI_Context *C, int fd)
 {
        size_t i, j, k, l;
@@ -1238,11 +1371,16 @@ void PSI_ContextDump(PSI_Context *C, int fd)
                        dprintf(fd, "%s ", decl->abi->convention);
                        dump_decl_arg(fd, decl->func);
                        dprintf(fd, "(");
-                       if (decl->args) for (j = 0; j < decl->args->count; ++j) {
-                               if (j) {
-                                       dprintf(fd, ", ");
+                       if (decl->args) {
+                               for (j = 0; j < decl->args->count; ++j) {
+                                       if (j) {
+                                               dprintf(fd, ", ");
+                                       }
+                                       dump_decl_arg(fd, decl->args->args[j]);
+                               }
+                               if (decl->args->varargs) {
+                                       dprintf(fd, ", ...");
                                }
-                               dump_decl_arg(fd, decl->args->args[j]);
                        }
                        dprintf(fd, ");\n");
                }
@@ -1253,16 +1391,26 @@ void PSI_ContextDump(PSI_Context *C, int fd)
                        impl *impl = C->impls->list[i];
 
                        dprintf(fd, "function %s(", impl->func->name);
-                       if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
-                               impl_arg *iarg = impl->func->args->args[j];
-
-                               dprintf(fd, "%s%s %s$%s",
-                                               j ? ", " : "",
-                                               iarg->type->name,
-                                               iarg->var->reference ? "&" : "",
-                                               iarg->var->name);
-                               if (iarg->def) {
-                                       dprintf(fd, " = %s", iarg->def->text);
+                       if (impl->func->args) {
+                               for (j = 0; j < impl->func->args->count; ++j) {
+                                       impl_arg *iarg = impl->func->args->args[j];
+
+                                       dprintf(fd, "%s%s %s$%s",
+                                                       j ? ", " : "",
+                                                       iarg->type->name,
+                                                       iarg->var->reference ? "&" : "",
+                                                       iarg->var->name);
+                                       if (iarg->def) {
+                                               dprintf(fd, " = %s", iarg->def->text);
+                                       }
+                               }
+                               if (impl->func->args->vararg) {
+                                       impl_arg *vararg = impl->func->args->vararg;
+
+                                       dprintf(fd, ", %s %s...$%s",
+                                                       vararg->type->name,
+                                                       vararg->var->reference ? "&" : "",
+                                                       vararg->var->name);
                                }
                        }
                        dprintf(fd, ") : %s%s {\n",
@@ -1274,20 +1422,29 @@ void PSI_ContextDump(PSI_Context *C, int fd)
 
                                        dprintf(fd, "\tlet %s", let->var->name);
                                        if (let->val) {
-                                               dprintf(fd, " = %s", let->val->is_reference ? "&" : "");
-                                               if (let->val->func) {
-                                                       dprintf(fd, "%s(", let->val->func->name);
-                                                       if (let->val->func->alloc) {
-                                                               dump_num_exp(fd, let->val->func->alloc->nmemb);
-                                                               dprintf(fd, ", ");
-                                                               dump_num_exp(fd, let->val->func->alloc->size);
-                                                       } else {
-                                                               dprintf(fd, "$%s", let->val->var->name);
-                                                       }
-                                                       dprintf(fd, ");\n");
-                                               } else {
-                                                       dprintf(fd, "NULL;\n");
+                                               dprintf(fd, " = %s", let->val->flags.one.is_reference ? "&" : "");
+                                               switch (let->val->kind) {
+                                               case PSI_LET_NULL:
+                                                       dprintf(fd, "NULL");
+                                                       break;
+                                               case PSI_LET_TMP:
+                                                       dump_decl_var(fd, let->val->data.var);
+                                                       break;
+                                               case PSI_LET_CALLOC:
+                                                       dprintf(fd, "calloc(");
+                                                       dump_num_exp(fd, let->val->data.alloc->nmemb);
+                                                       dprintf(fd, ", ");
+                                                       dump_num_exp(fd, let->val->data.alloc->size);
+                                                       dprintf(fd, ")");
+                                                       break;
+                                               case PSI_LET_FUNC:
+                                                       dprintf(fd, "%s($%s)", let->val->data.func->name,
+                                                                       let->val->data.func->var->name);
+                                                       break;
+
+                                               EMPTY_SWITCH_DEFAULT_CASE();
                                                }
+                                               dprintf(fd, ";\n");
                                        }
                                }
                                for (j = 0; j < impl->stmts->ret.count; ++j) {