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