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