parser.c

   1 #include <stdio.h>
   2 #include <stdarg.h>
   3
   4 #include "gmqcc.h"
   5 #include "lexer.h"
   6
   7 typedef struct {
   8     lex_file *lex;
   9     int      tok;
  10
  11     MEM_VECTOR_MAKE(ast_value*, globals);
  12     MEM_VECTOR_MAKE(ast_function*, functions);
  13     MEM_VECTOR_MAKE(ast_value*, imm_float);
  14
  15     ast_function *function;
  16     MEM_VECTOR_MAKE(ast_value*, locals);
  17     size_t blocklocal;
  18 } parser_t;
  19
  20 MEM_VEC_FUNCTIONS(parser_t, ast_value*, globals)
  21 MEM_VEC_FUNCTIONS(parser_t, ast_value*, imm_float)
  22 MEM_VEC_FUNCTIONS(parser_t, ast_value*, locals)
  23 MEM_VEC_FUNCTIONS(parser_t, ast_function*, functions)
  24
  25 void parseerror(parser_t *parser, const char *fmt, ...)
  26 {
  27         va_list ap;
  28
  29     if (parser)
  30             printf("error %s:%lu: ", parser->lex->tok->ctx.file, (unsigned long)parser->lex->tok->ctx.line);
  31         else
  32             printf("error: ");
  33
  34         va_start(ap, fmt);
  35         vprintf(fmt, ap);
  36         va_end(ap);
  37
  38         printf("\n");
  39 }
  40
  41 bool parser_next(parser_t *parser)
  42 {
  43     /* lex_do kills the previous token */
  44     parser->tok = lex_do(parser->lex);
  45     if (parser->tok == TOKEN_EOF || parser->tok >= TOKEN_ERROR)
  46         return false;
  47     return true;
  48 }
  49
  50 /* lift a token out of the parser so it's not destroyed by parser_next */
  51 token *parser_lift(parser_t *parser)
  52 {
  53     token *tok = parser->lex->tok;
  54     parser->lex->tok = NULL;
  55     return tok;
  56 }
  57
  58 #define parser_tokval(p) (p->lex->tok->value)
  59 #define parser_token(p)  (p->lex->tok)
  60 #define parser_ctx(p)    (p->lex->tok->ctx)
  61
  62 ast_value* parser_const_float(parser_t *parser, double d)
  63 {
  64     size_t i;
  65     ast_value *out;
  66     for (i = 0; i < parser->imm_float_count; ++i) {
  67         if (parser->imm_float[i]->constval.vfloat == d)
  68             return parser->imm_float[i];
  69     }
  70     out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_FLOAT);
  71     out->isconst = true;
  72     out->constval.vfloat = d;
  73     if (!parser_t_imm_float_add(parser, out)) {
  74         ast_value_delete(out);
  75         return NULL;
  76     }
  77     return out;
  78 }
  79
  80 ast_value* parser_find_global(parser_t *parser, const char *name)
  81 {
  82     size_t i;
  83     for (i = 0; i < parser->globals_count; ++i) {
  84         if (!strcmp(parser->globals[i]->name, name))
  85             return parser->globals[i];
  86     }
  87     return NULL;
  88 }
  89
  90 ast_value* parser_find_local(parser_t *parser, const char *name, size_t upto)
  91 {
  92     size_t i;
  93     ast_value *fun;
  94     for (i = parser->locals_count; i > upto;) {
  95         --i;
  96         if (!strcmp(parser->locals[i]->name, name))
  97             return parser->locals[i];
  98     }
  99     fun = parser->function->vtype;
 100     for (i = 0; i < fun->params_count; ++i) {
 101         if (!strcmp(fun->params[i]->name, name))
 102             return fun->params[i];
 103     }
 104     return NULL;
 105 }
 106
 107 ast_value* parser_find_var(parser_t *parser, const char *name)
 108 {
 109     ast_value *v;
 110     v         = parser_find_local(parser, name, 0);
 111     if (!v) v = parser_find_global(parser, name);
 112     return v;
 113 }
 114
 115 typedef struct {
 116     MEM_VECTOR_MAKE(ast_value*, p);
 117 } paramlist_t;
 118 MEM_VEC_FUNCTIONS(paramlist_t, ast_value*, p)
 119
 120 static ast_value *parser_parse_type(parser_t *parser, int basetype, bool *isfunc)
 121 {
 122     paramlist_t params;
 123     ast_value *var;
 124     lex_ctx   ctx = parser_ctx(parser);
 125     int vtype = basetype;
 126     int temptype;
 127     size_t i;
 128
 129     MEM_VECTOR_INIT(&params, p);
 130
 131     *isfunc = false;
 132
 133     if (parser->tok == '(') {
 134         *isfunc = true;
 135         while (true) {
 136             ast_value *param;
 137             bool dummy;
 138
 139             if (!parser_next(parser))
 140                 goto on_error;
 141
 142             if (parser->tok == ')')
 143                 break;
 144
 145             temptype = parser_token(parser)->constval.t;
 146             if (!parser_next(parser))
 147                 goto on_error;
 148
 149             param = parser_parse_type(parser, temptype, &dummy);
 150             (void)dummy;
 151
 152             if (!param)
 153                 goto on_error;
 154
 155             if (parser->tok == TOKEN_IDENT) {
 156                 /* named parameter */
 157                 if (!ast_value_set_name(param, parser_tokval(parser)))
 158                     goto on_error;
 159                 if (!parser_next(parser))
 160                     goto on_error;
 161             }
 162
 163             if (!paramlist_t_p_add(&params, param)) {
 164                 parseerror(parser, "Out of memory while parsing typename");
 165                 goto on_error;
 166             }
 167
 168             if (parser->tok == ',')
 169                 continue;
 170             if (parser->tok == ')')
 171                 break;
 172             parseerror(parser, "Unexpected token");
 173             goto on_error;
 174         }
 175         if (!parser_next(parser))
 176             goto on_error;
 177     }
 178
 179     var = ast_value_new(ctx, "<unnamed>", vtype);
 180     if (!var)
 181         goto on_error;
 182     MEM_VECTOR_MOVE(&params, p, var, params);
 183     return var;
 184 on_error:
 185     for (i = 0; i < params.p_count; ++i)
 186         ast_value_delete(params.p[i]);
 187     MEM_VECTOR_CLEAR(&params, p);
 188     return NULL;
 189 }
 190
 191 typedef struct
 192 {
 193     size_t etype; /* 0 = expression, others are operators */
 194     int             paren;
 195     ast_expression *out;
 196     ast_value      *value; /* need to know if we can assign */
 197     lex_ctx ctx;
 198 } sy_elem;
 199 typedef struct
 200 {
 201     MEM_VECTOR_MAKE(sy_elem, out);
 202     MEM_VECTOR_MAKE(sy_elem, ops);
 203 } shunt;
 204 MEM_VEC_FUNCTIONS(shunt, sy_elem, out)
 205 MEM_VEC_FUNCTIONS(shunt, sy_elem, ops)
 206
 207 static sy_elem syexp(lex_ctx ctx, ast_expression *v) {
 208     sy_elem e;
 209     e.etype = 0;
 210     e.out   = v;
 211     e.value = NULL;
 212     e.ctx   = ctx;
 213     e.paren = 0;
 214     return e;
 215 }
 216 static sy_elem syval(lex_ctx ctx, ast_value *v) {
 217     sy_elem e;
 218     e.etype = 0;
 219     e.out   = (ast_expression*)v;
 220     e.value = v;
 221     e.ctx   = ctx;
 222     e.paren = 0;
 223     return e;
 224 }
 225
 226 static sy_elem syop(lex_ctx ctx, const oper_info *op) {
 227     sy_elem e;
 228     e.etype = 1 + (op - operators);
 229     e.out   = NULL;
 230     e.value = NULL;
 231     e.ctx   = ctx;
 232     e.paren = 0;
 233     return e;
 234 }
 235
 236 static sy_elem syparen(lex_ctx ctx, int p) {
 237     sy_elem e;
 238     e.etype = 0;
 239     e.out   = NULL;
 240     e.value = NULL;
 241     e.ctx   = ctx;
 242     e.paren = p;
 243     return e;
 244 }
 245
 246 static bool parser_sy_pop(parser_t *parser, shunt *sy)
 247 {
 248     const oper_info *op;
 249     lex_ctx ctx;
 250     ast_expression *out = NULL;
 251     ast_expression *exprs[3];
 252     ast_value      *vars[3];
 253     size_t i;
 254
 255     if (!sy->ops_count) {
 256         parseerror(parser, "internal error: missing operator");
 257         return false;
 258     }
 259
 260     if (sy->ops[sy->ops_count-1].paren) {
 261         parseerror(parser, "unmatched parenthesis");
 262         return false;
 263     }
 264
 265     op = &operators[sy->ops[sy->ops_count-1].etype - 1];
 266     ctx = sy->ops[sy->ops_count-1].ctx;
 267
 268     if (sy->out_count < op->operands) {
 269         parseerror(parser, "internal error: not enough operands: %i", sy->out_count);
 270         return false;
 271     }
 272
 273     sy->ops_count--;
 274
 275     sy->out_count -= op->operands;
 276     for (i = 0; i < op->operands; ++i) {
 277         exprs[i] = sy->out[sy->out_count+i].out;
 278         vars[i]  = sy->out[sy->out_count+i].value;
 279     }
 280
 281     switch (op->id)
 282     {
 283         default:
 284             parseerror(parser, "internal error: unhandled operand");
 285             return false;
 286
 287         case opid1('+'):
 288             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 289                 parseerror(parser, "Cannot add type %s and %s",
 290                            type_name[exprs[0]->expression.vtype],
 291                            type_name[exprs[1]->expression.vtype]);
 292                 return false;
 293             }
 294             switch (exprs[0]->expression.vtype) {
 295                 case TYPE_FLOAT:
 296                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
 297                     break;
 298                 case TYPE_VECTOR:
 299                     out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
 300                     break;
 301                 default:
 302                     parseerror(parser, "Cannot add type %s and %s",
 303                                type_name[exprs[0]->expression.vtype],
 304                                type_name[exprs[1]->expression.vtype]);
 305                     return false;
 306             };
 307             break;
 308         case opid1('-'):
 309             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
 310                 parseerror(parser, "Cannot subtract type %s from %s",
 311                            type_name[exprs[1]->expression.vtype],
 312                            type_name[exprs[0]->expression.vtype]);
 313                 return false;
 314             }
 315             switch (exprs[0]->expression.vtype) {
 316                 case TYPE_FLOAT:
 317                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
 318                     break;
 319                 case TYPE_VECTOR:
 320                     out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
 321                     break;
 322                 default:
 323                     parseerror(parser, "Cannot add type %s from %s",
 324                                type_name[exprs[1]->expression.vtype],
 325                                type_name[exprs[0]->expression.vtype]);
 326                     return false;
 327             };
 328             break;
 329         case opid1('*'):
 330             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
 331                 exprs[0]->expression.vtype != TYPE_VECTOR &&
 332                 exprs[0]->expression.vtype != TYPE_FLOAT &&
 333                 exprs[1]->expression.vtype != TYPE_VECTOR &&
 334                 exprs[1]->expression.vtype != TYPE_FLOAT)
 335             {
 336                 parseerror(parser, "Cannot multiply type %s from %s",
 337                            type_name[exprs[1]->expression.vtype],
 338                            type_name[exprs[0]->expression.vtype]);
 339                 return false;
 340             }
 341             switch (exprs[0]->expression.vtype) {
 342                 case TYPE_FLOAT:
 343                     if (exprs[1]->expression.vtype == TYPE_VECTOR)
 344                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
 345                     else
 346                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
 347                     break;
 348                 case TYPE_VECTOR:
 349                     if (exprs[1]->expression.vtype == TYPE_FLOAT)
 350                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
 351                     else
 352                         out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
 353                     break;
 354                 default:
 355                     parseerror(parser, "Cannot add type %s from %s",
 356                                type_name[exprs[1]->expression.vtype],
 357                                type_name[exprs[0]->expression.vtype]);
 358                     return false;
 359             };
 360             break;
 361         case opid1('/'):
 362             if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
 363                 exprs[0]->expression.vtype != TYPE_FLOAT)
 364             {
 365                 parseerror(parser, "Cannot divide types %s and %s",
 366                            type_name[exprs[0]->expression.vtype],
 367                            type_name[exprs[1]->expression.vtype]);
 368                 return false;
 369             }
 370             out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
 371             break;
 372
 373
 374         case opid1('='):
 375             if (!vars[0]) {
 376                 parseerror(parser, "Cannot assign to non-variable");
 377                 return false;
 378             }
 379             out = (ast_expression*)ast_store_new(ctx,
 380                                                  type_store_instr[vars[0]->expression.vtype],
 381                                                  vars[0], exprs[1]);
 382             break;
 383     }
 384
 385     if (!out) {
 386         parseerror(parser, "failed to apply operand %s", op->op);
 387         return false;
 388     }
 389
 390     sy->out[sy->out_count++] = syexp(ctx, out);
 391     return true;
 392 }
 393
 394 static bool parser_close_paren(parser_t *parser, shunt *sy)
 395 {
 396     if (!sy->ops_count) {
 397         parseerror(parser, "unmatched closing paren");
 398         return false;
 399     }
 400     if (sy->ops[sy->ops_count-1].paren == 1) {
 401         parseerror(parser, "empty parenthesis expression");
 402         return false;
 403     }
 404     while (sy->ops_count) {
 405         if (sy->ops[sy->ops_count-1].paren == 1) {
 406             sy->ops_count--;
 407             break;
 408         }
 409         if (!parser_sy_pop(parser, sy))
 410             return false;
 411     }
 412     return true;
 413 }
 414
 415 static ast_expression* parser_expression(parser_t *parser)
 416 {
 417     ast_expression *expr = NULL;
 418     shunt sy;
 419     bool wantop = false;
 420
 421     MEM_VECTOR_INIT(&sy, out);
 422     MEM_VECTOR_INIT(&sy, ops);
 423
 424     while (true)
 425     {
 426         if (!wantop)
 427         {
 428             bool nextwant = true;
 429             if (parser->tok == TOKEN_IDENT)
 430             {
 431                 /* variable */
 432                 ast_value *var = parser_find_var(parser, parser_tokval(parser));
 433                 if (!var) {
 434                     parseerror(parser, "unexpected ident: %s", parser_tokval(parser));
 435                     goto onerr;
 436                 }
 437                 if (!shunt_out_add(&sy, syval(parser_ctx(parser), var))) {
 438                     parseerror(parser, "out of memory");
 439                     goto onerr;
 440                 }
 441             }
 442             else if (parser->tok == TOKEN_FLOATCONST) {
 443                 ast_value *val = parser_const_float(parser, (parser_token(parser)->constval.f));
 444                 if (!val)
 445                     return false;
 446                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 447                     parseerror(parser, "out of memory");
 448                     goto onerr;
 449                 }
 450             }
 451             else if (parser->tok == TOKEN_INTCONST) {
 452                 ast_value *val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
 453                 if (!val)
 454                     return false;
 455                 if (!shunt_out_add(&sy, syexp(parser_ctx(parser), (ast_expression*)val))) {
 456                     parseerror(parser, "out of memory");
 457                     goto onerr;
 458                 }
 459             }
 460             else if (parser->tok == '(') {
 461                 nextwant = false; /* not expecting an operator next */
 462                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 1))) {
 463                     parseerror(parser, "out of memory");
 464                     goto onerr;
 465                 }
 466             }
 467             else {
 468                 /* TODO: prefix operators */
 469                 parseerror(parser, "expected statement");
 470                 goto onerr;
 471             }
 472             wantop = nextwant;
 473             parser->lex->flags.noops = !wantop;
 474         } else {
 475             if (parser->tok == '(') {
 476                 /* we expected an operator, this is the function-call operator */
 477                 if (!shunt_ops_add(&sy, syparen(parser_ctx(parser), 'f'))) {
 478                     parseerror(parser, "out of memory");
 479                     goto onerr;
 480                 }
 481             }
 482             else if (parser->tok == ')') {
 483                 /* we do expect an operator next */
 484                 /* closing an opening paren */
 485                 if (!parser_close_paren(parser, &sy))
 486                     goto onerr;
 487             }
 488             else if (parser->tok != TOKEN_OPERATOR) {
 489                 parseerror(parser, "expected operator or end of statement");
 490                 goto onerr;
 491             }
 492             else {
 493                 /* classify the operator */
 494                 /* TODO: suffix operators */
 495                 const oper_info *op;
 496                 const oper_info *olast = NULL;
 497                 size_t o;
 498                 for (o = 0; o < operator_count; ++o) {
 499                     if (!(operators[o].flags & OP_PREFIX) &&
 500                         !(operators[o].flags & OP_SUFFIX) && /* remove this */
 501                         !strcmp(parser_tokval(parser), operators[o].op))
 502                     {
 503                         break;
 504                     }
 505                 }
 506                 if (o == operator_count) {
 507                     /* no operator found... must be the end of the statement */
 508                     break;
 509                 }
 510                 /* found an operator */
 511                 op = &operators[o];
 512
 513                 if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 514                     olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 515
 516                 while (olast && (
 517                         (op->prec < olast->prec) ||
 518                         (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
 519                 {
 520                     if (!parser_sy_pop(parser, &sy))
 521                         goto onerr;
 522                     if (sy.ops_count && !sy.ops[sy.ops_count-1].paren)
 523                         olast = &operators[sy.ops[sy.ops_count-1].etype-1];
 524                 }
 525
 526                 if (!shunt_ops_add(&sy, syop(parser_ctx(parser), op)))
 527                     goto onerr;
 528             }
 529             wantop = false;
 530             parser->lex->flags.noops = true;
 531         }
 532         if (!parser_next(parser)) {
 533             goto onerr;
 534         }
 535         if (parser->tok == ';') {
 536             break;
 537         }
 538     }
 539     if (!parser_next(parser)) {
 540         parseerror(parser, "Unexpected end of file");
 541         goto onerr;
 542     }
 543
 544     while (sy.ops_count) {
 545         if (!parser_sy_pop(parser, &sy))
 546             goto onerr;
 547     }
 548
 549     parser->lex->flags.noops = true;
 550     if (!sy.out_count) {
 551         parseerror(parser, "empty expression");
 552         expr = NULL;
 553     } else
 554         expr = sy.out[0].out;
 555     MEM_VECTOR_CLEAR(&sy, out);
 556     MEM_VECTOR_CLEAR(&sy, ops);
 557     return expr;
 558
 559 onerr:
 560     parser->lex->flags.noops = true;
 561     MEM_VECTOR_CLEAR(&sy, out);
 562     MEM_VECTOR_CLEAR(&sy, ops);
 563     return NULL;
 564 }
 565
 566 static bool parser_variable(parser_t *parser, ast_block *localblock);
 567 static bool parser_body_do(parser_t *parser, ast_block *block)
 568 {
 569     if (parser->tok == TOKEN_TYPENAME)
 570     {
 571         /* local variable */
 572         if (!parser_variable(parser, block))
 573             return false;
 574         return true;
 575     }
 576     else if (parser->tok == TOKEN_KEYWORD)
 577     {
 578         if (!strcmp(parser_tokval(parser), "return"))
 579         {
 580             ast_expression *exp;
 581             ast_return *ret;
 582
 583             if (!parser_next(parser)) {
 584                 parseerror(parser, "expected return expression");
 585                 return false;
 586             }
 587
 588             exp = parser_expression(parser);
 589             if (!exp)
 590                 return false;
 591             ret = ast_return_new(exp->expression.node.context, exp);
 592             if (!ret) {
 593                 ast_delete(exp);
 594                 return false;
 595             }
 596             if (!ast_block_exprs_add(block, (ast_expression*)ret)) {
 597                 ast_delete(ret);
 598                 return false;
 599             }
 600             return true;
 601         }
 602         parseerror(parser, "Unexpected keyword");
 603         return false;
 604     }
 605     else if (parser->tok == '{')
 606     {
 607         /* a block */
 608         parseerror(parser, "TODO: inner blocks: %s", parser_tokval(parser));
 609         return false;
 610     }
 611     else
 612     {
 613         ast_expression *exp = parser_expression(parser);
 614         if (!exp)
 615             return false;
 616         if (!ast_block_exprs_add(block, exp)) {
 617             ast_delete(exp);
 618             return false;
 619         }
 620         return true;
 621     }
 622 }
 623
 624 static ast_block* parser_parse_block(parser_t *parser)
 625 {
 626     size_t oldblocklocal;
 627     ast_block *block = NULL;
 628
 629     oldblocklocal = parser->blocklocal;
 630     parser->blocklocal = parser->locals_count;
 631
 632     if (!parser_next(parser)) { /* skip the '{' */
 633         parseerror(parser, "expected function body");
 634         goto cleanup;
 635     }
 636
 637     block = ast_block_new(parser_ctx(parser));
 638
 639     while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
 640     {
 641         if (parser->tok == '}')
 642             break;
 643
 644         if (!parser_body_do(parser, block)) {
 645             ast_block_delete(block);
 646             block = NULL;
 647             goto cleanup;
 648         }
 649     }
 650
 651     if (parser->tok != '}') {
 652         ast_block_delete(block);
 653         block = NULL;
 654     } else {
 655         (void)parser_next(parser);
 656     }
 657
 658 cleanup:
 659     parser->blocklocal = oldblocklocal;
 660     return block;
 661 }
 662
 663 static bool parser_variable(parser_t *parser, ast_block *localblock)
 664 {
 665     bool          isfunc = false;
 666     ast_function *func = NULL;
 667     lex_ctx       ctx;
 668     ast_value    *var;
 669
 670     int basetype = parser_token(parser)->constval.t;
 671
 672     while (true)
 673     {
 674         if (!parser_next(parser)) { /* skip basetype or comma */
 675             parseerror(parser, "expected variable declaration");
 676             return false;
 677         }
 678
 679         isfunc = false;
 680         func = NULL;
 681         ctx = parser_ctx(parser);
 682         var = parser_parse_type(parser, basetype, &isfunc);
 683
 684         if (!var)
 685             return false;
 686
 687         if (parser->tok != TOKEN_IDENT) {
 688             parseerror(parser, "expected variable name\n");
 689             return false;
 690         }
 691
 692         if (!localblock && parser_find_global(parser, parser_tokval(parser))) {
 693             ast_value_delete(var);
 694             parseerror(parser, "global already exists: %s\n", parser_tokval(parser));
 695             return false;
 696         }
 697
 698         if (localblock && parser_find_local(parser, parser_tokval(parser), parser->blocklocal)) {
 699             ast_value_delete(var);
 700             parseerror(parser, "local variable already exists: %s\n", parser_tokval(parser));
 701             return false;
 702         }
 703
 704         if (!ast_value_set_name(var, parser_tokval(parser))) {
 705             parseerror(parser, "failed to set variable name\n");
 706             ast_value_delete(var);
 707             return false;
 708         }
 709
 710         if (isfunc) {
 711             /* a function was defined */
 712             ast_value *fval;
 713
 714             /* turn var into a value of TYPE_FUNCTION, with the old var
 715              * as return type
 716              */
 717             fval = ast_value_new(ctx, var->name, TYPE_FUNCTION);
 718             func = ast_function_new(ctx, var->name, fval);
 719             if (!fval || !func) {
 720                 ast_value_delete(var);
 721                 if (fval) ast_value_delete(fval);
 722                 if (func) ast_function_delete(func);
 723                 return false;
 724             }
 725
 726             fval->expression.next = (ast_expression*)var;
 727             MEM_VECTOR_MOVE(var, params, fval, params);
 728
 729             if (!parser_t_functions_add(parser, func)) {
 730                 ast_value_delete(var);
 731                 if (fval) ast_value_delete(fval);
 732                 if (func) ast_function_delete(func);
 733                 return false;
 734             }
 735
 736             var = fval;
 737         }
 738
 739         if ( (!localblock && !parser_t_globals_add(parser, var)) ||
 740              ( localblock && !parser_t_locals_add(parser, var)) )
 741         {
 742             ast_value_delete(var);
 743             return false;
 744         }
 745         if (localblock && !ast_block_locals_add(localblock, var))
 746         {
 747             parser->locals_count--;
 748             ast_value_delete(var);
 749             return false;
 750         }
 751
 752         if (!parser_next(parser)) {
 753             ast_value_delete(var);
 754             return false;
 755         }
 756
 757         if (parser->tok == ';') {
 758             if (!parser_next(parser))
 759                 return parser->tok == TOKEN_EOF;
 760             return true;
 761         }
 762
 763         if (parser->tok == ',') {
 764             /* another var */
 765             continue;
 766         }
 767
 768         if (parser->tok != '=') {
 769             parseerror(parser, "expected '=' or ';'");
 770             return false;
 771         }
 772
 773         if (!parser_next(parser))
 774             return false;
 775
 776         if (parser->tok == '#') {
 777             if (localblock) {
 778                 parseerror(parser, "cannot declare builtins within functions");
 779                 return false;
 780             }
 781             if (!isfunc || !func) {
 782                 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name);
 783                 return false;
 784             }
 785             if (!parser_next(parser)) {
 786                 parseerror(parser, "expected builtin number");
 787                 return false;
 788             }
 789             if (parser->tok != TOKEN_INTCONST) {
 790                 parseerror(parser, "builtin number must be an integer constant");
 791                 return false;
 792             }
 793             if (parser_token(parser)->constval.i <= 0) {
 794                 parseerror(parser, "builtin number must be positive integer greater than zero");
 795                 return false;
 796             }
 797
 798             func->builtin = -parser_token(parser)->constval.i;
 799         } else if (parser->tok == '{') {
 800             /* function body */
 801             ast_block *block;
 802             ast_function *old = parser->function;
 803
 804             if (localblock) {
 805                 parseerror(parser, "cannot declare functions within functions");
 806                 return false;
 807             }
 808
 809             parser->function = func;
 810             block = parser_parse_block(parser);
 811             parser->function = old;
 812
 813             if (!block)
 814                 return false;
 815
 816             if (!ast_function_blocks_add(func, block)) {
 817                 ast_block_delete(block);
 818                 return false;
 819             }
 820             return true;
 821         } else {
 822             parseerror(parser, "TODO, const assignment");
 823         }
 824
 825         if (!parser_next(parser))
 826             return false;
 827
 828         if (parser->tok == ',') {
 829             /* another */
 830             continue;
 831         }
 832
 833         if (parser->tok != ';') {
 834             parseerror(parser, "expected semicolon");
 835             return false;
 836         }
 837
 838         (void)parser_next(parser);
 839
 840         return true;
 841     }
 842 }
 843
 844 static bool parser_do(parser_t *parser)
 845 {
 846     if (parser->tok == TOKEN_TYPENAME)
 847     {
 848         return parser_variable(parser, NULL);
 849     }
 850     else if (parser->tok == TOKEN_KEYWORD)
 851     {
 852         /* handle 'var' and 'const' */
 853         return false;
 854     }
 855     else if (parser->tok == '.')
 856     {
 857         /* entity-member declaration */
 858         return false;
 859     }
 860     else
 861     {
 862         parseerror(parser, "unexpected token: %s", parser->lex->tok->value);
 863         return false;
 864     }
 865     return true;
 866 }
 867
 868 bool parser_compile(const char *filename)
 869 {
 870     size_t i;
 871     parser_t *parser;
 872     ir_builder *ir;
 873
 874     parser = (parser_t*)mem_a(sizeof(parser_t));
 875     if (!parser)
 876         return false;
 877
 878     memset(parser, 0, sizeof(parser));
 879
 880     MEM_VECTOR_INIT(parser, globals);
 881     MEM_VECTOR_INIT(parser, locals);
 882     parser->lex = lex_open(filename);
 883
 884     if (!parser->lex) {
 885         printf("failed to open file \"%s\"\n", filename);
 886         return false;
 887     }
 888
 889     /* initial lexer/parser state */
 890     parser->lex->flags.noops = true;
 891
 892     if (parser_next(parser))
 893     {
 894         while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
 895         {
 896             if (!parser_do(parser)) {
 897                 if (parser->tok == TOKEN_EOF)
 898                     parseerror(parser, "unexpected eof");
 899                 else
 900                     parseerror(parser, "parse error\n");
 901                 lex_close(parser->lex);
 902                 mem_d(parser);
 903                 return false;
 904             }
 905         }
 906     }
 907
 908     lex_close(parser->lex);
 909
 910     ir = ir_builder_new("gmqcc_out");
 911     if (!ir) {
 912         printf("failed to allocate builder\n");
 913         goto cleanup;
 914     }
 915
 916     for (i = 0; i < parser->imm_float_count; ++i) {
 917         if (!ast_global_codegen(parser->imm_float[i], ir)) {
 918             printf("failed to generate global %s\n", parser->imm_float[i]->name);
 919         }
 920     }
 921     for (i = 0; i < parser->globals_count; ++i) {
 922         if (!ast_global_codegen(parser->globals[i], ir)) {
 923             printf("failed to generate global %s\n", parser->globals[i]->name);
 924         }
 925     }
 926     for (i = 0; i < parser->functions_count; ++i) {
 927         if (!ast_function_codegen(parser->functions[i], ir)) {
 928             printf("failed to generate function %s\n", parser->functions[i]->name);
 929         }
 930         if (!ir_function_finalize(parser->functions[i]->ir_func)) {
 931             printf("failed to finalize function %s\n", parser->functions[i]->name);
 932         }
 933     }
 934
 935     ir_builder_dump(ir, printf);
 936
 937 cleanup:
 938     for (i = 0; i < parser->globals_count; ++i) {
 939         ast_value_delete(parser->globals[i]);
 940     }
 941     MEM_VECTOR_CLEAR(parser, globals);
 942
 943     mem_d(parser);
 944     return true;
 945 }