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

   1 #include <sys/param.h>
   2 #include <dlfcn.h>
   3 #include <dirent.h>
   4 #include <fnmatch.h>
   5 #include <errno.h>
   6
   7 #ifdef HAVE_CONFIG_H
   8 # include "config.h"
   9 #endif
  10
  11 #include "php.h"
  12 #include "php_scandir.h"
  13 #include "php_psi.h"
  14 #include "context.h"
  15 #include "parser.h"
  16
  17 #define psi_predef_count(of) ((sizeof(psi_predef ##of## s)/sizeof(psi_predef ##of))-1)
  18 typedef struct psi_predef_type {
  19         token_t type_tag;
  20         const char *type_name;
  21         const char *alias;
  22 } psi_predef_type;
  23 static const psi_predef_type psi_predef_types[] = {
  24         PHP_PSI_TYPES{0}
  25 };
  26 #define psi_predef_type_count() psi_predef_count(_type)
  27
  28 typedef struct psi_predef_const {
  29         token_t type_tag;
  30         const char *type_name;
  31         const char *name;
  32         const char *val_text;
  33         token_t val_type_tag;
  34 } psi_predef_const;
  35 static const psi_predef_const psi_predef_consts[] = {
  36         PHP_PSI_CONSTS{0}
  37 };
  38 #define psi_predef_const_count() psi_predef_count(_const)
  39
  40 typedef struct psi_predef_struct_member {
  41         token_t type_tag;
  42         const char *type_name;
  43         const char *name;
  44         size_t off;
  45         size_t len;
  46         size_t pointer_level;
  47         size_t array_size;
  48 } psi_predef_struct_member;
  49 #define PSI_PREDEF_STRUCT_MEMBERS 32
  50 typedef struct psi_predef_struct {
  51         const char *name;
  52         size_t size;
  53         psi_predef_struct_member members[PSI_PREDEF_STRUCT_MEMBERS];
  54 } psi_predef_struct;
  55 static const psi_predef_struct psi_predef_structs[] = {
  56         PHP_PSI_STRUCTS{0}
  57 };
  58 #define psi_predef_struct_count() psi_predef_count(_struct)
  59
  60 typedef struct psi_predef_func {
  61         const char *name;
  62         void (*func)(void);
  63 } psi_predef_func;
  64 static psi_predef_func psi_predef_funcs[] = {
  65         PHP_PSI_FUNCS{0}
  66 };
  67 #define psi_predef_func_count() psi_predef_count(_func)
  68
  69 static int validate_lib(PSI_Data *data, void **dlopened) {
  70         char lib[MAXPATHLEN];
  71         const char *ptr = data->psi.file.ln;
  72         size_t len;
  73
  74         if (!ptr) {
  75                 /* FIXME: assume stdlib */
  76                 return 1;
  77         } else if (!strchr(ptr, '/')) {
  78 #ifdef DARWIN
  79                 len = snprintf(lib, MAXPATHLEN, "lib%s.dylib", ptr);
  80 #else
  81                 len = snprintf(lib, MAXPATHLEN, "lib%s.so", ptr);
  82 #endif
  83                 if (MAXPATHLEN == len) {
  84                         data->error(PSI_WARNING, "Library name too long: '%s'", ptr);
  85                 }
  86                 lib[len] = 0;
  87                 ptr = lib;
  88         }
  89         if (!(*dlopened = dlopen(ptr, RTLD_LAZY|RTLD_LOCAL))) {
  90                 data->error(PSI_WARNING, "Could not open library '%s': %s.",
  91                                 data->psi.file.ln, dlerror());
  92                 return 0;
  93         }
  94         return 1;
  95 }
  96
  97 static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
  98         size_t i;
  99
 100         if (type->real) {
 101                 return 1;
 102         }
 103         for (i = 0; i < defs->count; ++i) {
 104                 if (!strcmp(defs->list[i]->alias, type->name)) {
 105                         type->real = defs->list[i]->type;
 106                         return 1;
 107                 }
 108         }
 109         return 0;
 110 }
 111 static inline int locate_decl_type_struct(decl_structs *structs, decl_type *type) {
 112         size_t i;
 113
 114         if (type->strct) {
 115                 return 1;
 116         }
 117         for (i = 0; i < structs->count; ++i) {
 118                 if (!strcmp(structs->list[i]->name, type->name)) {
 119                         type->strct = structs->list[i];
 120                         return 1;
 121                 }
 122         }
 123         return 0;
 124 }
 125
 126 static inline int validate_decl_type(PSI_Data *data, decl_type *type) {
 127         switch (type->type) {
 128         case PSI_T_NAME:
 129                 if (!data->defs || !locate_decl_type_alias(data->defs, type)) {
 130                         return 0;
 131                 }
 132                 return validate_decl_type(data, type->real);
 133         case PSI_T_STRUCT:
 134                 if (!data->structs || !locate_decl_type_struct(data->structs, type)) {
 135                         return 0;
 136                 }
 137                 break;
 138         }
 139         return 1;
 140 }
 141 static inline int validate_decl_typedef(PSI_Data *data, decl_typedef *def) {
 142         if (!validate_decl_type(data, def->type)) {
 143                 data->error(PSI_WARNING, "Type '%s' cannot be aliased to '%s'",
 144                         def->type->name, def->alias);
 145                 return 0;
 146         }
 147         /* FIXME: check def->alias */
 148         return 1;
 149 }
 150
 151 static inline int validate_constant(PSI_Data *data, constant *c) {
 152         /* FIXME */
 153         return 1;
 154 }
 155
 156 static inline int validate_decl_arg(PSI_Data *data, decl_arg *arg) {
 157         if (!validate_decl_type(data, arg->type)) {
 158                 data->error(PSI_WARNING, "Cannot use '%s' as type for '%s'",
 159                         arg->type->name, arg->var->name);
 160                 return 0;
 161         }
 162         return 1;
 163 }
 164
 165 static inline int validate_decl_struct(PSI_Data *data, decl_struct *s) {
 166         size_t i;
 167
 168         for (i = 0; i < s->args->count; ++i) {
 169                 if (!validate_decl_arg(data, s->args->args[i])) {
 170                         return 0;
 171                 }
 172         }
 173
 174         for (i = 0; i < s->args->count; ++i) {
 175                 decl_arg *darg = s->args->args[i];
 176
 177                 if (!validate_decl_arg(data, darg)) {
 178                         return 0;
 179                 }
 180
 181                 ZEND_ASSERT(!darg->var->arg || darg->var->arg == darg);
 182                 darg->var->arg = darg;
 183
 184                 if (!darg->layout) {
 185                         token_t t;
 186
 187                         if (darg->var->pointer_level && (!darg->var->array_size || darg->var->pointer_level == 1)) {
 188                                 t = PSI_T_POINTER;
 189                         } else {
 190                                 t = real_decl_type(darg->type)->type;
 191                         }
 192
 193                         if (i) {
 194                                 decl_arg *last = s->args->args[i-1];
 195                                 darg->layout = init_decl_struct_layout(
 196                                                 psi_t_align(t, last->layout->pos + last->layout->len),
 197                                                 psi_t_size(t) * darg->var->array_size);
 198                         } else {
 199                                 darg->layout = init_decl_struct_layout(0, psi_t_size(t));
 200                         }
 201                 }
 202                 if (s->size < darg->layout->pos + darg->layout->len) {
 203                         s->size = darg->layout->pos + darg->layout->len;
 204                 }
 205         }
 206         return 1;
 207 }
 208
 209 static const char * const abi_ccs[] = {
 210                 "default", /* \                 */
 211                 "extern",  /*  > - all the same */
 212                 "cdecl",   /* /                 */
 213                 "stdcall",
 214                 "fastcall",
 215 };
 216 static inline int validate_decl_abi(PSI_Data *data, decl_abi *abi) {
 217         size_t i;
 218
 219         for (i = 0; i < sizeof(abi_ccs)/sizeof(char*); ++i) {
 220                 if (strcasecmp(abi->convention, abi_ccs[i])) {
 221                         return 1;
 222                 }
 223         }
 224         data->error(PSI_WARNING, "Invalid calling convention: '%s'", abi->convention);
 225         return 0;
 226 }
 227 static inline int validate_decl_func(PSI_Data *data, void *dl, decl *decl, decl_arg *func)
 228 {
 229         if (!strcmp(func->var->name, "dlsym")) {
 230                 data->error(PSI_WARNING, "Cannot dlsym dlsym (sic!)");
 231                 return 0;
 232         }
 233
 234         if (!validate_decl_arg(data, func)) {
 235                 return 0;
 236         }
 237 #ifndef RTLD_NEXT
 238 # define RTLD_NEXT ((void *) -1l)
 239 #endif
 240         decl->dlptr = dlsym(dl ?: RTLD_NEXT, func->var->name);
 241         if (!decl->dlptr) {
 242                 size_t i;
 243
 244                 for (i = 0; i < psi_predef_func_count(); ++i) {
 245                         psi_predef_func *pre = &psi_predef_funcs[i];
 246
 247                         if (!strcmp(func->var->name, pre->name)) {
 248                                 decl->dlptr = pre->func;
 249                                 break;
 250                         }
 251                 }
 252                 if (!decl->dlptr) {
 253                         data->error(PSI_WARNING, "Failed to locate symbol '%s': %s",
 254                                 func->var->name, dlerror());
 255                 }
 256         }
 257         return 1;
 258 }
 259
 260 static inline int validate_decl(PSI_Data *data, void *dl, decl *decl) {
 261         if (!validate_decl_abi(data, decl->abi)) {
 262                 return 0;
 263         }
 264         if (!validate_decl_func(data, dl, decl, decl->func)) {
 265                 return 0;
 266         }
 267         if (decl->args) {
 268                 size_t i;
 269
 270                 for (i = 0; i < decl->args->count; ++i) {
 271                         if (!validate_decl_arg(data, decl->args->args[i])) {
 272                                 return 0;
 273                         }
 274                 }
 275         }
 276         return 1;
 277 }
 278
 279 static inline decl_arg *locate_struct_member(decl_struct *s, decl_var *var) {
 280         size_t i;
 281
 282         ZEND_ASSERT(s);
 283         for (i = 0; i < s->args->count; ++i) {
 284                 decl_arg *darg = s->args->args[i];
 285
 286                 if (!strcmp(var->name, darg->var->name)) {
 287                         return var->arg = darg;
 288                 }
 289         }
 290
 291         return NULL;
 292 }
 293 static inline int validate_set_value(PSI_Data *data, set_value *set, decl_arg *ref) {
 294         size_t i;
 295         decl_type *ref_type = real_decl_type(ref->type);
 296         decl_var *set_var = set->vars->vars[0];
 297
 298         switch (set->func->type) {
 299         case PSI_T_TO_BOOL:
 300                 set->func->handler = psi_to_bool;
 301                 break;
 302         case PSI_T_TO_INT:
 303                 set->func->handler = psi_to_int;
 304                 break;
 305         case PSI_T_TO_FLOAT:
 306                 set->func->handler = psi_to_double;
 307                 break;
 308         case PSI_T_TO_STRING:
 309                 set->func->handler = psi_to_string;
 310                 break;
 311         case PSI_T_TO_ARRAY:
 312                 set->func->handler = psi_to_array;
 313                 break;
 314         EMPTY_SWITCH_DEFAULT_CASE();
 315         }
 316
 317         if (strcmp(set_var->name, ref->var->name)) {
 318                 return 0;
 319         }
 320
 321         if (set->count && (set->func->type != PSI_T_TO_ARRAY || ref_type->type != PSI_T_STRUCT)) {
 322                 data->error(E_WARNING, "Inner `set` statement casts only work with to_array() casts on structs");
 323                 return 0;
 324         }
 325         for (i = 0; i < set->count; ++i) {
 326                 decl_var *sub_var = set->inner[i]->vars->vars[0];
 327                 decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
 328
 329                 if (sub_ref) {
 330                         if (!validate_set_value(data, set->inner[i], sub_ref)) {
 331                                 return 0;
 332                         }
 333                 }
 334         }
 335
 336         return 1;
 337 }
 338 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
 339         size_t i;
 340
 341         for (i = 0; i < decls->count; ++i) {
 342                 if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
 343                         ret->decl = decls->list[i]->func;
 344                         return decls->list[i];
 345                 }
 346         }
 347         return NULL;
 348 }
 349 static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
 350         return_stmt *ret;
 351
 352         /* we must have exactly one ret stmt delcaring the native func to call */
 353         /* and which type cast to apply */
 354         if (impl->stmts->ret.count != 1) {
 355                 if (impl->stmts->ret.count > 1) {
 356                         data->error(PSI_WARNING, "Too many `return` statements for implmentation %s;"
 357                                         " found %zu, exactly one is needed",
 358                                         impl->func->name, impl->stmts->ret.count);
 359                 } else {
 360                         data->error(PSI_WARNING, "Missing `return` statement for implementation %s",
 361                                         impl->func->name);
 362                 }
 363                 return 0;
 364         }
 365
 366         ret = impl->stmts->ret.list[0];
 367
 368         if (!(impl->decl = locate_impl_decl(data->decls, ret))) {
 369                 data->error(PSI_WARNING, "Missing declaration for implementation %s",
 370                                 impl->func->name);
 371                 return 0;
 372         }
 373         if (!validate_set_value(data, ret->set, ret->decl)) {
 374                 return 0;
 375         }
 376
 377         return 1;
 378 }
 379 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
 380         size_t i, j;
 381         /* we can have multiple let stmts */
 382         /* check that we have a let stmt for every decl arg */
 383         if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
 384                 decl_arg *darg = impl->decl->args->args[i];
 385                 int check = 0;
 386
 387                 for (j = 0; j < impl->stmts->let.count; ++j) {
 388                         let_stmt *let = impl->stmts->let.list[j];
 389
 390                         if (!strcmp(let->var->name, darg->var->name)) {
 391                                 darg->let = let;
 392                                 check = 1;
 393                                 break;
 394                         }
 395                 }
 396                 if (!check) {
 397                         data->error(PSI_WARNING, "Missing `let` statement for arg '%s %.*s%s'"
 398                                         " of declaration '%s' for implementation '%s'",
 399                                         darg->type->name, (int) darg->var->pointer_level, "*****",
 400                                         darg->var->name, impl->decl->func->var->name, impl->func->name);
 401                         return 0;
 402                 }
 403         }
 404         /* check that the let_value references a known variable or NULL */
 405         for (i = 0; i < impl->stmts->let.count; ++i) {
 406                 let_stmt *let = impl->stmts->let.list[i];
 407                 int check = 0;
 408
 409                 if (let->val && let->val->func && let->val->func->alloc) {
 410                         if (!validate_decl_type(data, let->val->func->alloc->type)) {
 411                                 data->error(PSI_WARNING, "Cannot use '%s' as type for calloc in `let` statement",
 412                                         let->val->func->alloc->type->name);
 413                                 return 0;
 414                         }
 415                 }
 416                 if (let->val && let->val->var) {
 417                         if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
 418                                 impl_arg *iarg = impl->func->args->args[j];
 419
 420                                 if (!strcmp(let->val->var->name, iarg->var->name)) {
 421                                         let->arg = iarg;
 422                                         check = 1;
 423                                         break;
 424                                 }
 425                         }
 426                         if (!check) {
 427                                 data->error(PSI_WARNING, "Unknown value '$%s' of `let` statement"
 428                                                 " for variable '%s' of implementation '%s'",
 429                                                 let->val->var->name, let->var->name, impl->func->name);
 430                                 return 0;
 431                         }
 432                 }
 433         }
 434         return 1;
 435 }
 436 static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
 437         size_t i, j, k;
 438         /* we can have any count of set stmts; processing out vars */
 439         /* check that set stmts reference known variables */
 440         for (i = 0; i < impl->stmts->set.count; ++i) {
 441                 set_stmt *set = impl->stmts->set.list[i];
 442                 int check = 0;
 443
 444                 if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
 445                         impl_arg *iarg = impl->func->args->args[j];
 446
 447                         if (!strcmp(set->var->name, iarg->var->name)) {
 448                                 set->arg = iarg;
 449                                 check = 1;
 450                                 break;
 451                         }
 452                 }
 453                 if (!check) {
 454                         data->error(PSI_WARNING, "Unknown variable '$%s' of `set` statement"
 455                                         " of implementation '%s'",
 456                                         set->var->name, impl->func->name);
 457                         return 0;
 458                 }
 459
 460                 for (j = 0; j < set->val->vars->count; ++j) {
 461                         decl_var *set_var = set->val->vars->vars[j];
 462
 463                         check = 0;
 464                         if (impl->decl->args) for (k = 0; k < impl->decl->args->count; ++k) {
 465                                 decl_arg *set_arg = impl->decl->args->args[k];
 466
 467                                 if (!strcmp(set_var->name, set_arg->var->name)) {
 468                                         check = 1;
 469                                         if (!validate_set_value(data, set->val, set_arg)) {
 470                                                 return 0;
 471                                         }
 472                                         set_var->arg = set_arg;
 473                                         break;
 474                                 }
 475                         }
 476
 477                         if (!check) {
 478                                 data->error(PSI_WARNING, "Unknown value '%s' of `set` statement"
 479                                                 " for variable '$%s' of implementation '%s'",
 480                                                 set_var->name, set->arg->var->name, impl->func->name);
 481                                 return 0;
 482                         }
 483                 }
 484         }
 485         return 1;
 486 }
 487 static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
 488         size_t i, j, k;
 489         /* we can have any count of free stmts; freeing any out vars */
 490         for (i = 0; i < impl->stmts->fre.count; ++i) {
 491                 free_stmt *fre = impl->stmts->fre.list[i];
 492
 493                 for (j = 0; j < fre->vars->count; ++j) {
 494                         decl_var *free_var = fre->vars->vars[j];
 495                         int check = 0;
 496
 497                         if (!strcmp(free_var->name, impl->decl->func->var->name)) {
 498                                 continue;
 499                         }
 500                         if (impl->decl->args) for (k = 0; k < impl->decl->args->count; ++k) {
 501                                 decl_arg *free_arg = impl->decl->args->args[k];
 502
 503                                 if (!strcmp(free_var->name, free_arg->var->name)) {
 504                                         check = 1;
 505                                         free_var->arg = free_arg;
 506                                         break;
 507                                 }
 508                         }
 509
 510                         if (!check) {
 511                                 data->error(PSI_WARNING, "Unknown variable '%s' of `free` statement"
 512                                                 " of implementation '%s'",
 513                                                 free_var->name, impl->func->name);
 514                                 return 0;
 515                         }
 516                 }
 517         }
 518         return 1;
 519 }
 520 static inline int validate_impl_stmts(PSI_Data *data, impl *impl) {
 521         if (!impl->stmts) {
 522                 data->error(PSI_WARNING, "Missing body for implementation %s!",
 523                                 impl->func->name);
 524                 return 0;
 525         }
 526
 527         if (!validate_impl_ret_stmt(data, impl)) {
 528                 return 0;
 529         }
 530
 531         if (!validate_impl_let_stmts(data, impl)) {
 532                 return 0;
 533         }
 534         if (!validate_impl_set_stmts(data, impl)) {
 535                 return 0;
 536         }
 537         if (!validate_impl_free_stmts(data, impl)) {
 538                 return 0;
 539         }
 540
 541         return 1;
 542 }
 543
 544 PSI_Context *PSI_ContextInit(PSI_Context *C, PSI_ContextOps *ops, PSI_ContextErrorFunc error)
 545 {
 546         size_t i, j;
 547         PSI_Data T;
 548
 549         if (!C) {
 550                 C = malloc(sizeof(*C));
 551         }
 552         memset(C, 0, sizeof(*C));
 553
 554         C->error = error;
 555         C->ops = ops;
 556         ops->init(C);
 557
 558         /* build up predefs in a temporary PSI_Data for validation */
 559         memset(&T, 0, sizeof(T));
 560         T.error = error;
 561
 562         for (i = 0; i < psi_predef_type_count(); ++i) {
 563                 const psi_predef_type *pre = &psi_predef_types[i];
 564                 decl_type *type = init_decl_type(pre->type_tag, pre->type_name);
 565                 decl_typedef *def = init_decl_typedef(pre->alias, type);
 566
 567                 T.defs = add_decl_typedef(T.defs, def);
 568         }
 569         for (i = 0; i < psi_predef_const_count(); ++i) {
 570                 const psi_predef_const *pre = &psi_predef_consts[i];
 571                 impl_def_val *val = init_impl_def_val(pre->val_type_tag, pre->val_text);
 572                 const_type *type = init_const_type(pre->type_tag, pre->type_name);
 573                 constant *constant = init_constant(type, pre->name, val);
 574
 575                 T.consts = add_constant(T.consts, constant);
 576         }
 577         for (i = 0; i < psi_predef_struct_count(); ++i) {
 578                 const psi_predef_struct *pre = &psi_predef_structs[i];
 579                 decl_args *dargs = init_decl_args(NULL);
 580                 decl_struct *dstruct;
 581
 582                 for (j = 0; j < PSI_PREDEF_STRUCT_MEMBERS; ++j) {
 583                         const psi_predef_struct_member *member = &pre->members[j];
 584                         decl_type *type;
 585                         decl_var *dvar;
 586                         decl_arg *darg;
 587
 588                         if (!member->name) {
 589                                 break;
 590                         }
 591
 592                         type = init_decl_type(member->type_tag, member->type_name);
 593                         dvar = init_decl_var(member->name, member->pointer_level, member->array_size);
 594                         darg = init_decl_arg(type, dvar);
 595                         darg->layout = init_decl_struct_layout(member->off, member->len);
 596                         dargs = add_decl_arg(dargs, darg);
 597                 }
 598
 599                 dstruct = init_decl_struct(pre->name, dargs);
 600                 dstruct->size = pre->size;
 601                 T.structs = add_decl_struct(T.structs, dstruct);
 602         }
 603
 604         for (i = 0; i < psi_predef_type_count(); ++i) {
 605                 decl_typedef *def = T.defs->list[i];
 606
 607                 if (validate_decl_typedef(&T, def)) {
 608                         C->defs = add_decl_typedef(C->defs, def);
 609                 }
 610         }
 611
 612         for (i = 0; i < psi_predef_const_count(); ++i) {
 613                 constant *constant = T.consts->list[i];
 614
 615                 if (validate_constant(&T, constant)) {
 616                         C->consts = add_constant(C->consts, constant);
 617                 }
 618         }
 619
 620         for (i = 0; i < psi_predef_struct_count(); ++i) {
 621                 decl_struct *dstruct = T.structs->list[i];
 622
 623                 if (validate_decl_struct(&T, dstruct)) {
 624                         C->structs = add_decl_struct(C->structs, dstruct);
 625                 }
 626         }
 627
 628         C->count = 1;
 629         C->data = malloc(sizeof(*C->data));
 630         PSI_DataExchange(C->data, &T);
 631
 632         return C;
 633 }
 634
 635 int PSI_ContextValidate(PSI_Context *C, PSI_Parser *P)
 636 {
 637         PSI_Data *D;
 638         void *dlopened = NULL;
 639         size_t count = C->count++;
 640
 641         C->data = realloc(C->data, C->count * sizeof(*C->data));
 642         D = PSI_DataExchange(&C->data[count], PSI_DATA(P));
 643
 644         if (D->defs) {
 645                 size_t i;
 646
 647                 for (i = 0; i < D->defs->count; ++i) {
 648                         if (validate_decl_typedef(PSI_DATA(C), D->defs->list[i])) {
 649                                 C->defs = add_decl_typedef(C->defs, D->defs->list[i]);
 650                         }
 651                 }
 652         }
 653         if (D->structs) {
 654                 size_t i;
 655
 656                 for (i = 0; i < D->structs->count; ++i) {
 657                         if (validate_decl_struct(PSI_DATA(C), D->structs->list[i])) {
 658                                 C->structs = add_decl_struct(C->structs, D->structs->list[i]);
 659                         }
 660                 }
 661         }
 662         if (D->consts) {
 663                 size_t i;
 664
 665                 for (i = 0; i < D->consts->count; ++i) {
 666                         if (validate_constant(PSI_DATA(C), D->consts->list[i])) {
 667                                 C->consts = add_constant(C->consts, D->consts->list[i]);
 668                         }
 669                 }
 670         }
 671
 672         if (!validate_lib(D, &dlopened)) {
 673                 return 0;
 674         }
 675
 676         add_decl_lib(&C->psi.libs, dlopened);
 677
 678         if (D->decls) {
 679                 size_t i;
 680
 681                 for (i = 0; i < D->decls->count; ++i) {
 682                         if (validate_decl(PSI_DATA(C), dlopened, D->decls->list[i])) {
 683                                 C->decls = add_decl(C->decls, D->decls->list[i]);
 684                         }
 685                 }
 686         }
 687         if (D->impls) {
 688                 size_t i;
 689
 690                 for (i = 0; i < D->impls->count; ++i) {
 691                         if (validate_impl_stmts(PSI_DATA(C), D->impls->list[i])) {
 692                                 C->impls = add_impl(C->impls, D->impls->list[i]);
 693                         }
 694                 }
 695         }
 696
 697         return 1;
 698 }
 699
 700 static int psi_select_dirent(const struct dirent *entry)
 701 {
 702 #ifndef FNM_CASEFOLD
 703 #define FNM_CASEFOLD 0
 704 #endif
 705         return 0 == fnmatch("*.psi", entry->d_name, FNM_CASEFOLD);
 706 }
 707
 708 void PSI_ContextBuild(PSI_Context *C, const char *path)
 709 {
 710         int i, n;
 711         struct dirent **entries = NULL;
 712
 713         n = php_scandir(path, &entries, psi_select_dirent, alphasort);
 714
 715         if (n < 0) {
 716                 return;
 717         } else for (i = 0; i < n; ++i) {
 718                 char psi[MAXPATHLEN];
 719                 PSI_Parser P;
 720
 721                 if (MAXPATHLEN <= slprintf(psi, MAXPATHLEN, "%s/%s", path, entries[i]->d_name)) {
 722                         C->error(PSI_WARNING, "Path to PSI file too long: %s/%s",
 723                                 path, entries[i]->d_name);
 724                 }
 725                 if (!PSI_ParserInit(&P, psi, C->error, 0)) {
 726                         C->error(PSI_WARNING, "Failed to init PSI parser (%s): %s",
 727                                 psi, strerror(errno));
 728                         continue;
 729                 }
 730
 731                 while (-1 != PSI_ParserScan(&P)) {
 732                         PSI_ParserParse(&P, PSI_TokenAlloc(&P));
 733                 };
 734                 PSI_ParserParse(&P, NULL);
 735                 PSI_ContextValidate(C, &P);
 736                 PSI_ParserDtor(&P);
 737         }
 738
 739         if (PSI_ContextCompile(C) && SUCCESS != zend_register_functions(NULL, C->closures, NULL, MODULE_PERSISTENT)) {
 740                 C->error(PSI_WARNING, "Failed to register functions!");
 741         }
 742
 743         if (entries) {
 744                 for (i = 0; i < n; ++i) {
 745                         free(entries[i]);
 746                 }
 747                 free(entries);
 748         }
 749
 750 }
 751
 752 zend_function_entry *PSI_ContextCompile(PSI_Context *C)
 753 {
 754         size_t i;
 755
 756         if (C->consts) {
 757                 zend_constant zc;
 758
 759                 zc.flags = CONST_PERSISTENT|CONST_CS;
 760                 zc.module_number = EG(current_module)->module_number;
 761
 762                 for (i = 0; i < C->consts->count; ++i) {
 763                         constant *c = C->consts->list[i];
 764
 765                         zc.name = zend_string_init(c->name + (c->name[0] == '\\'), strlen(c->name) - (c->name[0] == '\\'), 1);
 766                         ZVAL_NEW_STR(&zc.value, zend_string_init(c->val->text, strlen(c->val->text), 1));
 767
 768                         switch (c->type->type) {
 769                         case PSI_T_BOOL:
 770                                 convert_to_boolean(&zc.value);
 771                                 break;
 772                         case PSI_T_INT:
 773                                 convert_to_long(&zc.value);
 774                                 break;
 775                         case PSI_T_FLOAT:
 776                                 convert_to_double(&zc.value);
 777                                 break;
 778                         }
 779                         zend_register_constant(&zc);
 780                 }
 781         }
 782
 783
 784         return C->closures = C->ops->compile(C);
 785 }
 786
 787 void PSI_ContextDtor(PSI_Context *C)
 788 {
 789         size_t i;
 790
 791         C->ops->dtor(C);
 792
 793         free_decl_libs(&C->psi.libs);
 794
 795         for (i = 0; i < C->count; ++i) {
 796                 PSI_DataDtor(&C->data[i]);
 797         }
 798
 799         free(C->data);
 800         free(C->closures);
 801
 802         if (C->consts) {
 803                 if (C->consts->list) {
 804                         free(C->consts->list);
 805                 }
 806                 free(C->consts);
 807         }
 808         if (C->defs) {
 809                 if (C->defs->list) {
 810                         free(C->defs->list);
 811                 }
 812                 free(C->defs);
 813         }
 814         if (C->structs) {
 815                 if (C->structs->list) {
 816                         free(C->structs->list);
 817                 }
 818                 free(C->structs);
 819         }
 820         if (C->decls) {
 821                 if (C->decls->list) {
 822                         free(C->decls->list);
 823                 }
 824                 free(C->decls);
 825         }
 826         if (C->impls) {
 827                 if (C->impls->list) {
 828                         free(C->impls->list);
 829                 }
 830                 free(C->impls);
 831         }
 832
 833         memset(C, 0, sizeof(*C));
 834 }
 835
 836 void PSI_ContextFree(PSI_Context **C)
 837 {
 838         if (*C) {
 839                 PSI_ContextDtor(*C);
 840                 free(*C);
 841                 *C = NULL;
 842         }
 843 }