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