git.m6w6.name Git - m6w6/ext-psi/blob - src/context.c

   1 #ifdef HAVE_CONFIG_H
   2 # include "config.h"
   3 #endif
   4
   5 #include "php.h"
   6
   7 #include <stdio.h>
   8 #ifdef HAVE_SYS_TYPES_H
   9 # include <sys/types.h>
  10 #endif
  11 #ifdef HAVE_SYS_STAT_H
  12 # include <sys/stat.h>
  13 #endif
  14 #ifdef STDC_HEADERS
  15 # include <stdlib.h>
  16 # include <stddef.h>
  17 #else
  18 # ifdef HAVE_STDLIB_H
  19 #  include <stdlib.h>
  20 # endif
  21 #endif
  22 #ifdef HAVE_STRING_H
  23 # if !defined STDC_HEADERS && defined HAVE_MEMORY_H
  24 #  include <memory.h>
  25 # endif
  26 # include <string.h>
  27 #endif
  28 #ifdef HAVE_STRINGS_H
  29 # include <strings.h>
  30 #endif
  31 #ifdef HAVE_INTTYPES_H
  32 # include <inttypes.h>
  33 #endif
  34 #ifdef HAVE_STDINT_H
  35 # include <stdint.h>
  36 #endif
  37 #ifdef HAVE_UNISTD_H
  38 # include <unistd.h>
  39 #endif
  40
  41 #ifdef HAVE_ERRNO_H
  42 # include <errno.h>
  43 #endif
  44 #ifdef HAVE_GLOB_H
  45 # include <glob.h>
  46 #endif
  47 #ifdef HAVE_NETINET_IN_H
  48 # include <netinet/in.h>
  49 #endif
  50 #ifdef HAVE_ARPA_NAMESER_H
  51 # include <arpa/nameser.h>
  52 #endif
  53 #ifdef HAVE_NETDB_H
  54 # include <netdb.h>
  55 #endif
  56 #ifdef HAVE_RESOLV_H
  57 # include <resolv.h>
  58 #endif
  59 #ifdef HAVE_SYS_SELECT_H
  60 # include <sys/select.h>
  61 #endif
  62 #ifdef HAVE_SYS_SOCKET_H
  63 # include <sys/socket.h>
  64 #endif
  65 #ifdef HAVE_SYS_TIME_H
  66 # include <sys/time.h>
  67 #endif
  68 #ifdef HAVE_SYS_TIMES_H
  69 # include <sys/times.h>
  70 #endif
  71 #ifdef HAVE_SYS_UIO_H
  72 # include <sys/uio.h>
  73 #endif
  74 #ifdef HAVE_SYS_UTSNAME_H
  75 # include <sys/utsname.h>
  76 #endif
  77 #ifdef HAVE_TIME_H
  78 # include <time.h>
  79 #endif
  80 #ifdef HAVE_WCHAR_H
  81 # include <wchar.h>
  82 #endif
  83
  84 #ifdef HAVE_DIRENT_H
  85 # include <dirent.h>
  86 # define NAMLEN(dirent) strlen ((dirent)->d_name)
  87 #else
  88 # define dirent direct
  89 # define NAMLEN(dirent) ((dirent)->d_namlen)
  90 # ifdef HAVE_SYS_NDIR_H
  91 #  include <sys/ndir.h>
  92 # endif
  93 # ifdef HAVE_SYS_DIR_H
  94 #  include <sys/dir.h>
  95 # endif
  96 # ifdef HAVE_NDIR_H
  97 #  include <ndir.h>
  98 # endif
  99 #endif
 100
 101 #include <sys/param.h>
 102 #include <dlfcn.h>
 103 #include <fnmatch.h>
 104
 105 #include "php.h"
 106 #include "php_scandir.h"
 107 #include "php_psi.h"
 108
 109 #include "libjit.h"
 110 #include "libffi.h"
 111
 112 static struct psi_std_type {
 113         token_t type_tag;
 114         const char *type_name;
 115 } psi_std_types[] = {
 116         {PSI_T_FLOAT, "float"},
 117         {PSI_T_DOUBLE, "double"},
 118         {PSI_T_INT8, "int8_t"},
 119         {PSI_T_INT16, "int16_t"},
 120         {PSI_T_INT32, "int32_t"},
 121         {PSI_T_INT64, "int64_t"},
 122         {PSI_T_UINT8, "uint8_t"},
 123         {PSI_T_UINT16, "uint16_t"},
 124         {PSI_T_UINT32, "uint32_t"},
 125         {PSI_T_UINT64, "uint64_t"},
 126         {0}
 127 };
 128
 129 static struct psi_predef_type {
 130         token_t type_tag;
 131         const char *type_name;
 132         const char *alias;
 133 } psi_predef_types[] = {
 134         PSI_TYPES
 135         {0}
 136 };
 137
 138 static struct psi_predef_const {
 139         token_t type_tag;
 140         const char *type_name;
 141         const char *var_name;
 142         const char *val_text;
 143         token_t val_type_tag;
 144 } psi_predef_consts[] = {
 145         PSI_CONSTS
 146         {0}
 147 };
 148
 149 PSI_MACROS
 150
 151 int psi_glob(const char *pattern, int flags,
 152                int (*errfunc) (const char *epath, int eerrno),
 153                glob_t *pglob) {
 154         size_t offs = flags & GLOB_DOOFFS ? pglob->gl_offs : 0;
 155         int rv = glob(pattern, flags, errfunc, pglob);
 156         if (pglob->gl_pathv) {
 157                 while (offs--) {
 158                         pglob->gl_pathv[offs] = NULL;
 159                 }
 160         }
 161         return rv;
 162 }
 163
 164 static struct psi_func_redir {
 165         const char *name;
 166         void (*func)(void);
 167 } psi_func_redirs[] = {
 168         {"glob", (void (*)(void)) psi_glob},
 169         PSI_REDIRS
 170         {0}
 171 };
 172
 173 static struct psi_predef_decl {
 174         token_t type_tag;
 175         const char *type_name;
 176         const char *var_name;
 177         size_t pointer_level;
 178         size_t array_size;
 179 } psi_predef_decls[] = {
 180         PSI_DECLS
 181         {0}
 182 };
 183 static struct psi_predef_decl psi_predef_vararg_decls[] = {
 184         PSI_VA_DECLS
 185         {0}
 186 };
 187
 188 static struct psi_predef_struct {
 189         token_t type_tag;
 190         const char *type_name;
 191         const char *var_name;
 192         size_t offset;
 193         size_t size;
 194         size_t pointer_level;
 195         size_t array_size;
 196 } psi_predef_structs[] = {
 197         PSI_STRUCTS
 198         {0}
 199 };
 200
 201 static int validate_lib(PSI_Data *data, void **dlopened) {
 202         char lib[MAXPATHLEN];
 203         const char *ptr = data->psi.file.ln;
 204         size_t len;
 205
 206         if (!ptr) {
 207                 /* FIXME: assume stdlib */
 208                 return 1;
 209         } else if (!strchr(ptr, '/')) {
 210                 len = snprintf(lib, MAXPATHLEN, "lib%s.%s", ptr, PHP_PSI_SHLIB_SUFFIX);
 211                 if (MAXPATHLEN == len) {
 212                         data->error(PSI_WARNING, "Library name too long: '%s'", ptr);
 213                 }
 214                 lib[len] = 0;
 215                 ptr = lib;
 216         }
 217         if (!(*dlopened = dlopen(ptr, RTLD_LAZY|RTLD_LOCAL))) {
 218                 data->error(PSI_WARNING, "Could not open library '%s': %s.",
 219                                 data->psi.file.ln, dlerror());
 220                 return 0;
 221         }
 222         return 1;
 223 }
 224
 225 static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
 226         size_t i;
 227         struct psi_std_type *stdtyp;
 228
 229         if (type->real) {
 230                 return 1;
 231         }
 232         for (i = 0; i < defs->count; ++i) {
 233                 decl_typedef *def = defs->list[i];
 234
 235                 if (def->type->type != type->type && !strcmp(def->alias, type->name)) {
 236                         type->real = def->type;
 237                         return 1;
 238                 }
 239         }
 240         for (stdtyp = &psi_std_types[0]; stdtyp->type_tag; ++stdtyp) {
 241                 if (!strcmp(type->name, stdtyp->type_name)) {
 242                         type->type = stdtyp->type_tag;
 243                         return 1;
 244                 }
 245         }
 246
 247         return 0;
 248 }
 249 static inline int locate_decl_type_struct(decl_structs *structs, decl_type *type) {
 250         size_t i;
 251
 252         if (type->strct) {
 253                 return 1;
 254         }
 255         for (i = 0; i < structs->count; ++i) {
 256                 if (!strcmp(structs->list[i]->name, type->name)) {
 257                         type->strct = structs->list[i];
 258                         return 1;
 259                 }
 260         }
 261         return 0;
 262 }
 263
 264 static inline int validate_decl_type(PSI_Data *data, decl_type *type) {
 265         switch (type->type) {
 266         case PSI_T_NAME:
 267                 if (!data->defs || !locate_decl_type_alias(data->defs, type)) {
 268                         return 0;
 269                 }
 270                 if (type->real) {
 271                         return validate_decl_type(data, type->real);
 272                 }
 273                 return 1;
 274         case PSI_T_STRUCT:
 275                 if (!data->structs || !locate_decl_type_struct(data->structs, type)) {
 276                         return 0;
 277                 }
 278                 break;
 279         }
 280         return 1;
 281 }
 282 static inline int validate_decl_typedef(PSI_Data *data, decl_typedef *def) {
 283         if (!validate_decl_type(data, def->type)) {
 284                 data->error(PSI_WARNING, "Type '%s' cannot be aliased to %s'%s'",
 285                         def->type->name, def->type->type == PSI_T_STRUCT?"struct ":"",def->alias);
 286                 return 0;
 287         }
 288         /* FIXME: check def->alias */
 289         return 1;
 290 }
 291
 292 static inline int validate_constant(PSI_Data *data, constant *c) {
 293         /* FIXME */
 294         return 1;
 295 }
 296
 297 static inline int validate_decl_arg(PSI_Data *data, decl_arg *arg) {
 298         if (!validate_decl_type(data, arg->type)) {
 299                 data->error(PSI_WARNING, "Cannot use '%s'(%d) as type for decl var '%s'"
 300                                 " in %s on line %zu at col %zu",
 301                         arg->type->name, arg->type->type, arg->var->name,
 302                         arg->type->token->file, arg->type->token->line, arg->type->token->col);
 303                 return 0;
 304         }
 305         return 1;
 306 }
 307
 308 static inline int validate_decl_struct(PSI_Data *data, decl_struct *s) {
 309         size_t i;
 310
 311         for (i = 0; i < s->args->count; ++i) {
 312                 if (!validate_decl_arg(data, s->args->args[i])) {
 313                         return 0;
 314                 }
 315         }
 316
 317         for (i = 0; i < s->args->count; ++i) {
 318                 decl_arg *darg = s->args->args[i];
 319
 320                 if (!validate_decl_arg(data, darg)) {
 321                         return 0;
 322                 }
 323
 324                 ZEND_ASSERT(!darg->var->arg || darg->var->arg == darg);
 325                 darg->var->arg = darg;
 326
 327                 if (!darg->layout) {
 328                         token_t t;
 329
 330                         if (darg->var->pointer_level && (!darg->var->array_size || darg->var->pointer_level == 1)) {
 331                                 t = PSI_T_POINTER;
 332                         } else {
 333                                 t = real_decl_type(darg->type)->type;
 334                         }
 335
 336                         if (i) {
 337                                 decl_arg *last = s->args->args[i-1];
 338                                 darg->layout = init_decl_struct_layout(
 339                                                 psi_t_align(t, last->layout->pos + last->layout->len),
 340                                                 psi_t_size(t) * darg->var->array_size);
 341                         } else {
 342                                 darg->layout = init_decl_struct_layout(0, psi_t_size(t));
 343                         }
 344                 }
 345                 if (s->size < darg->layout->pos + darg->layout->len) {
 346                         s->size = darg->layout->pos + darg->layout->len;
 347                 }
 348         }
 349         return 1;
 350 }
 351
 352 static const char * const abi_ccs[] = {
 353                 "default", /* \                 */
 354                 "extern",  /*  > - all the same */
 355                 "cdecl",   /* /                 */
 356                 "stdcall",
 357                 "fastcall",
 358 };
 359 static inline int validate_decl_abi(PSI_Data *data, decl_abi *abi) {
 360         size_t i;
 361
 362         for (i = 0; i < sizeof(abi_ccs)/sizeof(char*); ++i) {
 363                 if (strcasecmp(abi->convention, abi_ccs[i])) {
 364                         return 1;
 365                 }
 366         }
 367         data->error(PSI_WARNING, "Invalid calling convention: '%s'", abi->convention);
 368         return 0;
 369 }
 370 static inline int validate_decl_func(PSI_Data *data, void *dl, decl *decl, decl_arg *func)
 371 {
 372         struct psi_func_redir *redir;
 373
 374         if (!strcmp(func->var->name, "dlsym")) {
 375                 data->error(PSI_WARNING, "Cannot dlsym dlsym (sic!)");
 376                 return 0;
 377         }
 378
 379         if (!validate_decl_arg(data, func)) {
 380                 return 0;
 381         }
 382         for (redir = &psi_func_redirs[0]; redir->name; ++redir) {
 383                 if (!strcmp(func->var->name, redir->name)) {
 384                         decl->call.sym = redir->func;
 385                 }
 386         }
 387         if (!decl->call.sym) {
 388 #ifndef RTLD_NEXT
 389 # define RTLD_NEXT ((void *) -1l)
 390 #endif
 391                 decl->call.sym = dlsym(dl ?: RTLD_NEXT, func->var->name);
 392                 if (!decl->call.sym) {
 393                         data->error(PSI_WARNING, "Failed to locate symbol '%s': %s",
 394                                 func->var->name, dlerror());
 395                 }
 396         }
 397         return 1;
 398 }
 399
 400 static inline int validate_decl(PSI_Data *data, void *dl, decl *decl) {
 401         if (!validate_decl_abi(data, decl->abi)) {
 402                 return 0;
 403         }
 404         if (!validate_decl_func(data, dl, decl, decl->func)) {
 405                 return 0;
 406         }
 407         if (decl->args) {
 408                 size_t i;
 409
 410                 for (i = 0; i < decl->args->count; ++i) {
 411                         if (!validate_decl_arg(data, decl->args->args[i])) {
 412                                 return 0;
 413                         }
 414                 }
 415         }
 416         return 1;
 417 }
 418 static inline decl_arg *locate_decl_var_arg(decl_var *var, decl_args *args, decl_arg *func) {
 419         size_t i;
 420
 421         for (i = 0; i < args->count; ++i) {
 422                 decl_arg *arg = args->args[i];
 423
 424                 if (!strcmp(var->name, arg->var->name)) {
 425                         ZEND_ASSERT(!var->arg || var->arg == arg);
 426                         return var->arg = arg;
 427                 }
 428         }
 429         if (func && !strcmp(var->name, func->var->name)) {
 430                 return var->arg = func;
 431         }
 432
 433         return NULL;
 434 }
 435 static inline decl_arg *locate_struct_member(decl_struct *s, decl_var *var) {
 436         if (s->args) {
 437                 return locate_decl_var_arg(var, s->args, NULL);
 438         }
 439
 440         return NULL;
 441 }
 442 static inline constant *locate_num_exp_constant(num_exp *exp, constants *consts) {
 443         size_t i;
 444
 445         for (i = 0; i < consts->count; ++i) {
 446                 constant *cnst = consts->list[i];
 447
 448                 if (!strcmp(cnst->name, exp->u.numb)) {
 449                         free(exp->u.numb);
 450                         return exp->u.cnst = cnst;
 451                 }
 452         }
 453
 454         return NULL;
 455 }
 456 static inline int validate_num_exp(PSI_Data *data, decl_args *dargs, decl_arg *func, num_exp *exp) {
 457         if (exp->operand) {
 458                 switch (exp->operator) {
 459                 case PSI_T_PLUS:
 460                         exp->calculator = psi_calc_add;
 461                         break;
 462                 case PSI_T_MINUS:
 463                         exp->calculator = psi_calc_sub;
 464                         break;
 465                 case PSI_T_ASTERISK:
 466                         exp->calculator = psi_calc_mul;
 467                         break;
 468                 case PSI_T_SLASH:
 469                         exp->calculator = psi_calc_div;
 470                         break;
 471                 EMPTY_SWITCH_DEFAULT_CASE();
 472                 }
 473                 if (!validate_num_exp(data, dargs, func, exp->operand)) {
 474                         return 0;
 475                 }
 476         }
 477         switch (exp->t) {
 478         case PSI_T_NAME:
 479                 if (!locate_decl_var_arg(exp->u.dvar, dargs, func)) {
 480                         data->error(PSI_WARNING, "Unknown variable '%s' in numeric expression",
 481                                         exp->u.dvar->name);
 482                         return 0;
 483                 }
 484                 return 1;
 485         case PSI_T_NSNAME:
 486                 if (!locate_num_exp_constant(exp, data->consts)) {
 487                         data->error(PSI_WARNING, "Unknown constant '%s' in numeric expression",
 488                                         exp->u.numb);
 489                         return 0;
 490                 }
 491                 return 1;
 492         case PSI_T_NUMBER:
 493                 return 1;
 494         default:
 495                 return 0;
 496         }
 497 }
 498 static inline int validate_set_value_handler(set_value *set) {
 499         switch (set->func->type) {
 500         case PSI_T_TO_BOOL:
 501                 set->func->handler = psi_to_bool;
 502                 break;
 503         case PSI_T_TO_INT:
 504                 set->func->handler = psi_to_int;
 505                 break;
 506         case PSI_T_TO_FLOAT:
 507                 set->func->handler = psi_to_double;
 508                 break;
 509         case PSI_T_TO_STRING:
 510                 set->func->handler = psi_to_string;
 511                 break;
 512         case PSI_T_TO_ARRAY:
 513                 set->func->handler = psi_to_array;
 514                 break;
 515         case PSI_T_TO_OBJECT:
 516                 set->func->handler = psi_to_object;
 517                 break;
 518         case PSI_T_VOID:
 519                 set->func->handler = psi_to_void;
 520                 break;
 521         default:
 522                 return 0;
 523         }
 524         return 1;
 525 }
 526 static inline void decl_var_arg_v(decl_args *args, va_list argp) {
 527         int argc;
 528         decl_arg **argv;
 529
 530         memset(args, 0, sizeof(*args));
 531
 532         while ((argc = va_arg(argp, int))) {
 533                 argv = va_arg(argp, decl_arg **);
 534                 while (argc--) {
 535                         add_decl_arg(args, *argv++);
 536                 }
 537         }
 538 }
 539 static inline int validate_set_value_ex(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
 540         size_t i;
 541         decl_type *ref_type;
 542         decl_var *set_var = set->vars->vars[0];
 543
 544         if (!validate_set_value_handler(set)) {
 545                 data->error(PSI_WARNING, "Invalid cast '%s'", set->func->name);
 546                 return 0;
 547         }
 548
 549         for (i = 0; i < set->vars->count; ++i) {
 550                 decl_var *svar = set->vars->vars[i];
 551                 if (!svar->arg && !locate_decl_var_arg(svar, ref_list, NULL)) {
 552                         data->error(PSI_WARNING, "Unknown variable '%s'", svar->name);
 553                         return 0;
 554                 }
 555         }
 556
 557         if (!ref) {
 558                 ref = set_var->arg;
 559         }
 560         ref_type = real_decl_type(ref->type);
 561
 562         if (set->count) {
 563                 int is_to_array = (set->func->type == PSI_T_TO_ARRAY);
 564                 int is_pointer_to_struct = (ref_type->type == PSI_T_STRUCT && ref->var->pointer_level);
 565
 566                 if (!is_to_array && !is_pointer_to_struct) {
 567                         data->error(E_WARNING, "Inner `set` statement casts only work with "
 568                                         "to_array() casts on structs or pointers: %s(%s...", set->func->name, set->vars->vars[0]->name);
 569                         return 0;
 570                 }
 571         }
 572         if (set->num) {
 573                 if (!validate_num_exp(data, ref_list, ref, set->num)) {
 574                         return 0;
 575                 }
 576         }
 577
 578         if (ref_type->type == PSI_T_STRUCT) {
 579                 /* to_array(struct, to_...) */
 580                 for (i = 0; i < set->count; ++i) {
 581                         decl_var *sub_var = set->inner[i]->vars->vars[0];
 582                         decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
 583
 584                         set->inner[i]->outer.set = set;
 585                         if (sub_ref) {
 586                                 if (!validate_set_value_ex(data, set->inner[i], sub_ref, ref_type->strct->args)) {
 587                                         return 0;
 588                                 }
 589                         }
 590                 }
 591         } else if (set->count == 1) {
 592                 /* to_array(ptr, to_string(*ptr)) */
 593                 decl_var *sub_var = set->inner[0]->vars->vars[0];
 594                 decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list, ref);
 595
 596                 set->inner[0]->outer.set = set;
 597                 if (sub_ref) {
 598                         if (strcmp(sub_var->name, set_var->name)) {
 599                                 data->error(E_WARNING, "Inner `set` statement casts on pointers must reference the same variable");
 600                                 return 0;
 601                         }
 602                         if (!validate_set_value_ex(data, set->inner[0], sub_ref, ref_list)) {
 603                                 return 0;
 604                         }
 605                 }
 606         } else if (set->count > 1) {
 607                 data->error(E_WARNING, "Inner `set` statement casts on pointers may only occur once");
 608                 return 0;
 609         }
 610
 611         return 1;
 612 }
 613 static inline int validate_set_value(PSI_Data *data, set_value *set, ...) {
 614         va_list argp;
 615         decl_args args = {0};
 616         int check;
 617
 618         va_start(argp, set);
 619         decl_var_arg_v(&args, argp);
 620         va_end(argp);
 621
 622         check = validate_set_value_ex(data, set, NULL, &args);
 623         if (args.args) {
 624                 free(args.args);
 625         }
 626         return check;
 627 }
 628 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
 629         if (decls) {
 630                 size_t i;
 631
 632                 for (i = 0; i < decls->count; ++i) {
 633                         if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
 634                                 ret->decl = decls->list[i]->func;
 635                                 return decls->list[i];
 636                         }
 637                 }
 638         }
 639
 640         return NULL;
 641 }
 642 static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 643         return_stmt *ret;
 644
 645         /* we must have exactly one ret stmt delcaring the native func to call */
 646         /* and which type cast to apply */
 647         if (impl->stmts->ret.count != 1) {
 648                 if (impl->stmts->ret.count > 1) {
 649                         data->error(PSI_WARNING, "Too many `return` statements for implmentation %s;"
 650                                         " found %zu, exactly one is needed",
 651                                         impl->func->name, impl->stmts->ret.count);
 652                 } else {
 653                         data->error(PSI_WARNING, "Missing `return` statement for implementation %s",
 654                                         impl->func->name);
 655                 }
 656                 return 0;
 657         }
 658
 659         ret = impl->stmts->ret.list[0];
 660
 661         if (!(impl->decl = locate_impl_decl(data->decls, ret))) {
 662                 data->error(PSI_WARNING, "Missing declaration for implementation %s",
 663                                 impl->func->name);
 664                 return 0;
 665         }
 666
 667         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)) {
 668                 return 0;
 669         }
 670
 671         impl->decl->impl = impl;
 672
 673         return 1;
 674 }
 675
 676 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
 677         size_t i, j;
 678         /* we can have multiple let stmts */
 679
 680         /* check that we have a decl arg for every let stmt */
 681         for (i = 0; i < impl->stmts->let.count; ++i) {
 682                 let_stmt *let = impl->stmts->let.list[i];
 683                 decl_var *let_var;
 684                 int check = 0;
 685
 686                 if (let->val && let->val->kind == PSI_LET_TMP) {
 687                         let_var = let->val->data.var;
 688                 } else {
 689                         let_var = let->var;
 690                 }
 691
 692                 if (!locate_decl_var_arg(let_var, impl->decl->args, impl->decl->func)) {
 693                         data->error(PSI_WARNING, "Unknown variable '%s' in `let` statement"
 694                                         " of implementation '%s'", let_var->name, impl->func->name);
 695                         return 0;
 696                 }
 697
 698                 switch (let->val->kind) {
 699                 case PSI_LET_NULL:
 700                         break;
 701                 case PSI_LET_TMP:
 702                         /* e.g. let bar = &strval($bar); // decl_arg(char **bar) */
 703                         /* e.g. let foo = *bar;  */
 704                         let->var->pointer_level = let->val->data.var->pointer_level;
 705                         let->var->arg = init_decl_arg(
 706                                         init_decl_type(
 707                                                         real_decl_type(let->val->data.var->arg->type)->type,
 708                                                         real_decl_type(let->val->data.var->arg->type)->name),
 709                                         init_decl_var(
 710                                                         let->var->name,
 711                                                         let->var->pointer_level,
 712                                                         let->var->array_size));
 713                         break;
 714                 case PSI_LET_NUMEXP:
 715                         if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.num)) {
 716                                 return 0;
 717                         }
 718                         break;
 719                 case PSI_LET_CALLOC:
 720                         if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.alloc->nmemb)) {
 721                                 return 0;
 722                         }
 723                         if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.alloc->size)) {
 724                                 return 0;
 725                         }
 726                         break;
 727                 case PSI_LET_FUNC:
 728                         if (impl->func->args) {
 729                                 for (j = 0; j < impl->func->args->count; ++j) {
 730                                         impl_arg *iarg = impl->func->args->args[j];
 731
 732                                         if (!strcmp(let->val->data.func->var->name, iarg->var->name)) {
 733                                                 let->val->data.func->arg = iarg;
 734                                                 check = 1;
 735                                                 break;
 736                                         }
 737                                 }
 738                         }
 739                         if (!check) {
 740                                 data->error(PSI_WARNING, "Unknown value '$%s' of `let` statement"
 741                                                 " for variable '%s' of implementation '%s'",
 742                                                 let->val->data.func->var->name, let->var->name, impl->func->name);
 743                                 return 0;
 744                         }
 745                         break;
 746                 }
 747         }
 748
 749         /* check that we have a let stmt for every decl arg */
 750         if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
 751                 decl_arg *darg = impl->decl->args->args[i];
 752                 int check = 0;
 753
 754                 for (j = 0; j < impl->stmts->let.count; ++j) {
 755                         let_stmt *let = impl->stmts->let.list[j];
 756
 757                         if (!strcmp(let->var->name, darg->var->name)) {
 758                                 darg->let = let;
 759                                 check = 1;
 760                                 break;
 761                         }
 762                 }
 763                 if (!check) {
 764                         data->error(PSI_WARNING, "Missing `let` statement for arg '%s %.*s%s'"
 765                                         " of declaration '%s' for implementation '%s'",
 766                                         darg->type->name, (int) darg->var->pointer_level, "*****",
 767                                         darg->var->name, impl->decl->func->var->name, impl->func->name);
 768                         return 0;
 769                 }
 770         }
 771
 772         return 1;
 773 }
 774 static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
 775         size_t i, j, k;
 776         /* we can have any count of set stmts; processing out vars */
 777         /* check that set stmts reference known variables */
 778         for (i = 0; i < impl->stmts->set.count; ++i) {
 779                 set_stmt *set = impl->stmts->set.list[i];
 780                 int check = 0;
 781
 782                 if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
 783                         impl_arg *iarg = impl->func->args->args[j];
 784
 785                         if (!strcmp(set->var->name, iarg->var->name)) {
 786                                 set->arg = iarg;
 787                                 check = 1;
 788                                 break;
 789                         }
 790                 }
 791                 if (!check) {
 792                         data->error(PSI_WARNING, "Unknown variable '$%s' of `set` statement"
 793                                         " of implementation '%s'",
 794                                         set->var->name, impl->func->name);
 795                         return 0;
 796                 }
 797
 798                 for (j = 0; j < set->val->vars->count; ++j) {
 799                         decl_var *set_var = set->val->vars->vars[j];
 800
 801                         check = 0;
 802                         if (impl->decl->args) {
 803                                 for (k = 0; k < impl->decl->args->count; ++k) {
 804                                         decl_arg *set_arg = impl->decl->args->args[k];
 805
 806                                         if (!strcmp(set_var->name, set_arg->var->name)) {
 807                                                 check = 1;
 808                                                 set_var->arg = set_arg;
 809                                                 if (!validate_set_value(data, set->val, 1, &set_arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
 810                                                         return 0;
 811                                                 }
 812                                                 break;
 813                                         }
 814                                 }
 815                         }
 816                         if (!check) {
 817                                 for (k = 0; k < impl->stmts->let.count; ++k) {
 818                                         let_stmt *let = impl->stmts->let.list[k];
 819
 820                                         /* check temp vars */
 821                                         if (let->val && let->val->kind == PSI_LET_TMP) {
 822                                                 if (!strcmp(set_var->name, let->var->name)) {
 823                                                         check = 1;
 824                                                         set_var->arg = let->var->arg;
 825                                                         if (!validate_set_value(data, set->val, 1, &set_var->arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
 826                                                                 return 0;
 827                                                         }
 828                                                         break;
 829                                                 }
 830                                         }
 831                                 }
 832                         }
 833
 834                         if (!check) {
 835                                 data->error(PSI_WARNING, "Unknown value '%s' of `set` statement"
 836                                                 " for variable '$%s' of implementation '%s'",
 837                                                 set_var->name, set->arg->var->name, impl->func->name);
 838                                 return 0;
 839                         }
 840                 }
 841         }
 842         return 1;
 843 }
 844 static inline decl *locate_free_decl(decls *decls, free_call *f) {
 845         if (decls)  {
 846                 size_t i;
 847
 848                 for (i = 0; i < decls->count; ++i) {
 849                         if (!strcmp(decls->list[i]->func->var->name, f->func)) {
 850                                 f->decl = decls->list[i];
 851                                 return decls->list[i];
 852                         }
 853                 }
 854         }
 855
 856         return NULL;
 857 }
 858 static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
 859         size_t i, j, k, l;
 860         /* we can have any count of free stmts; freeing any out vars */
 861         for (i = 0; i < impl->stmts->fre.count; ++i) {
 862                 free_stmt *fre = impl->stmts->fre.list[i];
 863
 864                 for (j = 0; j < fre->calls->count; ++j) {
 865                         free_call *free_call = fre->calls->list[j];
 866
 867                         /* first find the decl of the free func */
 868                         if (!locate_free_decl(data->decls, free_call)) {
 869                                 data->error(PSI_WARNING, "Unknown function '%s' in `free` statement"
 870                                                 " of implementation '%s'", free_call->func, impl->func->name);
 871                                 return 0;
 872                         }
 873                         if (!impl->decl->args) {
 874                                 data->error(PSI_WARNING, "Declaration '%s' of implementation '%s'"
 875                                                 " does not have any arguments to free",
 876                                                 impl->decl->func->var->name, impl->func->name);
 877                         }
 878
 879                         /* now check for known vars */
 880                         for (l = 0; l < free_call->vars->count; ++l) {
 881                                 int check = 0;
 882                                 decl_var *free_var = free_call->vars->vars[l];
 883
 884                                 for (k = 0; k < impl->decl->args->count; ++k) {
 885                                         decl_arg *free_arg = impl->decl->args->args[k];
 886
 887                                         if (!strcmp(free_var->name, free_arg->var->name)) {
 888                                                 check = 1;
 889                                                 free_var->arg = free_arg;
 890                                                 break;
 891                                         }
 892                                 }
 893
 894                                 if (!check) {
 895                                         data->error(PSI_WARNING, "Unknown variable '%s' of `free` statement"
 896                                                         " of implementation '%s'",
 897                                                         free_var->name, impl->func->name);
 898                                         return 0;
 899                                 }
 900                         }
 901                 }
 902         }
 903         return 1;
 904 }
 905 static inline int validate_impl_stmts(PSI_Data *data, impl *impl) {
 906         if (!impl->stmts) {
 907                 data->error(PSI_WARNING, "Missing body for implementation %s!",
 908                                 impl->func->name);
 909                 return 0;
 910         }
 911
 912         if (!validate_impl_ret_stmt(data, impl)) {
 913                 return 0;
 914         }
 915
 916         if (!validate_impl_let_stmts(data, impl)) {
 917                 return 0;
 918         }
 919         if (!validate_impl_set_stmts(data, impl)) {
 920                 return 0;
 921         }
 922         if (!validate_impl_free_stmts(data, impl)) {
 923                 return 0;
 924         }
 925
 926         return 1;
 927 }
 928
 929 PSI_Context *PSI_ContextInit(PSI_Context *C, PSI_ContextOps *ops, PSI_ContextErrorFunc error)
 930 {
 931         size_t i;
 932         PSI_Data T;
 933         struct psi_predef_type *predef_type;
 934         struct psi_predef_const *predef_const;
 935         struct psi_predef_struct *predef_struct;
 936         struct psi_predef_decl *predef_decl;
 937
 938         if (!C) {
 939                 C = malloc(sizeof(*C));
 940         }
 941         memset(C, 0, sizeof(*C));
 942
 943         C->error = error;
 944         C->ops = ops;
 945
 946         if (ops->init) {
 947                 ops->init(C);
 948         }
 949
 950         ZEND_ASSERT(ops->call != NULL);
 951         ZEND_ASSERT(ops->compile != NULL);
 952
 953         /* build up predefs in a temporary PSI_Data for validation */
 954         memset(&T, 0, sizeof(T));
 955         T.error = error;
 956
 957         for (predef_type = &psi_predef_types[0]; predef_type->type_tag; ++predef_type) {
 958                 decl_type *type = init_decl_type(predef_type->type_tag, predef_type->type_name);
 959                 decl_typedef *def = init_decl_typedef(predef_type->alias, type);
 960
 961                 T.defs = add_decl_typedef(T.defs, def);
 962         }
 963         for (predef_const = &psi_predef_consts[0]; predef_const->type_tag; ++predef_const) {
 964                 impl_def_val *val = init_impl_def_val(predef_const->val_type_tag, predef_const->val_text);
 965                 const_type *type = init_const_type(predef_const->type_tag, predef_const->type_name);
 966                 constant *constant = init_constant(type, predef_const->var_name, val);
 967
 968                 T.consts = add_constant(T.consts, constant);
 969         }
 970         for (predef_struct = &psi_predef_structs[0]; predef_struct->type_tag; ++predef_struct) {
 971                 struct psi_predef_struct *member;
 972                 decl_args *dargs = init_decl_args(NULL);
 973                 decl_struct *dstruct = init_decl_struct(predef_struct->var_name, dargs);
 974
 975                 dstruct->size = predef_struct->size;
 976                 for (member = &predef_struct[1]; member->type_tag; ++member) {
 977                         decl_type *type;
 978                         decl_var *dvar;
 979                         decl_arg *darg;
 980
 981                         type = init_decl_type(member->type_tag, member->type_name);
 982                         dvar = init_decl_var(member->var_name, member->pointer_level, member->array_size);
 983                         darg = init_decl_arg(type, dvar);
 984                         darg->layout = init_decl_struct_layout(member->offset, member->size);
 985                         dargs = add_decl_arg(dargs, darg);
 986                 }
 987
 988                 T.structs = add_decl_struct(T.structs, dstruct);
 989                 predef_struct = member;
 990         }
 991         for (predef_decl = &psi_predef_decls[0]; predef_decl->type_tag; ++predef_decl) {
 992                 struct psi_predef_decl *farg;
 993                 decl_type *ftype = init_decl_type(predef_decl->type_tag, predef_decl->type_name);
 994                 decl_var *fname = init_decl_var(predef_decl->var_name, predef_decl->pointer_level, predef_decl->array_size);
 995                 decl_arg *func = init_decl_arg(ftype, fname);
 996                 decl_args *args = init_decl_args(NULL);
 997                 decl *decl = init_decl(init_decl_abi("default"), func, args);
 998
 999                 for (farg = &predef_decl[1]; farg->type_tag; ++farg) {
1000                         decl_type *arg_type = init_decl_type(farg->type_tag, farg->type_name);
1001                         decl_var *arg_var = init_decl_var(farg->var_name, farg->pointer_level, farg->array_size);
1002                         decl_arg *darg = init_decl_arg(arg_type, arg_var);
1003                         args = add_decl_arg(args, darg);
1004                 }
1005
1006                 T.decls = add_decl(T.decls, decl);
1007                 predef_decl = farg;
1008         }
1009
1010         for (predef_decl = &psi_predef_vararg_decls[0]; predef_decl->type_tag; ++predef_decl) {
1011                 struct psi_predef_decl *farg;
1012                 decl_type *ftype = init_decl_type(predef_decl->type_tag, predef_decl->type_name);
1013                 decl_var *fname = init_decl_var(predef_decl->var_name, predef_decl->pointer_level, predef_decl->array_size);
1014                 decl_arg *func = init_decl_arg(ftype, fname);
1015                 decl_args *args = init_decl_args(NULL);
1016                 decl *decl = init_decl(init_decl_abi("default"), func, args);
1017
1018                 for (farg = &predef_decl[1]; farg->type_tag; ++farg) {
1019                         decl_type *arg_type = init_decl_type(farg->type_tag, farg->type_name);
1020                         decl_var *arg_var = init_decl_var(farg->var_name, farg->pointer_level, farg->array_size);
1021                         decl_arg *darg = init_decl_arg(arg_type, arg_var);
1022                         args = add_decl_arg(args, darg);
1023                 }
1024                 args->varargs = 1;
1025
1026                 T.decls = add_decl(T.decls, decl);
1027                 predef_decl = farg;
1028         }
1029
1030         for (i = 0; i < T.defs->count; ++i) {
1031                 decl_typedef *def = T.defs->list[i];
1032
1033                 if (validate_decl_typedef(&T, def)) {
1034                         C->defs = add_decl_typedef(C->defs, def);
1035                 }
1036         }
1037
1038         for (i = 0; i < T.consts->count; ++i) {
1039                 constant *constant = T.consts->list[i];
1040
1041                 if (validate_constant(&T, constant)) {
1042                         C->consts = add_constant(C->consts, constant);
1043                 }
1044         }
1045
1046         for (i = 0; i < T.structs->count; ++i) {
1047                 decl_struct *dstruct = T.structs->list[i];
1048
1049                 if (validate_decl_struct(&T, dstruct)) {
1050                         C->structs = add_decl_struct(C->structs, dstruct);
1051                 }
1052         }
1053
1054         for (i = 0; i < T.decls->count; ++i) {
1055                 decl *decl = T.decls->list[i];
1056
1057                 if (validate_decl(&T, NULL, decl)) {
1058                         C->decls = add_decl(C->decls, decl);
1059                 }
1060         }
1061
1062         C->count = 1;
1063         C->data = malloc(sizeof(*C->data));
1064         PSI_DataExchange(C->data, &T);
1065
1066         return C;
1067 }
1068
1069 int PSI_ContextValidate(PSI_Context *C, PSI_Parser *P)
1070 {
1071         PSI_Data *D;
1072         void *dlopened = NULL;
1073         size_t count = C->count++;
1074
1075         C->data = realloc(C->data, C->count * sizeof(*C->data));
1076         D = PSI_DataExchange(&C->data[count], PSI_DATA(P));
1077
1078         if (D->defs) {
1079                 size_t i;
1080
1081                 for (i = 0; i < D->defs->count; ++i) {
1082                         if (validate_decl_typedef(PSI_DATA(C), D->defs->list[i])) {
1083                                 C->defs = add_decl_typedef(C->defs, D->defs->list[i]);
1084                         }
1085                 }
1086         }
1087         if (D->structs) {
1088                 size_t i;
1089
1090                 for (i = 0; i < D->structs->count; ++i) {
1091                         if (validate_decl_struct(PSI_DATA(C), D->structs->list[i])) {
1092                                 C->structs = add_decl_struct(C->structs, D->structs->list[i]);
1093                         }
1094                 }
1095         }
1096         if (D->consts) {
1097                 size_t i;
1098
1099                 for (i = 0; i < D->consts->count; ++i) {
1100                         if (validate_constant(PSI_DATA(C), D->consts->list[i])) {
1101                                 C->consts = add_constant(C->consts, D->consts->list[i]);
1102                         }
1103                 }
1104         }
1105
1106         if (!validate_lib(D, &dlopened)) {
1107                 return 0;
1108         }
1109
1110         add_decl_lib(&C->psi.libs, dlopened);
1111
1112         if (D->decls) {
1113                 size_t i;
1114
1115                 for (i = 0; i < D->decls->count; ++i) {
1116                         if (validate_decl(PSI_DATA(C), dlopened, D->decls->list[i])) {
1117                                 C->decls = add_decl(C->decls, D->decls->list[i]);
1118                         }
1119                 }
1120         }
1121         if (D->impls) {
1122                 size_t i;
1123
1124                 for (i = 0; i < D->impls->count; ++i) {
1125                         if (validate_impl_stmts(PSI_DATA(C), D->impls->list[i])) {
1126                                 C->impls = add_impl(C->impls, D->impls->list[i]);
1127                         }
1128                 }
1129         }
1130
1131         return 1;
1132 }
1133
1134 static int psi_select_dirent(const struct dirent *entry)
1135 {
1136 #ifndef FNM_CASEFOLD
1137 #define FNM_CASEFOLD 0
1138 #endif
1139         return 0 == fnmatch("*.psi", entry->d_name, FNM_CASEFOLD);
1140 }
1141
1142 void PSI_ContextBuild(PSI_Context *C, const char *paths)
1143 {
1144         int i, n, flags = psi_check_env("PSI_DEBUG") ? PSI_PARSER_DEBUG : 0;
1145         char *sep = NULL, *cpy = strdup(paths), *ptr = cpy;
1146         struct dirent **entries = NULL;
1147
1148
1149         do {
1150                 sep = strchr(ptr, ':');
1151
1152                 if (sep) {
1153                         *sep = 0;
1154                 }
1155
1156                 n = php_scandir(ptr, &entries, psi_select_dirent, alphasort);
1157
1158                 if (n > 0) {
1159                         for (i = 0; i < n; ++i) {
1160                                 char psi[MAXPATHLEN];
1161                                 PSI_Parser P;
1162
1163                                 if (MAXPATHLEN <= slprintf(psi, MAXPATHLEN, "%s/%s", ptr, entries[i]->d_name)) {
1164                                         C->error(PSI_WARNING, "Path to PSI file too long: %s/%s",
1165                                                 ptr, entries[i]->d_name);
1166                                 }
1167                                 if (!PSI_ParserInit(&P, psi, C->error, flags)) {
1168                                         C->error(PSI_WARNING, "Failed to init PSI parser (%s): %s",
1169                                                 psi, strerror(errno));
1170                                         continue;
1171                                 }
1172
1173                                 while (0 < PSI_ParserScan(&P)) {
1174                                         PSI_ParserParse(&P, PSI_TokenAlloc(&P));
1175                                         if (P.num == PSI_T_EOF) {
1176                                                 break;
1177                                         }
1178                                 }
1179
1180                                 PSI_ParserParse(&P, NULL);
1181                                 PSI_ContextValidate(C, &P);
1182                                 PSI_ParserDtor(&P);
1183                         }
1184                 }
1185
1186                 if (entries) {
1187                         for (i = 0; i < n; ++i) {
1188                                 free(entries[i]);
1189                         }
1190                         free(entries);
1191                 }
1192
1193                 ptr = sep + 1;
1194         } while (sep);
1195
1196
1197         if (PSI_ContextCompile(C) && SUCCESS != zend_register_functions(NULL, C->closures, NULL, MODULE_PERSISTENT)) {
1198                 C->error(PSI_WARNING, "Failed to register functions!");
1199         }
1200
1201         free(cpy);
1202
1203 }
1204
1205 zend_function_entry *PSI_ContextCompile(PSI_Context *C)
1206 {
1207         size_t i;
1208
1209         if (C->consts) {
1210                 zend_constant zc;
1211
1212                 zc.flags = CONST_PERSISTENT|CONST_CS;
1213                 zc.module_number = EG(current_module)->module_number;
1214
1215                 for (i = 0; i < C->consts->count; ++i) {
1216                         constant *c = C->consts->list[i];
1217
1218                         zc.name = zend_string_init(c->name + (c->name[0] == '\\'), strlen(c->name) - (c->name[0] == '\\'), 1);
1219                         ZVAL_NEW_STR(&zc.value, zend_string_init(c->val->text, strlen(c->val->text), 1));
1220
1221                         switch (c->type->type) {
1222                         case PSI_T_BOOL:
1223                                 convert_to_boolean(&zc.value);
1224                                 break;
1225                         case PSI_T_INT:
1226                                 convert_to_long(&zc.value);
1227                                 break;
1228                         case PSI_T_FLOAT:
1229                                 convert_to_double(&zc.value);
1230                                 break;
1231                         }
1232                         zend_register_constant(&zc);
1233                 }
1234         }
1235
1236         return C->closures = C->ops->compile(C);
1237 }
1238
1239
1240 void PSI_ContextCall(PSI_Context *C, decl_callinfo *decl_call, impl_vararg *va)
1241 {
1242         C->ops->call(C, decl_call, va);
1243 }
1244
1245 static inline void dump_decl_type(int fd, decl_type *t) {
1246         const char *pre;
1247
1248         switch (t->type) {
1249         case PSI_T_STRUCT:
1250                 pre = "struct ";
1251                 break;
1252         default:
1253                 pre = "";
1254         }
1255         dprintf(fd, "%s%s", pre, t->name);
1256 }
1257 static inline void dump_decl_var(int fd, decl_var *v) {
1258         dprintf(fd, "%.*s%s", v->pointer_level-!!v->array_size, "**********", v->name);
1259         if (v->array_size) {
1260                 dprintf(fd, "[%u]", v->array_size);
1261         }
1262 }
1263 static inline void dump_decl_arg(int fd, decl_arg *a) {
1264         dump_decl_type(fd, a->type);
1265         dprintf(fd, " ");
1266         dump_decl_var(fd, a->var);
1267 }
1268 static inline void dump_level(int fd, unsigned level) {
1269         dprintf(fd, "%.*s", level, "\t\t\t\t\t\t\t\t\t");
1270 }
1271 static inline void dump_num_exp(int fd, num_exp *exp) {
1272         while (exp) {
1273                 switch (exp->t) {
1274                 case PSI_T_NUMBER:
1275                         dprintf(fd, "%s", exp->u.numb);
1276                         break;
1277                 case PSI_T_NSNAME:
1278                         dprintf(fd, "%s", exp->u.cnst->name);
1279                         break;
1280                 case PSI_T_NAME:
1281                         dump_decl_var(fd, exp->u.dvar);
1282                         break;
1283                 EMPTY_SWITCH_DEFAULT_CASE();
1284                 }
1285                 if (exp->operand) {
1286                         char op;
1287
1288                         switch (exp->operator) {
1289                         case PSI_T_PLUS:        op = '+'; break;
1290                         case PSI_T_MINUS:       op = '-'; break;
1291                         case PSI_T_ASTERISK:op = '*'; break;
1292                         case PSI_T_SLASH:       op = '/'; break;
1293                         EMPTY_SWITCH_DEFAULT_CASE();
1294                         }
1295                         dprintf(fd, " %c ", op);
1296                 }
1297                 exp = exp->operand;
1298         }
1299 }
1300 static inline void dump_impl_set_value(int fd, set_value *set, unsigned level) {
1301         size_t i;
1302
1303         if (level > 1) {
1304                 /* only if not directly after `set ...` */
1305                 dump_level(fd, level);
1306         }
1307         dprintf(fd, "%s(", set->func->name);
1308
1309         for (i = 0; i < set->vars->count; ++i) {
1310                 decl_var *svar = set->vars->vars[i];
1311                 if (i) {
1312                         dprintf(fd, ", ");
1313                 }
1314                 dump_decl_var(fd, svar);
1315         }
1316         if (set->num) {
1317                 dprintf(fd, ", ");
1318                 dump_num_exp(fd, set->num);
1319         }
1320         if (set->inner) {
1321                 dprintf(fd, ",\n");
1322                 for (i = 0; i < set->count; ++i) {
1323                         dump_impl_set_value(fd, set->inner[i], level+1);
1324                 }
1325                 /* only if inner stmts, i.e. with new lines, were dumped */
1326                 dump_level(fd, level);
1327         }
1328         if (level > 1) {
1329                 dprintf(fd, "),\n");
1330         } else {
1331                 dprintf(fd, ");\n");
1332         }
1333 }
1334 void PSI_ContextDump(PSI_Context *C, int fd)
1335 {
1336         size_t i, j, k, l;
1337
1338 #ifdef HAVE_LIBJIT
1339         if (C->ops == PSI_Libjit()) {
1340                 dprintf(fd, "// psi.engine=jit\n");
1341         }
1342 #endif
1343 #ifdef HAVE_LIBFFI
1344         if (C->ops == PSI_Libffi()) {
1345                 dprintf(fd, "// psi.engine=ffi\n");
1346         }
1347 #endif
1348         dprintf(fd, "\n");
1349
1350         if (C->defs) {
1351                 for (i = 0; i < C->defs->count; ++i) {
1352                         decl_typedef *tdef = C->defs->list[i];
1353
1354                         dprintf(fd, "typedef ");
1355                         dump_decl_type(fd, tdef->type);
1356                         dprintf(fd, " %s;\n", tdef->alias);
1357                 }
1358                 dprintf(fd, "\n");
1359         }
1360
1361         if (C->structs) {
1362                 for (i = 0; i < C->structs->count; ++i) {
1363                         decl_struct *strct = C->structs->list[i];
1364
1365                         dprintf(fd, "struct %s::(%zu) {\n", strct->name, strct->size);
1366                         if (strct->args) for (j = 0; j < strct->args->count; ++j) {
1367                                 decl_arg *sarg = strct->args->args[j];
1368
1369                                 dprintf(fd, "\t");
1370                                 dump_decl_arg(fd, sarg);
1371                                 dprintf(fd, "::(%zu, %zu);\n", sarg->layout->pos, sarg->layout->len);
1372                         }
1373                         dprintf(fd, "}\n");
1374                 }
1375                 dprintf(fd, "\n");
1376         }
1377         if (C->consts) {
1378                 for (i = 0; i < C->consts->count; ++i) {
1379                         constant *cnst = C->consts->list[i];
1380
1381                         dprintf(fd, "const %s %s = ", cnst->type->name, cnst->name);
1382                         if (cnst->val->type == PSI_T_QUOTED_STRING) {
1383                                 dprintf(fd, "\"%s\";\n", cnst->val->text);
1384                         } else {
1385                                 dprintf(fd, "%s;\n", cnst->val->text);
1386                         }
1387                 }
1388                 dprintf(fd, "\n");
1389         }
1390         if (C->decls) {
1391                 for (i = 0; i < C->decls->count; ++i) {
1392                         decl *decl = C->decls->list[i];
1393
1394                         dprintf(fd, "%s ", decl->abi->convention);
1395                         dump_decl_arg(fd, decl->func);
1396                         dprintf(fd, "(");
1397                         if (decl->args) {
1398                                 for (j = 0; j < decl->args->count; ++j) {
1399                                         if (j) {
1400                                                 dprintf(fd, ", ");
1401                                         }
1402                                         dump_decl_arg(fd, decl->args->args[j]);
1403                                 }
1404                                 if (decl->args->varargs) {
1405                                         dprintf(fd, ", ...");
1406                                 }
1407                         }
1408                         dprintf(fd, ");\n");
1409                 }
1410                 dprintf(fd, "\n");
1411         }
1412         if (C->impls) {
1413                 for (i = 0; i < C->impls->count; ++i) {
1414                         impl *impl = C->impls->list[i];
1415
1416                         dprintf(fd, "function %s(", impl->func->name);
1417                         if (impl->func->args) {
1418                                 for (j = 0; j < impl->func->args->count; ++j) {
1419                                         impl_arg *iarg = impl->func->args->args[j];
1420
1421                                         dprintf(fd, "%s%s %s$%s",
1422                                                         j ? ", " : "",
1423                                                         iarg->type->name,
1424                                                         iarg->var->reference ? "&" : "",
1425                                                         iarg->var->name);
1426                                         if (iarg->def) {
1427                                                 dprintf(fd, " = %s", iarg->def->text);
1428                                         }
1429                                 }
1430                                 if (impl->func->args->vararg.name) {
1431                                         impl_arg *vararg = impl->func->args->vararg.name;
1432
1433                                         dprintf(fd, ", %s %s...$%s",
1434                                                         vararg->type->name,
1435                                                         vararg->var->reference ? "&" : "",
1436                                                         vararg->var->name);
1437                                 }
1438                         }
1439                         dprintf(fd, ") : %s%s {\n",
1440                                         impl->func->return_reference ? "&":"",
1441                                         impl->func->return_type->name);
1442                         if (impl->stmts) {
1443                                 for (j = 0; j < impl->stmts->let.count; ++j) {
1444                                         let_stmt *let = impl->stmts->let.list[j];
1445
1446                                         dprintf(fd, "\tlet %s", let->var->name);
1447                                         if (let->val) {
1448                                                 dprintf(fd, " = %s", let->val->flags.one.is_reference ? "&" : "");
1449                                                 switch (let->val->kind) {
1450                                                 case PSI_LET_NULL:
1451                                                         dprintf(fd, "NULL");
1452                                                         break;
1453                                                 case PSI_LET_TMP:
1454                                                         dump_decl_var(fd, let->val->data.var);
1455                                                         break;
1456                                                 case PSI_LET_CALLOC:
1457                                                         dprintf(fd, "calloc(");
1458                                                         dump_num_exp(fd, let->val->data.alloc->nmemb);
1459                                                         dprintf(fd, ", ");
1460                                                         dump_num_exp(fd, let->val->data.alloc->size);
1461                                                         dprintf(fd, ")");
1462                                                         break;
1463                                                 case PSI_LET_FUNC:
1464                                                         dprintf(fd, "%s($%s)", let->val->data.func->name,
1465                                                                         let->val->data.func->var->name);
1466                                                         break;
1467
1468                                                 EMPTY_SWITCH_DEFAULT_CASE();
1469                                                 }
1470                                                 dprintf(fd, ";\n");
1471                                         }
1472                                 }
1473                                 for (j = 0; j < impl->stmts->ret.count; ++j) {
1474                                         return_stmt *ret = impl->stmts->ret.list[j];
1475
1476                                         dprintf(fd, "\treturn ");
1477                                         dump_impl_set_value(fd, ret->set, 1);
1478                                 }
1479                                 for (j = 0; j < impl->stmts->set.count; ++j) {
1480                                         set_stmt *set = impl->stmts->set.list[j];
1481
1482                                         dprintf(fd, "\tset $%s = ", set->var->name);
1483                                         dump_impl_set_value(fd, set->val, 1);
1484                                 }
1485                                 for (j = 0; j < impl->stmts->fre.count; ++j) {
1486                                         free_stmt *fre = impl->stmts->fre.list[j];
1487
1488                                         dprintf(fd, "\tfree ");
1489                                         for (k = 0; k < fre->calls->count; ++k) {
1490                                                 free_call *call = fre->calls->list[k];
1491
1492                                                 if (k) {
1493                                                         dprintf(fd, ", ");
1494                                                 }
1495                                                 dprintf(fd, "%s(", call->func);
1496                                                 for (l = 0; l < call->vars->count; ++l) {
1497                                                         decl_var *fvar = call->vars->vars[l];
1498
1499                                                         dump_decl_var(fd, fvar);
1500                                                 }
1501                                                 dprintf(fd, ");\n");
1502                                         }
1503                                 }
1504                         }
1505                         dprintf(fd, "}\n");
1506                 }
1507                 dprintf(fd, "\n");
1508         }
1509 }
1510
1511 void PSI_ContextDtor(PSI_Context *C)
1512 {
1513         size_t i;
1514         zend_function_entry *zfe;
1515
1516         if (C->ops->dtor) {
1517                 C->ops->dtor(C);
1518         }
1519
1520         free_decl_libs(&C->psi.libs);
1521
1522         if (C->data) {
1523                 for (i = 0; i < C->count; ++i) {
1524                         PSI_DataDtor(&C->data[i]);
1525                 }
1526                 free(C->data);
1527         }
1528
1529         if (C->closures) {
1530                 for (zfe = C->closures; zfe->fname; ++zfe) {
1531                         free((void *) zfe->arg_info);
1532                 }
1533                 free(C->closures);
1534         }
1535
1536         if (C->consts) {
1537                 if (C->consts->list) {
1538                         free(C->consts->list);
1539                 }
1540                 free(C->consts);
1541         }
1542         if (C->defs) {
1543                 if (C->defs->list) {
1544                         free(C->defs->list);
1545                 }
1546                 free(C->defs);
1547         }
1548         if (C->structs) {
1549                 if (C->structs->list) {
1550                         free(C->structs->list);
1551                 }
1552                 free(C->structs);
1553         }
1554         if (C->decls) {
1555                 if (C->decls->list) {
1556                         free(C->decls->list);
1557                 }
1558                 free(C->decls);
1559         }
1560         if (C->impls) {
1561                 if (C->impls->list) {
1562                         free(C->impls->list);
1563                 }
1564                 free(C->impls);
1565         }
1566
1567         memset(C, 0, sizeof(*C));
1568 }
1569
1570 void PSI_ContextFree(PSI_Context **C)
1571 {
1572         if (*C) {
1573                 PSI_ContextDtor(*C);
1574                 free(*C);
1575                 *C = NULL;
1576         }
1577 }