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_SYS_TIME_H
  48 # include <sys/time.h>
  49 #endif
  50 #ifdef HAVE_SYS_TIMES_H
  51 # include <sys/times.h>
  52 #endif
  53 #ifdef HAVE_SYS_UTSNAME_H
  54 # include <sys/utsname.h>
  55 #endif
  56 #ifdef HAVE_TIME_H
  57 # include <time.h>
  58 #endif
  59 #ifdef HAVE_WCHAR_H
  60 # include <wchar.h>
  61 #endif
  62
  63 #ifdef HAVE_DIRENT_H
  64 # include <dirent.h>
  65 # define NAMLEN(dirent) strlen ((dirent)->d_name)
  66 #else
  67 # define dirent direct
  68 # define NAMLEN(dirent) ((dirent)->d_namlen)
  69 # ifdef HAVE_SYS_NDIR_H
  70 #  include <sys/ndir.h>
  71 # endif
  72 # ifdef HAVE_SYS_DIR_H
  73 #  include <sys/dir.h>
  74 # endif
  75 # ifdef HAVE_NDIR_H
  76 #  include <ndir.h>
  77 # endif
  78 #endif
  79
  80 #include <sys/param.h>
  81 #include <dlfcn.h>
  82 #include <fnmatch.h>
  83
  84 #include "php.h"
  85 #include "php_scandir.h"
  86 #include "php_psi.h"
  87 #include "context.h"
  88 #include "parser.h"
  89
  90 #include "libjit.h"
  91 #include "libffi.h"
  92
  93 #define psi_predef_count(of) ((sizeof(psi_predef ##of## s)/sizeof(psi_predef ##of))-1)
  94 typedef struct psi_predef_type {
  95         token_t type_tag;
  96         const char *type_name;
  97         const char *alias;
  98 } psi_predef_type;
  99 static const psi_predef_type psi_predef_types[] = {
 100         PHP_PSI_TYPES{0}
 101 };
 102 #define psi_predef_type_count() psi_predef_count(_type)
 103
 104 typedef struct psi_predef_const {
 105         token_t type_tag;
 106         const char *type_name;
 107         const char *name;
 108         const char *val_text;
 109         token_t val_type_tag;
 110 } psi_predef_const;
 111 static const psi_predef_const psi_predef_consts[] = {
 112         PHP_PSI_CONSTS{0}
 113 };
 114 #define psi_predef_const_count() psi_predef_count(_const)
 115
 116 typedef struct psi_predef_struct_member {
 117         token_t type_tag;
 118         const char *type_name;
 119         const char *name;
 120         size_t off;
 121         size_t len;
 122         size_t pointer_level;
 123         size_t array_size;
 124 } psi_predef_struct_member;
 125 #define PSI_PREDEF_STRUCT_MEMBERS 32
 126 typedef struct psi_predef_struct {
 127         const char *name;
 128         size_t size;
 129         psi_predef_struct_member members[PSI_PREDEF_STRUCT_MEMBERS];
 130 } psi_predef_struct;
 131 static const psi_predef_struct psi_predef_structs[] = {
 132         PHP_PSI_STRUCTS{0}
 133 };
 134 #define psi_predef_struct_count() psi_predef_count(_struct)
 135
 136 PHP_PSI_MACROS
 137
 138 typedef struct psi_predef_func {
 139         const char *name;
 140         void (*func)(void);
 141 } psi_predef_func;
 142 static psi_predef_func psi_predef_funcs[] = {
 143         PHP_PSI_FUNCS{0}
 144 };
 145 #define psi_predef_func_count() psi_predef_count(_func)
 146
 147 static int validate_lib(PSI_Data *data, void **dlopened) {
 148         char lib[MAXPATHLEN];
 149         const char *ptr = data->psi.file.ln;
 150         size_t len;
 151
 152         if (!ptr) {
 153                 /* FIXME: assume stdlib */
 154                 return 1;
 155         } else if (!strchr(ptr, '/')) {
 156                 len = snprintf(lib, MAXPATHLEN, "lib%s.%s", ptr, PHP_PSI_SHLIB_SUFFIX);
 157                 if (MAXPATHLEN == len) {
 158                         data->error(PSI_WARNING, "Library name too long: '%s'", ptr);
 159                 }
 160                 lib[len] = 0;
 161                 ptr = lib;
 162         }
 163         if (!(*dlopened = dlopen(ptr, RTLD_LAZY|RTLD_LOCAL))) {
 164                 data->error(PSI_WARNING, "Could not open library '%s': %s.",
 165                                 data->psi.file.ln, dlerror());
 166                 return 0;
 167         }
 168         return 1;
 169 }
 170
 171 static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
 172         size_t i;
 173
 174         if (type->real) {
 175                 return 1;
 176         }
 177         for (i = 0; i < defs->count; ++i) {
 178                 if (!strcmp(defs->list[i]->alias, type->name)) {
 179                         type->real = defs->list[i]->type;
 180                         return 1;
 181                 }
 182         }
 183         return 0;
 184 }
 185 static inline int locate_decl_type_struct(decl_structs *structs, decl_type *type) {
 186         size_t i;
 187
 188         if (type->strct) {
 189                 return 1;
 190         }
 191         for (i = 0; i < structs->count; ++i) {
 192                 if (!strcmp(structs->list[i]->name, type->name)) {
 193                         type->strct = structs->list[i];
 194                         return 1;
 195                 }
 196         }
 197         return 0;
 198 }
 199
 200 static inline int validate_decl_type(PSI_Data *data, decl_type *type) {
 201         switch (type->type) {
 202         case PSI_T_NAME:
 203                 if (!data->defs |!locate_decl_type_alias(data->defs, type)) {
 204                         return 0;
 205                 }
 206                 return validate_decl_type(data, type->real);
 207         case PSI_T_STRUCT:
 208                 if (!data->structs || !locate_decl_type_struct(data->structs, type)) {
 209                         return 0;
 210                 }
 211                 break;
 212         }
 213         return 1;
 214 }
 215 static inline int validate_decl_typedef(PSI_Data *data, decl_typedef *def) {
 216         if (!validate_decl_type(data, def->type)) {
 217                 data->error(PSI_WARNING, "Type '%s' cannot be aliased to '%s'",
 218                         def->type->name, def->alias);
 219                 return 0;
 220         }
 221         /* FIXME: check def->alias */
 222         return 1;
 223 }
 224
 225 static inline int validate_constant(PSI_Data *data, constant *c) {
 226         /* FIXME */
 227         return 1;
 228 }
 229
 230 static inline int validate_decl_arg(PSI_Data *data, decl_arg *arg) {
 231         if (!validate_decl_type(data, arg->type)) {
 232                 data->error(PSI_WARNING, "Cannot use '%s' as type for '%s'",
 233                         arg->type->name, arg->var->name);
 234                 return 0;
 235         }
 236         return 1;
 237 }
 238
 239 static inline int validate_decl_struct(PSI_Data *data, decl_struct *s) {
 240         size_t i;
 241
 242         for (i = 0; i < s->args->count; ++i) {
 243                 if (!validate_decl_arg(data, s->args->args[i])) {
 244                         return 0;
 245                 }
 246         }
 247
 248         for (i = 0; i < s->args->count; ++i) {
 249                 decl_arg *darg = s->args->args[i];
 250
 251                 if (!validate_decl_arg(data, darg)) {
 252                         return 0;
 253                 }
 254
 255                 ZEND_ASSERT(!darg->var->arg || darg->var->arg == darg);
 256                 darg->var->arg = darg;
 257
 258                 if (!darg->layout) {
 259                         token_t t;
 260
 261                         if (darg->var->pointer_level && (!darg->var->array_size || darg->var->pointer_level == 1)) {
 262                                 t = PSI_T_POINTER;
 263                         } else {
 264                                 t = real_decl_type(darg->type)->type;
 265                         }
 266
 267                         if (i) {
 268                                 decl_arg *last = s->args->args[i-1];
 269                                 darg->layout = init_decl_struct_layout(
 270                                                 psi_t_align(t, last->layout->pos + last->layout->len),
 271                                                 psi_t_size(t) * darg->var->array_size);
 272                         } else {
 273                                 darg->layout = init_decl_struct_layout(0, psi_t_size(t));
 274                         }
 275                 }
 276                 if (s->size < darg->layout->pos + darg->layout->len) {
 277                         s->size = darg->layout->pos + darg->layout->len;
 278                 }
 279         }
 280         return 1;
 281 }
 282
 283 static const char * const abi_ccs[] = {
 284                 "default", /* \                 */
 285                 "extern",  /*  > - all the same */
 286                 "cdecl",   /* /                 */
 287                 "stdcall",
 288                 "fastcall",
 289 };
 290 static inline int validate_decl_abi(PSI_Data *data, decl_abi *abi) {
 291         size_t i;
 292
 293         for (i = 0; i < sizeof(abi_ccs)/sizeof(char*); ++i) {
 294                 if (strcasecmp(abi->convention, abi_ccs[i])) {
 295                         return 1;
 296                 }
 297         }
 298         data->error(PSI_WARNING, "Invalid calling convention: '%s'", abi->convention);
 299         return 0;
 300 }
 301 static inline int validate_decl_func(PSI_Data *data, void *dl, decl *decl, decl_arg *func)
 302 {
 303         size_t i;
 304
 305         if (!strcmp(func->var->name, "dlsym")) {
 306                 data->error(PSI_WARNING, "Cannot dlsym dlsym (sic!)");
 307                 return 0;
 308         }
 309
 310         if (!validate_decl_arg(data, func)) {
 311                 return 0;
 312         }
 313         for (i = 0; i < psi_predef_func_count(); ++i) {
 314                 psi_predef_func *pre = &psi_predef_funcs[i];
 315
 316                 if (!strcmp(func->var->name, pre->name)) {
 317                         decl->call.sym = pre->func;
 318                         break;
 319                 }
 320         }
 321         if (!decl->call.sym) {
 322 #ifndef RTLD_NEXT
 323 # define RTLD_NEXT ((void *) -1l)
 324 #endif
 325                 decl->call.sym = dlsym(dl ?: RTLD_NEXT, func->var->name);
 326                 if (!decl->call.sym) {
 327                         data->error(PSI_WARNING, "Failed to locate symbol '%s': %s",
 328                                 func->var->name, dlerror());
 329                 }
 330         }
 331         return 1;
 332 }
 333
 334 static inline int validate_decl(PSI_Data *data, void *dl, decl *decl) {
 335         if (!validate_decl_abi(data, decl->abi)) {
 336                 return 0;
 337         }
 338         if (!validate_decl_func(data, dl, decl, decl->func)) {
 339                 return 0;
 340         }
 341         if (decl->args) {
 342                 size_t i;
 343
 344                 for (i = 0; i < decl->args->count; ++i) {
 345                         if (!validate_decl_arg(data, decl->args->args[i])) {
 346                                 return 0;
 347                         }
 348                 }
 349         }
 350         return 1;
 351 }
 352 static inline decl_arg *locate_decl_var_arg(decl_var *var, decl_args *args) {
 353         size_t i;
 354
 355         for (i = 0; i < args->count; ++i) {
 356                 decl_arg *arg = args->args[i];
 357
 358                 if (!strcmp(var->name, arg->var->name)) {
 359                         return var->arg = arg;
 360                 }
 361         }
 362
 363         return NULL;
 364 }
 365 static inline decl_arg *locate_struct_member(decl_struct *s, decl_var *var) {
 366         if (s->args) {
 367                 return locate_decl_var_arg(var, s->args);
 368         }
 369
 370         return NULL;
 371 }
 372 static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
 373         size_t i;
 374         decl_type *ref_type = real_decl_type(ref->type);
 375         decl_var *set_var = set->vars->vars[0];
 376
 377         switch (set->func->type) {
 378         case PSI_T_TO_BOOL:
 379                 set->func->handler = psi_to_bool;
 380                 break;
 381         case PSI_T_TO_INT:
 382                 set->func->handler = psi_to_int;
 383                 break;
 384         case PSI_T_TO_FLOAT:
 385                 set->func->handler = psi_to_double;
 386                 break;
 387         case PSI_T_TO_STRING:
 388                 set->func->handler = psi_to_string;
 389                 break;
 390         case PSI_T_TO_ARRAY:
 391                 set->func->handler = psi_to_array;
 392                 break;
 393         EMPTY_SWITCH_DEFAULT_CASE();
 394         }
 395
 396         for (i = 1; i < set->vars->count; ++i) {
 397                 if (!locate_decl_var_arg(set->vars->vars[i], ref_list)) {
 398                         return 0;
 399                 }
 400         }
 401         set_var->arg = ref;
 402
 403         if (set->count && (set->func->type != PSI_T_TO_ARRAY || (ref_type->type != PSI_T_STRUCT && !ref->var->arg->var->pointer_level))) {
 404                 data->error(E_WARNING, "Inner `set` statement casts only work with to_array() casts on structs or pointers");
 405                 return 0;
 406         }
 407
 408         if (ref_type->type == PSI_T_STRUCT) {
 409                 /* to_array(struct, to_...) */
 410                 for (i = 0; i < set->count; ++i) {
 411                         decl_var *sub_var = set->inner[i]->vars->vars[0];
 412                         decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
 413
 414                         if (sub_ref) {
 415                                 if (!validate_set_value(data, set->inner[i], sub_ref, ref_type->strct->args)) {
 416                                         return 0;
 417                                 }
 418                                 set->inner[i]->outer.set = set;
 419                         }
 420                 }
 421         } else if (set->count == 1) {
 422                 /* to_array(ptr, to_string(*ptr)) */
 423                 decl_var *sub_var = set->inner[0]->vars->vars[0];
 424                 decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list);
 425
 426                 if (sub_ref) {
 427                         if (strcmp(sub_var->name, set_var->name)) {
 428                                 data->error(E_WARNING, "Inner `set` statement casts on pointers must reference the same variable");
 429                                 return 0;
 430                         }
 431                         if (!validate_set_value(data, set->inner[0], sub_ref, ref_list)) {
 432                                 return 0;
 433                         }
 434                         set->inner[0]->outer.set = set;
 435                 }
 436         } else if (set->count > 1) {
 437                 data->error(E_WARNING, "Inner `set` statement casts on pointers may only occur once");
 438                 return 0;
 439         }
 440
 441         return 1;
 442 }
 443 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
 444         if (decls) {
 445                 size_t i;
 446
 447                 for (i = 0; i < decls->count; ++i) {
 448                         if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
 449                                 ret->decl = decls->list[i]->func;
 450                                 return decls->list[i];
 451                         }
 452                 }
 453         }
 454
 455         return NULL;
 456 }
 457 static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 458         return_stmt *ret;
 459
 460         /* we must have exactly one ret stmt delcaring the native func to call */
 461         /* and which type cast to apply */
 462         if (impl->stmts->ret.count != 1) {
 463                 if (impl->stmts->ret.count > 1) {
 464                         data->error(PSI_WARNING, "Too many `return` statements for implmentation %s;"
 465                                         " found %zu, exactly one is needed",
 466                                         impl->func->name, impl->stmts->ret.count);
 467                 } else {
 468                         data->error(PSI_WARNING, "Missing `return` statement for implementation %s",
 469                                         impl->func->name);
 470                 }
 471                 return 0;
 472         }
 473
 474         ret = impl->stmts->ret.list[0];
 475
 476         if (!(impl->decl = locate_impl_decl(data->decls, ret))) {
 477                 data->error(PSI_WARNING, "Missing declaration for implementation %s",
 478                                 impl->func->name);
 479                 return 0;
 480         }
 481
 482         if (!validate_set_value(data, ret->set, ret->decl, impl->decl->args)) {
 483                 return 0;
 484         }
 485
 486         impl->decl->impl = impl;
 487
 488         return 1;
 489 }
 490 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
 491         size_t i, j;
 492         /* we can have multiple let stmts */
 493         /* check that we have a let stmt for every decl arg */
 494         if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
 495                 decl_arg *darg = impl->decl->args->args[i];
 496                 int check = 0;
 497
 498                 for (j = 0; j < impl->stmts->let.count; ++j) {
 499                         let_stmt *let = impl->stmts->let.list[j];
 500
 501                         if (!strcmp(let->var->name, darg->var->name)) {
 502                                 darg->let = let;
 503                                 check = 1;
 504                                 break;
 505                         }
 506                 }
 507                 if (!check) {
 508                         data->error(PSI_WARNING, "Missing `let` statement for arg '%s %.*s%s'"
 509                                         " of declaration '%s' for implementation '%s'",
 510                                         darg->type->name, (int) darg->var->pointer_level, "*****",
 511                                         darg->var->name, impl->decl->func->var->name, impl->func->name);
 512                         return 0;
 513                 }
 514         }
 515         /* check that the let_value references a known variable or NULL */
 516         for (i = 0; i < impl->stmts->let.count; ++i) {
 517                 let_stmt *let = impl->stmts->let.list[i];
 518                 int check = 0;
 519
 520                 if (let->val && let->val->func && let->val->func->alloc) {
 521                         if (!validate_decl_type(data, let->val->func->alloc->type)) {
 522                                 data->error(PSI_WARNING, "Cannot use '%s' as type for calloc in `let` statement",
 523                                         let->val->func->alloc->type->name);
 524                                 return 0;
 525                         }
 526                 }
 527                 if (let->val && let->val->var) {
 528                         if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
 529                                 impl_arg *iarg = impl->func->args->args[j];
 530
 531                                 if (!strcmp(let->val->var->name, iarg->var->name)) {
 532                                         let->arg = iarg;
 533                                         check = 1;
 534                                         break;
 535                                 }
 536                         }
 537                         if (!check) {
 538                                 data->error(PSI_WARNING, "Unknown value '$%s' of `let` statement"
 539                                                 " for variable '%s' of implementation '%s'",
 540                                                 let->val->var->name, let->var->name, impl->func->name);
 541                                 return 0;
 542                         }
 543                 }
 544         }
 545         return 1;
 546 }
 547 static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
 548         size_t i, j, k;
 549         /* we can have any count of set stmts; processing out vars */
 550         /* check that set stmts reference known variables */
 551         for (i = 0; i < impl->stmts->set.count; ++i) {
 552                 set_stmt *set = impl->stmts->set.list[i];
 553                 int check = 0;
 554
 555                 if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
 556                         impl_arg *iarg = impl->func->args->args[j];
 557
 558                         if (!strcmp(set->var->name, iarg->var->name)) {
 559                                 set->arg = iarg;
 560                                 check = 1;
 561                                 break;
 562                         }
 563                 }
 564                 if (!check) {
 565                         data->error(PSI_WARNING, "Unknown variable '$%s' of `set` statement"
 566                                         " of implementation '%s'",
 567                                         set->var->name, impl->func->name);
 568                         return 0;
 569                 }
 570
 571                 for (j = 0; j < set->val->vars->count; ++j) {
 572                         decl_var *set_var = set->val->vars->vars[j];
 573
 574                         check = 0;
 575                         if (impl->decl->args) for (k = 0; k < impl->decl->args->count; ++k) {
 576                                 decl_arg *set_arg = impl->decl->args->args[k];
 577
 578                                 if (!strcmp(set_var->name, set_arg->var->name)) {
 579                                         check = 1;
 580                                         if (!validate_set_value(data, set->val, set_arg, impl->decl->args)) {
 581                                                 return 0;
 582                                         }
 583                                         set_var->arg = set_arg;
 584                                         break;
 585                                 }
 586                         }
 587
 588                         if (!check) {
 589                                 data->error(PSI_WARNING, "Unknown value '%s' of `set` statement"
 590                                                 " for variable '$%s' of implementation '%s'",
 591                                                 set_var->name, set->arg->var->name, impl->func->name);
 592                                 return 0;
 593                         }
 594                 }
 595         }
 596         return 1;
 597 }
 598 static inline decl *locate_free_decl(decls *decls, free_call *f) {
 599         if (decls)  {
 600                 size_t i;
 601
 602                 for (i = 0; i < decls->count; ++i) {
 603                         if (!strcmp(decls->list[i]->func->var->name, f->func)) {
 604                                 f->decl = decls->list[i];
 605                                 return decls->list[i];
 606                         }
 607                 }
 608         }
 609
 610         return NULL;
 611 }
 612 static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
 613         size_t i, j, k, l;
 614         /* we can have any count of free stmts; freeing any out vars */
 615         for (i = 0; i < impl->stmts->fre.count; ++i) {
 616                 free_stmt *fre = impl->stmts->fre.list[i];
 617
 618                 for (j = 0; j < fre->calls->count; ++j) {
 619                         free_call *free_call = fre->calls->list[j];
 620
 621                         /* first find the decl of the free func */
 622                         if (!locate_free_decl(data->decls, free_call)) {
 623                                 data->error(PSI_WARNING, "Unknown function '%s' in `free` statement"
 624                                                 " of implementation '%s'", free_call->func, impl->func->name);
 625                                 return 0;
 626                         }
 627                         if (!impl->decl->args) {
 628                                 data->error(PSI_WARNING, "Declaration '%s' of implementation '%s'"
 629                                                 " does not have any arguments to free",
 630                                                 impl->decl->func->var->name, impl->func->name);
 631                         }
 632
 633                         /* now check for known vars */
 634                         for (l = 0; l < free_call->vars->count; ++l) {
 635                                 int check = 0;
 636                                 decl_var *free_var = free_call->vars->vars[l];
 637
 638                                 for (k = 0; k < impl->decl->args->count; ++k) {
 639                                         decl_arg *free_arg = impl->decl->args->args[k];
 640
 641                                         if (!strcmp(free_var->name, free_arg->var->name)) {
 642                                                 check = 1;
 643                                                 free_var->arg = free_arg;
 644                                                 break;
 645                                         }
 646                                 }
 647
 648                                 if (!check) {
 649                                         data->error(PSI_WARNING, "Unknown variable '%s' of `free` statement"
 650                                                         " of implementation '%s'",
 651                                                         free_var->name, impl->func->name);
 652                                         return 0;
 653                                 }
 654                         }
 655                 }
 656         }
 657         return 1;
 658 }
 659 static inline int validate_impl_stmts(PSI_Data *data, impl *impl) {
 660         if (!impl->stmts) {
 661                 data->error(PSI_WARNING, "Missing body for implementation %s!",
 662                                 impl->func->name);
 663                 return 0;
 664         }
 665
 666         if (!validate_impl_ret_stmt(data, impl)) {
 667                 return 0;
 668         }
 669
 670         if (!validate_impl_let_stmts(data, impl)) {
 671                 return 0;
 672         }
 673         if (!validate_impl_set_stmts(data, impl)) {
 674                 return 0;
 675         }
 676         if (!validate_impl_free_stmts(data, impl)) {
 677                 return 0;
 678         }
 679
 680         return 1;
 681 }
 682
 683 PSI_Context *PSI_ContextInit(PSI_Context *C, PSI_ContextOps *ops, PSI_ContextErrorFunc error)
 684 {
 685         size_t i, j;
 686         PSI_Data T;
 687
 688         if (!C) {
 689                 C = malloc(sizeof(*C));
 690         }
 691         memset(C, 0, sizeof(*C));
 692
 693         C->error = error;
 694         C->ops = ops;
 695
 696         if (ops->init) {
 697                 ops->init(C);
 698         }
 699
 700         ZEND_ASSERT(ops->call != NULL);
 701         ZEND_ASSERT(ops->compile != NULL);
 702
 703         /* build up predefs in a temporary PSI_Data for validation */
 704         memset(&T, 0, sizeof(T));
 705         T.error = error;
 706
 707         for (i = 0; i < psi_predef_type_count(); ++i) {
 708                 const psi_predef_type *pre = &psi_predef_types[i];
 709                 decl_type *type = init_decl_type(pre->type_tag, pre->type_name);
 710                 decl_typedef *def = init_decl_typedef(pre->alias, type);
 711
 712                 T.defs = add_decl_typedef(T.defs, def);
 713         }
 714         for (i = 0; i < psi_predef_const_count(); ++i) {
 715                 const psi_predef_const *pre = &psi_predef_consts[i];
 716                 impl_def_val *val = init_impl_def_val(pre->val_type_tag, pre->val_text);
 717                 const_type *type = init_const_type(pre->type_tag, pre->type_name);
 718                 constant *constant = init_constant(type, pre->name, val);
 719
 720                 T.consts = add_constant(T.consts, constant);
 721         }
 722         for (i = 0; i < psi_predef_struct_count(); ++i) {
 723                 const psi_predef_struct *pre = &psi_predef_structs[i];
 724                 decl_args *dargs = init_decl_args(NULL);
 725                 decl_struct *dstruct;
 726
 727                 for (j = 0; j < PSI_PREDEF_STRUCT_MEMBERS; ++j) {
 728                         const psi_predef_struct_member *member = &pre->members[j];
 729                         decl_type *type;
 730                         decl_var *dvar;
 731                         decl_arg *darg;
 732
 733                         if (!member->name) {
 734                                 break;
 735                         }
 736
 737                         type = init_decl_type(member->type_tag, member->type_name);
 738                         dvar = init_decl_var(member->name, member->pointer_level, member->array_size);
 739                         darg = init_decl_arg(type, dvar);
 740                         darg->layout = init_decl_struct_layout(member->off, member->len);
 741                         dargs = add_decl_arg(dargs, darg);
 742                 }
 743
 744                 dstruct = init_decl_struct(pre->name, dargs);
 745                 dstruct->size = pre->size;
 746                 T.structs = add_decl_struct(T.structs, dstruct);
 747         }
 748
 749         for (i = 0; i < psi_predef_type_count(); ++i) {
 750                 decl_typedef *def = T.defs->list[i];
 751
 752                 if (validate_decl_typedef(&T, def)) {
 753                         C->defs = add_decl_typedef(C->defs, def);
 754                 }
 755         }
 756
 757         for (i = 0; i < psi_predef_const_count(); ++i) {
 758                 constant *constant = T.consts->list[i];
 759
 760                 if (validate_constant(&T, constant)) {
 761                         C->consts = add_constant(C->consts, constant);
 762                 }
 763         }
 764
 765         for (i = 0; i < psi_predef_struct_count(); ++i) {
 766                 decl_struct *dstruct = T.structs->list[i];
 767
 768                 if (validate_decl_struct(&T, dstruct)) {
 769                         C->structs = add_decl_struct(C->structs, dstruct);
 770                 }
 771         }
 772
 773         C->count = 1;
 774         C->data = malloc(sizeof(*C->data));
 775         PSI_DataExchange(C->data, &T);
 776
 777         return C;
 778 }
 779
 780 int PSI_ContextValidate(PSI_Context *C, PSI_Parser *P)
 781 {
 782         PSI_Data *D;
 783         void *dlopened = NULL;
 784         size_t count = C->count++;
 785
 786         C->data = realloc(C->data, C->count * sizeof(*C->data));
 787         D = PSI_DataExchange(&C->data[count], PSI_DATA(P));
 788
 789         if (D->defs) {
 790                 size_t i;
 791
 792                 for (i = 0; i < D->defs->count; ++i) {
 793                         if (validate_decl_typedef(PSI_DATA(C), D->defs->list[i])) {
 794                                 C->defs = add_decl_typedef(C->defs, D->defs->list[i]);
 795                         }
 796                 }
 797         }
 798         if (D->structs) {
 799                 size_t i;
 800
 801                 for (i = 0; i < D->structs->count; ++i) {
 802                         if (validate_decl_struct(PSI_DATA(C), D->structs->list[i])) {
 803                                 C->structs = add_decl_struct(C->structs, D->structs->list[i]);
 804                         }
 805                 }
 806         }
 807         if (D->consts) {
 808                 size_t i;
 809
 810                 for (i = 0; i < D->consts->count; ++i) {
 811                         if (validate_constant(PSI_DATA(C), D->consts->list[i])) {
 812                                 C->consts = add_constant(C->consts, D->consts->list[i]);
 813                         }
 814                 }
 815         }
 816
 817         if (!validate_lib(D, &dlopened)) {
 818                 return 0;
 819         }
 820
 821         add_decl_lib(&C->psi.libs, dlopened);
 822
 823         if (D->decls) {
 824                 size_t i;
 825
 826                 for (i = 0; i < D->decls->count; ++i) {
 827                         if (validate_decl(PSI_DATA(C), dlopened, D->decls->list[i])) {
 828                                 C->decls = add_decl(C->decls, D->decls->list[i]);
 829                         }
 830                 }
 831         }
 832         if (D->impls) {
 833                 size_t i;
 834
 835                 for (i = 0; i < D->impls->count; ++i) {
 836                         if (validate_impl_stmts(PSI_DATA(C), D->impls->list[i])) {
 837                                 C->impls = add_impl(C->impls, D->impls->list[i]);
 838                         }
 839                 }
 840         }
 841
 842         return 1;
 843 }
 844
 845 static int psi_select_dirent(const struct dirent *entry)
 846 {
 847 #ifndef FNM_CASEFOLD
 848 #define FNM_CASEFOLD 0
 849 #endif
 850         return 0 == fnmatch("*.psi", entry->d_name, FNM_CASEFOLD);
 851 }
 852
 853 void PSI_ContextBuild(PSI_Context *C, const char *paths)
 854 {
 855         int i, n, flags = getenv("PSI_DEBUG") ? PSI_PARSER_DEBUG : 0;
 856         char *sep = NULL, *cpy = strdup(paths), *ptr = cpy;
 857         struct dirent **entries = NULL;
 858
 859
 860         do {
 861                 sep = strchr(ptr, ':');
 862
 863                 if (sep) {
 864                         *sep = 0;
 865                 }
 866
 867                 n = php_scandir(ptr, &entries, psi_select_dirent, alphasort);
 868
 869                 if (n > 0) {
 870                         for (i = 0; i < n; ++i) {
 871                                 char psi[MAXPATHLEN];
 872                                 PSI_Parser P;
 873
 874                                 if (MAXPATHLEN <= slprintf(psi, MAXPATHLEN, "%s/%s", ptr, entries[i]->d_name)) {
 875                                         C->error(PSI_WARNING, "Path to PSI file too long: %s/%s",
 876                                                 ptr, entries[i]->d_name);
 877                                 }
 878                                 if (!PSI_ParserInit(&P, psi, C->error, flags)) {
 879                                         C->error(PSI_WARNING, "Failed to init PSI parser (%s): %s",
 880                                                 psi, strerror(errno));
 881                                         continue;
 882                                 }
 883
 884                                 while (-1 != PSI_ParserScan(&P)) {
 885                                         PSI_ParserParse(&P, PSI_TokenAlloc(&P));
 886                                 };
 887                                 PSI_ParserParse(&P, NULL);
 888                                 PSI_ContextValidate(C, &P);
 889                                 PSI_ParserDtor(&P);
 890                         }
 891                 }
 892
 893                 if (entries) {
 894                         for (i = 0; i < n; ++i) {
 895                                 free(entries[i]);
 896                         }
 897                         free(entries);
 898                 }
 899
 900                 ptr = sep + 1;
 901         } while (sep);
 902
 903
 904         if (PSI_ContextCompile(C) && SUCCESS != zend_register_functions(NULL, C->closures, NULL, MODULE_PERSISTENT)) {
 905                 C->error(PSI_WARNING, "Failed to register functions!");
 906         }
 907
 908         free(cpy);
 909
 910 }
 911
 912 zend_function_entry *PSI_ContextCompile(PSI_Context *C)
 913 {
 914         size_t i;
 915
 916         if (C->consts) {
 917                 zend_constant zc;
 918
 919                 zc.flags = CONST_PERSISTENT|CONST_CS;
 920                 zc.module_number = EG(current_module)->module_number;
 921
 922                 for (i = 0; i < C->consts->count; ++i) {
 923                         constant *c = C->consts->list[i];
 924
 925                         zc.name = zend_string_init(c->name + (c->name[0] == '\\'), strlen(c->name) - (c->name[0] == '\\'), 1);
 926                         ZVAL_NEW_STR(&zc.value, zend_string_init(c->val->text, strlen(c->val->text), 1));
 927
 928                         switch (c->type->type) {
 929                         case PSI_T_BOOL:
 930                                 convert_to_boolean(&zc.value);
 931                                 break;
 932                         case PSI_T_INT:
 933                                 convert_to_long(&zc.value);
 934                                 break;
 935                         case PSI_T_FLOAT:
 936                                 convert_to_double(&zc.value);
 937                                 break;
 938                         }
 939                         zend_register_constant(&zc);
 940                 }
 941         }
 942
 943         return C->closures = C->ops->compile(C);
 944 }
 945
 946
 947 void PSI_ContextCall(PSI_Context *C, impl_val *ret_val, decl *decl)
 948 {
 949         C->ops->call(C, ret_val, decl);
 950 }
 951
 952 static inline void dump_decl_type(int fd, decl_type *t) {
 953         const char *pre;
 954
 955         switch (t->type) {
 956         case PSI_T_STRUCT:
 957                 pre = "struct ";
 958                 break;
 959         default:
 960                 pre = "";
 961         }
 962         dprintf(fd, "%s%s", pre, t->name);
 963 }
 964 static inline void dump_decl_var(int fd, decl_var *v) {
 965         dprintf(fd, "%.*s%s", v->pointer_level-!!v->array_size, "**********", v->name);
 966         if (v->array_size) {
 967                 dprintf(fd, "[%u]", v->array_size);
 968         }
 969 }
 970 static inline void dump_decl_arg(int fd, decl_arg *a) {
 971         dump_decl_type(fd, a->type);
 972         dprintf(fd, " ");
 973         dump_decl_var(fd, a->var);
 974 }
 975 void PSI_ContextDump(PSI_Context *C, int fd)
 976 {
 977         size_t i, j;
 978
 979 #ifdef HAVE_LIBJIT
 980         if (C->ops == PSI_Libjit()) {
 981                 dprintf(fd, "#PSI(libjit)\n");
 982         }
 983 #endif
 984 #ifdef HAVE_LIBFFI
 985         if (C->ops == PSI_Libffi()) {
 986                 dprintf(fd, "#PSI(libffi)\n");
 987         }
 988 #endif
 989
 990         if (C->defs) for (i = 0; i < C->defs->count; ++i) {
 991                 decl_typedef *tdef = C->defs->list[i];
 992
 993                 dprintf(fd, "typedef ");
 994                 dump_decl_type(fd, tdef->type);
 995                 dprintf(fd, " %s;\n", tdef->alias);
 996         }
 997         if (C->structs) for (i = 0; i < C->structs->count; ++i) {
 998                 decl_struct *strct = C->structs->list[i];
 999                 decl_arg *sarg = NULL;
1000
1001                 dprintf(fd, "struct %s::(%zu) {\n", strct->name, strct->size);
1002                 for (j = 0; j < strct->args->count; ++j) {
1003                         sarg = strct->args->args[j];
1004                         dprintf(fd, "\t");
1005                         dump_decl_arg(fd, sarg);
1006                         dprintf(fd, "::(%zu, %zu);\n", sarg->layout->pos, sarg->layout->len);
1007                 }
1008                 dprintf(fd, "}\n");
1009         }
1010
1011         dprintf(fd, "\n");
1012 }
1013
1014 void PSI_ContextDtor(PSI_Context *C)
1015 {
1016         size_t i;
1017         zend_function_entry *zfe;
1018
1019         if (C->ops->dtor) {
1020                 C->ops->dtor(C);
1021         }
1022
1023         free_decl_libs(&C->psi.libs);
1024
1025         if (C->data) {
1026                 for (i = 0; i < C->count; ++i) {
1027                         PSI_DataDtor(&C->data[i]);
1028                 }
1029                 free(C->data);
1030         }
1031
1032         if (C->closures) {
1033                 for (zfe = C->closures; zfe->fname; ++zfe) {
1034                         free((void *) zfe->arg_info);
1035                 }
1036                 free(C->closures);
1037         }
1038
1039         if (C->consts) {
1040                 if (C->consts->list) {
1041                         free(C->consts->list);
1042                 }
1043                 free(C->consts);
1044         }
1045         if (C->defs) {
1046                 if (C->defs->list) {
1047                         free(C->defs->list);
1048                 }
1049                 free(C->defs);
1050         }
1051         if (C->structs) {
1052                 if (C->structs->list) {
1053                         free(C->structs->list);
1054                 }
1055                 free(C->structs);
1056         }
1057         if (C->decls) {
1058                 if (C->decls->list) {
1059                         free(C->decls->list);
1060                 }
1061                 free(C->decls);
1062         }
1063         if (C->impls) {
1064                 if (C->impls->list) {
1065                         free(C->impls->list);
1066                 }
1067                 free(C->impls);
1068         }
1069
1070         memset(C, 0, sizeof(*C));
1071 }
1072
1073 void PSI_ContextFree(PSI_Context **C)
1074 {
1075         if (*C) {
1076                 PSI_ContextDtor(*C);
1077                 free(*C);
1078                 *C = NULL;
1079         }
1080 }