git.m6w6.name Git - m6w6/ext-psi/blob - src/parser.h

   1 #ifndef _PSI_PARSER_H
   2 #define _PSI_PARSER_H
   3
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <stdarg.h>
   7 #include <string.h>
   8
   9 #include <Zend/zend_types.h>
  10
  11 #include "parser_proc.h"
  12
  13 #define BSIZE 256
  14
  15 #define PSI_T_POINTER PSI_T_ASTERISK
  16 #define PSI_T_LONG_DOUBLE (PSI_T_DOUBLE << 16)
  17
  18 typedef int token_t;
  19
  20 /* in php_psi.h */
  21 size_t psi_t_alignment(token_t);
  22 size_t psi_t_size(token_t);
  23
  24 typedef struct PSI_Token {
  25         token_t type;
  26         unsigned size, *line;
  27         char *text, *file;
  28         char buf[1];
  29 } PSI_Token;
  30
  31 typedef union impl_val {
  32         char cval;
  33         int8_t i8;
  34         uint8_t u8;
  35         short sval;
  36         int16_t i16;
  37         uint16_t u16;
  38         int ival;
  39         int32_t i32;
  40         uint32_t u32;
  41         long lval;
  42         int64_t i64;
  43         uint64_t u64;
  44         float fval;
  45         double dval;
  46         union {
  47                 zend_bool bval;
  48                 zend_long lval;
  49                 zend_string *str;
  50         } zend;
  51         void *ptr;
  52         uint8_t _dbg[sizeof(void *)];
  53 } impl_val;
  54
  55 typedef struct decl_type {
  56         PSI_Token *token;
  57         char *name;
  58         token_t type;
  59         struct decl_type *real;
  60         struct decl_struct *strct;
  61 } decl_type;
  62
  63 static inline decl_type *init_decl_type(token_t type, const char *name) {
  64         decl_type *t = calloc(1, sizeof(*t));
  65         t->type = type;
  66         t->name = strdup(name);
  67         return t;
  68 }
  69
  70 static inline decl_type *init_decl_type_ex(token_t type, int argc, ...) {
  71         va_list argv;
  72         char *ptr, *arg;
  73         unsigned i;
  74         size_t len, pos = 0, all = 0;
  75         decl_type *t = calloc(1, sizeof(*t));
  76
  77         va_start(argv, argc);
  78         for (i = 0; i < argc; ++i) {
  79                 arg = va_arg(argv, char *);
  80                 len = va_arg(argv, size_t);
  81
  82                 if (len) {
  83                         if (all) {
  84                                 pos = all;
  85                                 ptr = realloc(ptr, 1 + (all += len));
  86                         } else {
  87                                 ptr = malloc(1 + (all = len));
  88                         }
  89                         memcpy(ptr + pos, arg, len);
  90                 }
  91         }
  92         va_end(argv);
  93
  94         if (!all) {
  95                 ptr = calloc(1, 1);
  96         } else {
  97                 ptr[all] = 0;
  98         }
  99         t->type = type;
 100         t->name = ptr;
 101         return t;
 102 }
 103
 104 static inline decl_type *real_decl_type(decl_type *type) {
 105         while (type->real) {
 106                 type = type->real;
 107         }
 108         return type;
 109 }
 110
 111 static inline void free_decl_type(decl_type *type) {
 112         if (type->token) {
 113                 free(type->token);
 114         }
 115         free(type->name);
 116         free(type);
 117 }
 118
 119 typedef struct decl_typedef {
 120         PSI_Token *token;
 121         char *alias;
 122         decl_type *type;
 123 } decl_typedef;
 124
 125 static inline decl_typedef *init_decl_typedef(const char *name, decl_type *type) {
 126         decl_typedef *t = calloc(1, sizeof(*t));
 127         t->alias = strdup(name);
 128         t->type = type;
 129         return t;
 130 }
 131
 132 static inline void free_decl_typedef(decl_typedef *t) {
 133         if (t->token) {
 134                 free(t->token);
 135         }
 136         free(t->alias);
 137         free_decl_type(t->type);
 138         free(t);
 139 }
 140
 141 typedef struct decl_typedefs {
 142         size_t count;
 143         decl_typedef **list;
 144 } decl_typedefs;
 145
 146 static inline decl_typedefs *add_decl_typedef(decl_typedefs *defs, decl_typedef *def) {
 147         if (!defs) {
 148                 defs = calloc(1, sizeof(*defs));
 149         }
 150         defs->list = realloc(defs->list, ++defs->count * sizeof(*defs->list));
 151         defs->list[defs->count-1] = def;
 152         return defs;
 153 }
 154
 155 static void free_decl_typedefs(decl_typedefs *defs) {
 156         size_t i;
 157
 158         for (i = 0; i < defs->count; ++i) {
 159                 free_decl_typedef(defs->list[i]);
 160         }
 161         free(defs->list);
 162         free(defs);
 163 }
 164
 165 typedef struct decl_var {
 166         PSI_Token *token;
 167         char *name;
 168         unsigned pointer_level;
 169         unsigned array_size;
 170         struct decl_arg *arg;
 171 } decl_var;
 172
 173 static inline decl_var *init_decl_var(const char *name, unsigned pl, unsigned as) {
 174         decl_var *v = calloc(1, sizeof(*v));
 175         v->name = (char *) strdup((const char *) name);
 176         v->pointer_level = pl;
 177         v->array_size = as;
 178         return v;
 179 }
 180
 181 static inline void free_decl_var(decl_var *var) {
 182         if (var->token) {
 183                 free(var->token);
 184         }
 185         free(var->name);
 186         free(var);
 187 }
 188
 189 typedef struct decl_struct_layout {
 190         size_t pos;
 191         size_t len;
 192 } decl_struct_layout;
 193
 194 static inline decl_struct_layout *init_decl_struct_layout(size_t pos, size_t len) {
 195         decl_struct_layout *l = calloc(1, sizeof(*l));
 196
 197         l->pos = pos;
 198         l->len = len;
 199         return l;
 200 }
 201
 202 static inline void free_decl_struct_layout(decl_struct_layout *l) {
 203         free(l);
 204 }
 205
 206 typedef struct decl_arg {
 207         PSI_Token *token;
 208         decl_type *type;
 209         decl_var *var;
 210         decl_struct_layout *layout;
 211         struct let_stmt *let;
 212         impl_val val;
 213         void *ptr;
 214         void *mem;
 215 } decl_arg;
 216
 217 static inline decl_arg *init_decl_arg(decl_type *type, decl_var *var) {
 218         decl_arg *arg = calloc(1, sizeof(*arg));
 219         arg->token = var->token;
 220         arg->type = type;
 221         arg->var = var;
 222         var->arg = arg;
 223         arg->ptr = &arg->val;
 224         return arg;
 225 }
 226
 227 static inline void free_decl_arg(decl_arg *arg) {
 228         free_decl_type(arg->type);
 229         free_decl_var(arg->var);
 230         if (arg->layout) {
 231                 free_decl_struct_layout(arg->layout);
 232         }
 233         free(arg);
 234 }
 235
 236 typedef struct decl_vars {
 237         decl_var **vars;
 238         size_t count;
 239 } decl_vars;
 240
 241 static inline decl_vars *init_decl_vars(decl_var *var) {
 242         decl_vars *vars = calloc(1, sizeof(*vars));
 243         if (var) {
 244                 vars->count = 1;
 245                 vars->vars = calloc(1, sizeof(*vars->vars));
 246                 vars->vars[0] = var;
 247         }
 248         return vars;
 249 }
 250
 251 static inline decl_vars *add_decl_var(decl_vars *vars, decl_var *var) {
 252         vars->vars = realloc(vars->vars, ++vars->count * sizeof(*vars->vars));
 253         vars->vars[vars->count-1] = var;
 254         return vars;
 255 }
 256
 257 static inline void free_decl_vars(decl_vars *vars) {
 258         size_t i;
 259
 260         for (i = 0; i < vars->count; ++i) {
 261                 free_decl_var(vars->vars[i]);
 262         }
 263         free(vars->vars);
 264         free(vars);
 265 }
 266
 267 typedef struct decl_args {
 268         decl_arg **args;
 269         size_t count;
 270         unsigned varargs:1;
 271 } decl_args;
 272
 273 static inline decl_args *init_decl_args(decl_arg *arg) {
 274         decl_args *args = calloc(1, sizeof(*args));
 275         if (arg) {
 276                 args->count = 1;
 277                 args->args = calloc(1, sizeof(*args->args));
 278                 args->args[0] = arg;
 279         }
 280         return args;
 281 }
 282
 283 static inline decl_args *add_decl_arg(decl_args *args, decl_arg *arg) {
 284         args->args = realloc(args->args, ++args->count * sizeof(*args->args));
 285         args->args[args->count-1] = arg;
 286         return args;
 287 }
 288
 289 static inline void free_decl_args(decl_args *args) {
 290         size_t i;
 291
 292         for (i = 0; i < args->count; ++i) {
 293                 free_decl_arg(args->args[i]);
 294         }
 295         free(args->args);
 296         free(args);
 297 }
 298
 299 typedef struct decl_abi {
 300         PSI_Token *token;
 301         char *convention;
 302 } decl_abi;
 303
 304 static inline decl_abi *init_decl_abi(const char *convention) {
 305         decl_abi *abi = calloc(1, sizeof(*abi));
 306         abi->convention = strdup(convention);
 307         return abi;
 308 }
 309
 310 static inline void free_decl_abi(decl_abi *abi) {
 311         if (abi->token) {
 312                 free(abi->token);
 313         }
 314         free(abi->convention);
 315         free(abi);
 316 }
 317
 318 typedef struct decl_callinfo {
 319         void *sym;
 320         void *info;
 321         size_t argc;
 322         void **args;
 323         void *rval;
 324 } decl_callinfo;
 325
 326 typedef struct decl {
 327         decl_abi *abi;
 328         decl_arg *func;
 329         decl_args *args;
 330         struct impl *impl;
 331         decl_callinfo call;
 332 } decl;
 333
 334 static inline decl* init_decl(decl_abi *abi, decl_arg *func, decl_args *args) {
 335         decl *d = calloc(1, sizeof(*d));
 336         d->abi = abi;
 337         d->func = func;
 338         d->args = args;
 339         return d;
 340 }
 341
 342 static inline void free_decl(decl *d) {
 343         free_decl_abi(d->abi);
 344         free_decl_arg(d->func);
 345         if (d->args) {
 346                 free_decl_args(d->args);
 347         }
 348         free(d);
 349 }
 350
 351 typedef struct decls {
 352         size_t count;
 353         decl **list;
 354 } decls;
 355
 356 static inline decls *add_decl(decls *decls, decl *decl) {
 357         if (!decls) {
 358                 decls = calloc(1, sizeof(*decls));
 359         }
 360         decls->list = realloc(decls->list, ++decls->count * sizeof(*decls->list));
 361         decls->list[decls->count-1] = decl;
 362         return decls;
 363 }
 364
 365 static inline void free_decls(decls *decls) {
 366         size_t i;
 367
 368         for (i = 0; i < decls->count; ++i) {
 369                 free_decl(decls->list[i]);
 370         }
 371         free(decls->list);
 372         free(decls);
 373 }
 374
 375 typedef struct decl_struct {
 376         PSI_Token *token;
 377         char *name;
 378         decl_args *args;
 379         size_t size;
 380         struct {
 381                 void *type;
 382                 void (*dtor)(void *type);
 383         } engine;
 384 } decl_struct;
 385
 386 static inline decl_struct *init_decl_struct(const char *name, decl_args *args) {
 387         decl_struct *s = calloc(1, sizeof(*s));
 388         s->name = strdup(name);
 389         s->args = args;
 390         return s;
 391 }
 392
 393 static inline void free_decl_struct(decl_struct *s) {
 394         if (s->token) {
 395                 free(s->token);
 396         }
 397         if (s->args) {
 398                 free_decl_args(s->args);
 399         }
 400         free(s->name);
 401         free(s);
 402 }
 403
 404 typedef struct decl_structs {
 405         size_t count;
 406         decl_struct **list;
 407 } decl_structs;
 408
 409 static inline decl_structs *add_decl_struct(decl_structs *ss, decl_struct *s) {
 410         if (!ss) {
 411                 ss = calloc(1, sizeof(*ss));
 412         }
 413         ss->list = realloc(ss->list, ++ss->count * sizeof(*ss->list));
 414         ss->list[ss->count-1] = s;
 415         return ss;
 416 }
 417
 418 static inline void free_decl_structs(decl_structs *ss) {
 419         size_t i;
 420
 421         for (i = 0; i < ss->count; ++i) {
 422                 free_decl_struct(ss->list[i]);
 423         }
 424         free(ss->list);
 425         free(ss);
 426 }
 427
 428 typedef struct impl_type {
 429         char *name;
 430         token_t type;
 431 } impl_type;
 432
 433 static inline impl_type *init_impl_type(token_t type, const char *name) {
 434         impl_type *t = calloc(1, sizeof(*t));
 435
 436         t->type = type;
 437         t->name = strdup(name);
 438         return t;
 439 }
 440
 441 static inline void free_impl_type(impl_type *type) {
 442         free(type->name);
 443         free(type);
 444 }
 445
 446 typedef struct impl_var {
 447         PSI_Token *token;
 448         char *name;
 449         unsigned reference:1;
 450 } impl_var;
 451
 452 static inline impl_var *init_impl_var(const char *name, int is_reference) {
 453         impl_var *var = calloc(1, sizeof(*var));
 454         var->name = strdup(name);
 455         var->reference = is_reference;
 456         return var;
 457 }
 458
 459 static inline void free_impl_var(impl_var *var) {
 460         if (var->token) {
 461                 free(var->token);
 462         }
 463         free(var->name);
 464         free(var);
 465 }
 466
 467 typedef struct impl_def_val {
 468         token_t type;
 469         char *text;
 470 } impl_def_val;
 471
 472 static inline impl_def_val *init_impl_def_val(token_t t, const char *text) {
 473         impl_def_val *def = calloc(1, sizeof(*def));
 474         def->type = t;
 475         def->text = strdup(text);
 476         return def;
 477 }
 478
 479 static inline void free_impl_def_val(impl_def_val *def) {
 480         free(def->text);
 481         free(def);
 482 }
 483
 484 typedef struct const_type {
 485         token_t type;
 486         char *name;
 487 } const_type;
 488
 489 static inline const_type *init_const_type(token_t type, const char *name) {
 490         const_type *ct = calloc(1, sizeof(*ct));
 491         ct->type = type;
 492         ct->name = strdup(name);
 493         return ct;
 494 }
 495
 496 static inline void free_const_type(const_type *type) {
 497         free(type->name);
 498         free(type);
 499 }
 500
 501 typedef struct constant {
 502         const_type *type;
 503         char *name;
 504         impl_def_val *val;
 505 } constant;
 506
 507 static inline constant *init_constant(const_type *type, const char *name, impl_def_val *val) {
 508         constant *c = calloc(1, sizeof(*c));
 509         c->type = type;
 510         c->name = strdup(name);
 511         c->val = val;
 512         return c;
 513 }
 514
 515 static inline void free_constant(constant *constant) {
 516         free_const_type(constant->type);
 517         free(constant->name);
 518         free_impl_def_val(constant->val);
 519         free(constant);
 520 }
 521
 522 typedef struct constants {
 523         size_t count;
 524         constant **list;
 525 } constants;
 526
 527 static inline constants *add_constant(constants *constants, constant *constant) {
 528         if (!constants) {
 529                 constants = calloc(1, sizeof(*constants));
 530         }
 531         constants->list = realloc(constants->list, ++constants->count * sizeof(*constants->list));
 532         constants->list[constants->count-1] = constant;
 533         return constants;
 534 }
 535
 536 static inline void free_constants(constants *c) {
 537         size_t i;
 538
 539         for (i = 0; i < c->count; ++i) {
 540                 free_constant(c->list[i]);
 541         }
 542         free(c->list);
 543         free(c);
 544 }
 545
 546 typedef struct impl_arg {
 547         impl_type *type;
 548         impl_var *var;
 549         impl_def_val *def;
 550         impl_val val;
 551         zval *_zv;
 552 } impl_arg;
 553
 554 static inline impl_arg *init_impl_arg(impl_type *type, impl_var *var, impl_def_val *def) {
 555         impl_arg *arg = calloc(1, sizeof(*arg));
 556         arg->type = type;
 557         arg->var = var;
 558         arg->def = def;
 559         return arg;
 560 }
 561
 562 static inline void free_impl_arg(impl_arg *arg) {
 563         free_impl_type(arg->type);
 564         free_impl_var(arg->var);
 565         if (arg->def) {
 566                 free_impl_def_val(arg->def);
 567         }
 568         free(arg);
 569 }
 570
 571 typedef struct impl_vararg {
 572         impl_arg *name;
 573         struct impl_args *args;
 574         token_t *types;
 575         impl_val *values;
 576         void **free_list;
 577 } impl_vararg;
 578
 579 typedef struct impl_args {
 580         impl_arg **args;
 581         size_t count;
 582         impl_vararg vararg;
 583 } impl_args;
 584
 585 static inline impl_args *init_impl_args(impl_arg *arg) {
 586         impl_args *args = calloc(1, sizeof(*args));
 587         if (arg) {
 588                 args->count = 1;
 589                 args->args = calloc(1, sizeof(*args->args));
 590                 args->args[0] = arg;
 591         }
 592         return args;
 593 }
 594
 595 static inline impl_args *add_impl_arg(impl_args *args, impl_arg *arg) {
 596         args->args = realloc(args->args, ++args->count * sizeof(*args->args));
 597         args->args[args->count-1] = arg;
 598         return args;
 599 }
 600
 601 static inline void free_impl_args(impl_args *args) {
 602         size_t i;
 603
 604         for (i = 0; i < args->count; ++i) {
 605                 free_impl_arg(args->args[i]);
 606         }
 607         if (args->vararg.name) {
 608                 free_impl_arg(args->vararg.name);
 609         }
 610         free(args->args);
 611         free(args);
 612 }
 613
 614 typedef struct impl_func {
 615         PSI_Token *token;
 616         char *name;
 617         impl_args *args;
 618         impl_type *return_type;
 619         unsigned return_reference:1;
 620 } impl_func;
 621
 622 static inline impl_func *init_impl_func(char *name, impl_args *args, impl_type *type, int ret_reference) {
 623         impl_func *func = calloc(1, sizeof(*func));
 624         func->name = strdup(name);
 625         func->args = args ? args : init_impl_args(NULL);
 626         func->return_type = type;
 627         func->return_reference = ret_reference;
 628         return func;
 629 }
 630
 631 static inline void free_impl_func(impl_func *f) {
 632         if (f->token) {
 633                 free(f->token);
 634         }
 635         free_impl_type(f->return_type);
 636         free_impl_args(f->args);
 637         free(f->name);
 638         free(f);
 639 }
 640
 641 typedef struct num_exp {
 642         PSI_Token *token;
 643         token_t t;
 644         union {
 645                 char *numb;
 646                 constant *cnst;
 647                 decl_var *dvar;
 648         } u;
 649         token_t operator;
 650         int (*calculator)(int t1, impl_val *v1, int t2, impl_val *v2, impl_val *res);
 651         struct num_exp *operand;
 652 } num_exp;
 653
 654 static inline num_exp *init_num_exp(token_t t, void *num) {
 655         num_exp *exp = calloc(1, sizeof(*exp));
 656         switch (exp->t = t) {
 657         case PSI_T_NUMBER:
 658         case PSI_T_NSNAME:
 659                 exp->u.numb = strdup(num);
 660                 break;
 661         case PSI_T_NAME:
 662                 exp->u.dvar = num;
 663                 break;
 664         EMPTY_SWITCH_DEFAULT_CASE();
 665         }
 666         return exp;
 667 }
 668
 669 static inline void free_num_exp(num_exp *exp) {
 670         if (exp->token) {
 671                 free(exp->token);
 672         }
 673         switch (exp->t) {
 674         case PSI_T_NUMBER:
 675                 free(exp->u.numb);
 676                 break;
 677         case PSI_T_NSNAME:
 678                 break;
 679         case PSI_T_NAME:
 680                 free_decl_var(exp->u.dvar);
 681                 break;
 682         EMPTY_SWITCH_DEFAULT_CASE();
 683         }
 684         if (exp->operand) {
 685                 free_num_exp(exp->operand);
 686         }
 687         free(exp);
 688 }
 689
 690 typedef struct let_calloc {
 691         num_exp *nmemb;
 692         num_exp *size;
 693 } let_calloc;
 694
 695 static inline let_calloc *init_let_calloc(num_exp *nmemb, num_exp *size) {
 696         let_calloc *alloc = calloc(1, sizeof(*alloc));
 697         alloc->nmemb = nmemb;
 698         alloc->size = size;
 699         return alloc;
 700 }
 701
 702 static inline void free_let_calloc(let_calloc *alloc) {
 703         free_num_exp(alloc->nmemb);
 704         free_num_exp(alloc->size);
 705         free(alloc);
 706 }
 707
 708 typedef struct let_func {
 709         token_t type;
 710         char *name;
 711         impl_var *var;
 712         impl_arg *arg;
 713 } let_func;
 714
 715 static inline let_func *init_let_func(token_t type, const char *name, impl_var *var) {
 716         let_func *func = calloc(1, sizeof(*func));
 717         func->type = type;
 718         func->name = strdup(name);
 719         func->var = var;
 720         return func;
 721 }
 722
 723 static inline void free_let_func(let_func *func) {
 724         free_impl_var(func->var);
 725         free(func->name);
 726         free(func);
 727 }
 728
 729 #define PSI_LET_REFERENCE 0x1;
 730 typedef struct let_val {
 731         enum let_val_kind {
 732                 PSI_LET_NULL,
 733                 PSI_LET_NUMEXP,
 734                 PSI_LET_CALLOC,
 735                 PSI_LET_FUNC,
 736                 PSI_LET_TMP,
 737         } kind;
 738         union {
 739                 num_exp *num;
 740                 let_calloc *alloc;
 741                 let_func *func;
 742                 decl_var *var;
 743         } data;
 744         union {
 745                 struct {
 746                         unsigned is_reference:1;
 747                 } one;
 748                 unsigned all;
 749         } flags;
 750 } let_val;
 751
 752 static inline let_val *init_let_val(enum let_val_kind kind, void *data) {
 753         let_val *let = calloc(1, sizeof(*let));
 754         switch (let->kind = kind) {
 755         case PSI_LET_NULL:
 756                 break;
 757         case PSI_LET_NUMEXP:
 758                 let->data.num = data;
 759                 break;
 760         case PSI_LET_CALLOC:
 761                 let->data.alloc = data;
 762                 break;
 763         case PSI_LET_FUNC:
 764                 let->data.func = data;
 765                 break;
 766         case PSI_LET_TMP:
 767                 let->data.var = data;
 768                 break;
 769         EMPTY_SWITCH_DEFAULT_CASE();
 770         }
 771         return let;
 772 }
 773
 774 static inline void free_let_val(let_val *let) {
 775         switch (let->kind) {
 776         case PSI_LET_NULL:
 777                 break;
 778         case PSI_LET_NUMEXP:
 779                 free_num_exp(let->data.num);
 780                 break;
 781         case PSI_LET_CALLOC:
 782                 free_let_calloc(let->data.alloc);
 783                 break;
 784         case PSI_LET_FUNC:
 785                 free_let_func(let->data.func);
 786                 break;
 787         case PSI_LET_TMP:
 788                 free_decl_var(let->data.var);
 789                 break;
 790         EMPTY_SWITCH_DEFAULT_CASE();
 791         }
 792         free(let);
 793 }
 794
 795 typedef struct let_stmt {
 796         decl_var *var;
 797         let_val *val;
 798
 799         void *ptr;
 800 } let_stmt;
 801
 802 static inline let_stmt *init_let_stmt(decl_var *var, let_val *val) {
 803         let_stmt *let = calloc(1, sizeof(*let));
 804         let->var = var;
 805         let->val = val;
 806         return let;
 807 }
 808
 809 static inline void free_let_stmt(let_stmt *stmt) {
 810         if (stmt->val) {
 811                 if (stmt->val->kind == PSI_LET_TMP && stmt->var->arg) {
 812                         free_decl_arg(stmt->var->arg);
 813                 }
 814                 free_let_val(stmt->val);
 815         }
 816         free_decl_var(stmt->var);
 817         free(stmt);
 818 }
 819
 820 struct set_value;
 821
 822 typedef struct set_func {
 823         PSI_Token *token;
 824         token_t type;
 825         char *name;
 826         void (*handler)(zval *, struct set_value *set, impl_val *ret_val);
 827 } set_func;
 828
 829 static inline set_func *init_set_func(token_t type, const char *name) {
 830         set_func *func = calloc(1, sizeof(*func));
 831         func->type = type;
 832         func->name = strdup(name);
 833         return func;
 834 }
 835
 836 static inline void free_set_func(set_func *func) {
 837         if (func->token) {
 838                 free(func->token);
 839         }
 840         free(func->name);
 841         free(func);
 842 }
 843
 844 typedef struct set_value {
 845         set_func *func;
 846         decl_vars *vars;
 847         num_exp *num;
 848         struct {
 849                 struct set_value *set;
 850                 impl_val *val;
 851         } outer;
 852         struct set_value **inner;
 853         size_t count;
 854 } set_value;
 855
 856 static inline set_value *init_set_value(set_func *func, decl_vars *vars) {
 857         set_value *val = calloc(1, sizeof(*val));
 858         val->func = func;
 859         val->vars = vars;
 860         return val;
 861 }
 862 static inline set_value *add_inner_set_value(set_value *val, set_value *inner) {
 863         val->inner = realloc(val->inner, ++val->count * sizeof(*val->inner));
 864         val->inner[val->count-1] = inner;
 865         inner->outer.set = val;
 866         return val;
 867 }
 868
 869 static inline void free_set_value(set_value *val) {
 870         if (val->func) {
 871                 free_set_func(val->func);
 872         }
 873         if (val->vars) {
 874                 free_decl_vars(val->vars);
 875         }
 876         if (val->inner && (!val->outer.set || val->outer.set->inner != val->inner)) {
 877                 size_t i;
 878                 for (i = 0; i < val->count; ++i) {
 879                         free_set_value(val->inner[i]);
 880                 }
 881                 free(val->inner);
 882         }
 883         if (val->num) {
 884                 free_num_exp(val->num);
 885         }
 886         free(val);
 887 }
 888
 889 typedef struct set_stmt {
 890         impl_var *var;
 891         set_value *val;
 892         impl_arg *arg;
 893 } set_stmt;
 894
 895 static inline set_stmt *init_set_stmt(impl_var *var, set_value *val) {
 896         set_stmt *set = calloc(1, sizeof(*set));
 897         set->var = var;
 898         set->val = val;
 899         return set;
 900 }
 901
 902 static inline void free_set_stmt(set_stmt *set) {
 903         free_impl_var(set->var);
 904         free_set_value(set->val);
 905         free(set);
 906 }
 907
 908 typedef struct return_stmt {
 909         PSI_Token *token;
 910         set_value *set;
 911         decl_arg *decl;
 912 } return_stmt;
 913
 914 static inline return_stmt *init_return_stmt(set_value *val) {
 915         return_stmt *ret = calloc(1, sizeof(*ret));
 916         ret->set = val;
 917         return ret;
 918 }
 919
 920 static inline void free_return_stmt(return_stmt *ret) {
 921         if (ret->token) {
 922                 free(ret->token);
 923         }
 924         free_set_value(ret->set);
 925         free(ret);
 926 }
 927
 928 typedef struct free_call {
 929         PSI_Token *token;
 930         char *func;
 931         decl_vars *vars;
 932         decl *decl;
 933 } free_call;
 934
 935 static inline free_call *init_free_call(const char *func, decl_vars *vars) {
 936         free_call *f = calloc(1, sizeof(*f));
 937         f->func = strdup(func);
 938         f->vars = vars;
 939         return f;
 940 }
 941
 942 static inline void free_free_call(free_call *f) {
 943         if (f->token) {
 944                 free(f->token);
 945         }
 946         free(f->func);
 947         free_decl_vars(f->vars);
 948         free(f);
 949 }
 950
 951 typedef struct free_calls {
 952         free_call **list;
 953         size_t count;
 954 } free_calls;
 955
 956 static inline free_calls *init_free_calls(free_call *f) {
 957         free_calls *fcs = calloc(1, sizeof(*fcs));
 958         if (f) {
 959                 fcs->count = 1;
 960                 fcs->list = calloc(1, sizeof(*fcs->list));
 961                 fcs->list[0] = f;
 962         }
 963         return fcs;
 964 }
 965
 966 static inline void free_free_calls(free_calls *fcs) {
 967         size_t i;
 968
 969         for (i = 0; i < fcs->count; ++i) {
 970                 free_free_call(fcs->list[i]);
 971         }
 972         free(fcs->list);
 973         free(fcs);
 974 }
 975
 976 static inline free_calls *add_free_call(free_calls *fcs, free_call *f) {
 977         fcs->list = realloc(fcs->list, ++fcs->count * sizeof(*fcs->list));
 978         fcs->list[fcs->count-1] = f;
 979         return fcs;
 980 }
 981
 982 typedef struct free_stmt {
 983         free_calls *calls;
 984 } free_stmt;
 985
 986 static inline free_stmt *init_free_stmt(free_calls *calls) {
 987         free_stmt *f = calloc(1, sizeof(*f));
 988         f->calls = calls;
 989         return f;
 990 }
 991
 992 static inline void free_free_stmt(free_stmt *f) {
 993         free_free_calls(f->calls);
 994         free(f);
 995 }
 996
 997 typedef struct impl_stmt {
 998         token_t type;
 999         union {
1000                 let_stmt *let;
1001                 set_stmt *set;
1002                 return_stmt *ret;
1003                 free_stmt *fre;
1004                 void *ptr;
1005         } s;
1006 } impl_stmt;
1007
1008 static inline impl_stmt *init_impl_stmt(token_t type, void *ptr) {
1009         impl_stmt *stmt = calloc(1, sizeof(*stmt));
1010         stmt->type = type;
1011         stmt->s.ptr = ptr;
1012         return stmt;
1013 }
1014
1015 static inline void free_impl_stmt(impl_stmt *stmt) {
1016         switch (stmt->type) {
1017         case PSI_T_LET:
1018                 free_let_stmt(stmt->s.let);
1019                 break;
1020         case PSI_T_SET:
1021                 free_set_stmt(stmt->s.set);
1022                 break;
1023         case PSI_T_RETURN:
1024                 free_return_stmt(stmt->s.ret);
1025                 break;
1026         case PSI_T_FREE:
1027                 free_free_stmt(stmt->s.fre);
1028                 break;
1029         }
1030         free(stmt);
1031 }
1032
1033 typedef struct impl_stmts {
1034         struct {
1035                 return_stmt **list;
1036                 size_t count;
1037         } ret;
1038         struct {
1039                 let_stmt **list;
1040                 size_t count;
1041         } let;
1042         struct {
1043                 set_stmt **list;
1044                 size_t count;
1045         } set;
1046         struct {
1047                 free_stmt **list;
1048                 size_t count;
1049         } fre;
1050 } impl_stmts;
1051
1052 static inline void *add_impl_stmt_ex(void *list, size_t count, void *stmt) {
1053         list = realloc(list, count * sizeof(list));
1054         ((void **)list)[count-1] = stmt;
1055         return list;
1056 }
1057
1058 static inline impl_stmts *add_impl_stmt(impl_stmts *stmts, impl_stmt *stmt) {
1059         switch (stmt->type) {
1060         case PSI_T_RETURN:
1061                 stmts->ret.list = add_impl_stmt_ex(stmts->ret.list, ++stmts->ret.count, stmt->s.ret);
1062                 break;
1063         case PSI_T_LET:
1064                 stmts->let.list = add_impl_stmt_ex(stmts->let.list, ++stmts->let.count, stmt->s.let);
1065                 break;
1066         case PSI_T_SET:
1067                 stmts->set.list = add_impl_stmt_ex(stmts->set.list, ++stmts->set.count, stmt->s.set);
1068                 break;
1069         case PSI_T_FREE:
1070                 stmts->fre.list = add_impl_stmt_ex(stmts->fre.list, ++stmts->fre.count, stmt->s.fre);
1071                 break;
1072         }
1073         free(stmt);
1074         return stmts;
1075 }
1076
1077 static inline impl_stmts *init_impl_stmts(impl_stmt *stmt) {
1078         impl_stmts *stmts = calloc(1, sizeof(*stmts));
1079         return add_impl_stmt(stmts, stmt);
1080 }
1081
1082 static inline void free_impl_stmts(impl_stmts *stmts) {
1083         size_t i;
1084
1085         for (i = 0; i < stmts->let.count; ++i) {
1086                 free_let_stmt(stmts->let.list[i]);
1087         }
1088         free(stmts->let.list);
1089         for (i = 0; i < stmts->ret.count; ++i) {
1090                 free_return_stmt(stmts->ret.list[i]);
1091         }
1092         free(stmts->ret.list);
1093         for (i = 0; i < stmts->set.count; ++i) {
1094                 free_set_stmt(stmts->set.list[i]);
1095         }
1096         free(stmts->set.list);
1097         for (i = 0; i < stmts->fre.count; ++i) {
1098                 free_free_stmt(stmts->fre.list[i]);
1099         }
1100         free(stmts->fre.list);
1101         free(stmts);
1102 }
1103
1104 typedef struct impl {
1105         impl_func *func;
1106         impl_stmts *stmts;
1107         decl *decl;
1108 } impl;
1109
1110 static inline impl *init_impl(impl_func *func, impl_stmts *stmts) {
1111         impl *i = calloc(1, sizeof(*i));
1112         i->func = func;
1113         i->stmts = stmts;
1114         return i;
1115 }
1116
1117 static inline void free_impl(impl *impl) {
1118         free_impl_func(impl->func);
1119         free_impl_stmts(impl->stmts);
1120         free(impl);
1121 }
1122
1123 typedef struct impls {
1124         size_t count;
1125         impl **list;
1126 } impls;
1127
1128 static inline impls *add_impl(impls *impls, impl *impl) {
1129         if (!impls) {
1130                 impls = calloc(1, sizeof(*impls));
1131         }
1132         impls->list = realloc(impls->list, ++impls->count * sizeof(*impls->list));
1133         impls->list[impls->count-1] = impl;
1134         return impls;
1135 }
1136
1137 static void free_impls(impls *impls) {
1138         size_t i;
1139
1140         for (i = 0; i < impls->count; ++i) {
1141                 free_impl(impls->list[i]);
1142         }
1143         free(impls->list);
1144         free(impls);
1145 }
1146
1147 typedef struct decl_file {
1148         char *ln;
1149         char *fn;
1150 } decl_file;
1151
1152 static inline void free_decl_file(decl_file *file) {
1153         if (file->ln) {
1154                 free(file->ln);
1155         }
1156         if (file->fn) {
1157                 free(file->fn);
1158         }
1159         memset(file, 0, sizeof(*file));
1160 }
1161
1162 typedef struct decl_libs {
1163         void **dl;
1164         size_t count;
1165 } decl_libs;
1166
1167 static inline void free_decl_libs(decl_libs *libs) {
1168         if (libs->dl) {
1169                 size_t i;
1170                 for (i = 0; i < libs->count; ++i) {
1171                         if (libs->dl[i]) {
1172                                 dlclose(libs->dl[i]);
1173                         }
1174                 }
1175                 free(libs->dl);
1176         }
1177         memset(libs, 0, sizeof(*libs));
1178 }
1179
1180 static inline void add_decl_lib(decl_libs *libs, void *dlopened) {
1181         libs->dl = realloc(libs->dl, ++libs->count * sizeof(*libs->dl));
1182         libs->dl[libs->count-1] = dlopened;
1183 }
1184
1185 static inline impl_val *deref_impl_val(impl_val *ret_val, decl_var *var) {
1186         unsigned i;
1187
1188         ZEND_ASSERT(var->arg->var != var);
1189 #if 0
1190         fprintf(stderr, "deref: %s pl=%u:%u as=%u:%u %p\n",
1191                         var->name, var->pointer_level, var->arg->var->pointer_level,
1192                         var->array_size, var->arg->var->array_size, ret_val);
1193 #endif
1194         for (i = 0; i < var->pointer_level; ++i) {
1195 #if 0
1196                 fprintf(stderr, "-- %p %p %p\n", ret_val, *(void**)ret_val, ret_val->ptr);
1197 #endif
1198                 ret_val = *(void **) ret_val;
1199         }
1200         return ret_val;
1201 }
1202
1203 static inline impl_val *enref_impl_val(void *ptr, decl_var *var) {
1204         impl_val *val, *val_ptr;
1205         unsigned i;
1206
1207         ZEND_ASSERT(var->arg->var == var);
1208 #if 0
1209         fprintf(stderr, "enref: %s pl=%u:%u as=%u:%u\n",
1210                         var->name, var->pointer_level, var->arg->var->pointer_level,
1211                         var->array_size, var->arg->var->array_size);
1212 #endif
1213         if (!var->pointer_level && real_decl_type(var->arg->type)->type != PSI_T_STRUCT) {
1214                 return ptr;
1215         }
1216
1217         val = val_ptr = calloc(var->pointer_level + 1, sizeof(void *));
1218         for (i = !var->arg->var->array_size; i < var->pointer_level; ++i) {
1219 #if 0
1220                 fprintf(stderr, "++\n");
1221 #endif
1222                 val_ptr->ptr = (void **) val_ptr + 1;
1223                 val_ptr = val_ptr->ptr;
1224         }
1225         val_ptr->ptr = ptr;
1226         return val;
1227 }
1228
1229 static inline impl_val *struct_member_ref(decl_arg *set_arg, impl_val *struct_ptr, impl_val **to_free) {
1230         void *ptr = (char *) struct_ptr + set_arg->layout->pos;
1231 #if 0
1232         fprintf(stderr, "struct member %s: %p\n", set_arg->var->name, ptr);
1233 #endif
1234         return ptr;
1235 }
1236
1237
1238 #define PSI_ERROR 16
1239 #define PSI_WARNING 32
1240 typedef void (*psi_error_cb)(PSI_Token *token, int type, const char *msg, ...);
1241
1242 #define PSI_DATA(D) ((PSI_Data *) (D))
1243 #define PSI_DATA_MEMBERS \
1244         constants *consts; \
1245         decl_typedefs *defs; \
1246         decl_structs *structs; \
1247         decls *decls; \
1248         impls *impls; \
1249         union { \
1250                 decl_file file; \
1251                 decl_libs libs; \
1252         } psi; \
1253         psi_error_cb error
1254 typedef struct PSI_Data {
1255         PSI_DATA_MEMBERS;
1256 } PSI_Data;
1257
1258 static inline PSI_Data *PSI_DataExchange(PSI_Data *dest, PSI_Data *src) {
1259         if (!dest) {
1260                 dest = malloc(sizeof(*dest));
1261         }
1262         memcpy(dest, src, sizeof(*dest));
1263         memset(src, 0, sizeof(*src));
1264         return dest;
1265 }
1266
1267 static inline void PSI_DataDtor(PSI_Data *data) {
1268         if (data->consts) {
1269                 free_constants(data->consts);
1270         }
1271         if (data->defs) {
1272                 free_decl_typedefs(data->defs);
1273         }
1274         if (data->structs) {
1275                 free_decl_structs(data->structs);
1276         }
1277         if (data->decls) {
1278                 free_decls(data->decls);
1279         }
1280         if (data->impls) {
1281                 free_impls(data->impls);
1282         }
1283         free_decl_file(&data->psi.file);
1284 }
1285
1286 typedef struct PSI_Parser {
1287         PSI_DATA_MEMBERS;
1288         FILE *fp;
1289         token_t num;
1290         void *proc;
1291         unsigned flags, errors, line, col;
1292         char *cur, *tok, *lim, *eof, *ctx, *mrk, buf[BSIZE];
1293 } PSI_Parser;
1294
1295 static inline size_t PSI_TokenAllocSize(size_t token_len, size_t fname_len) {
1296         return sizeof(PSI_Token) + token_len + fname_len + sizeof(unsigned) + 2;
1297 }
1298
1299 static inline PSI_Token *PSI_TokenAlloc(PSI_Parser *P) {
1300         PSI_Token *T;
1301         size_t token_len, fname_len;
1302         token_t token_typ;
1303
1304         if (P->cur < P->tok) {
1305                 return NULL;
1306         }
1307
1308         token_typ = P->num;
1309         token_len = P->cur - P->tok;
1310         fname_len = strlen(P->psi.file.fn);
1311
1312         T = calloc(1, PSI_TokenAllocSize(token_len, fname_len));
1313         T->type = token_typ;
1314         T->size = token_len;
1315         T->text = &T->buf[0];
1316         T->file = &T->buf[token_len + 1];
1317         T->line = (void *) &T->buf[fname_len + token_len + 2];
1318
1319         memcpy(T->text, P->tok, token_len);
1320         memcpy(T->file, P->psi.file.fn, fname_len);
1321         memcpy(T->line, &P->line, sizeof(unsigned));
1322
1323         return T;
1324 }
1325
1326 static inline PSI_Token *PSI_TokenCopy(PSI_Token *src) {
1327         size_t strct_len = PSI_TokenAllocSize(src->size, strlen(src->file));
1328         PSI_Token *ptr = malloc(strct_len);
1329
1330         memcpy(ptr, src, strct_len);
1331
1332         ptr->text = &ptr->buf[0];
1333         ptr->file = &ptr->buf[ptr->size + 1];
1334
1335         return ptr;
1336 }
1337
1338 static inline PSI_Token *PSI_TokenCat(unsigned argc, ...) {
1339         va_list argv;
1340         unsigned i;
1341         PSI_Token *T = NULL;
1342
1343         va_start(argv, argc);
1344         for (i = 0; i < argc; ++i) {
1345                 PSI_Token *arg = va_arg(argv, PSI_Token *);
1346
1347                 if (T) {
1348                         size_t fname_len = strlen(T->file);
1349
1350                         T = realloc(T, PSI_TokenAllocSize(T->size + arg->size, fname_len));
1351                         memmove(&T->buf[T->size + 1], T->file, fname_len + 1);
1352                         memcpy(T->file - 1, arg->text, arg->size + 1);
1353                         T->file = &T->buf[T->size + 1];
1354                 } else {
1355                         T = PSI_TokenCopy(arg);
1356                         T->type = PSI_T_NAME;
1357                 }
1358         }
1359         va_end(argv);
1360
1361         return T;
1362 }
1363
1364 static inline const char *PSI_TokenLocation(PSI_Token *t) {
1365         return t ? t->file : "<builtin>:0:0";
1366 }
1367
1368 #define PSI_PARSER_DEBUG 0x1
1369
1370 PSI_Parser *PSI_ParserInit(PSI_Parser *P, const char *filename, psi_error_cb error, unsigned flags);
1371 void PSI_ParserSyntaxError(PSI_Parser *P, const char *fn, size_t ln, const char *msg, ...);
1372 size_t PSI_ParserFill(PSI_Parser *P, size_t n);
1373 token_t PSI_ParserScan(PSI_Parser *P);
1374 void PSI_ParserParse(PSI_Parser *P, PSI_Token *src);
1375 void PSI_ParserDtor(PSI_Parser *P);
1376 void PSI_ParserFree(PSI_Parser **P);
1377
1378 #endif