Update license headers
[xonotic/gmqcc.git] / parser.c
1 /*
2  * Copyright (C) 2012, 2013, 2014, 2015
3  *     Wolfgang Bumiller
4  *     Dale Weiler
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy of
7  * this software and associated documentation files (the "Software"), to deal in
8  * the Software without restriction, including without limitation the rights to
9  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
10  * of the Software, and to permit persons to whom the Software is furnished to do
11  * so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #include <string.h>
25 #include <math.h>
26
27 #include "parser.h"
28
29 #define PARSER_HT_LOCALS  2
30 #define PARSER_HT_SIZE    512
31 #define TYPEDEF_HT_SIZE   512
32
33 static void parser_enterblock(parser_t *parser);
34 static bool parser_leaveblock(parser_t *parser);
35 static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e);
36 static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e);
37 static bool parse_typedef(parser_t *parser);
38 static bool parse_variable(parser_t *parser, ast_block *localblock, bool nofields, int qualifier, ast_value *cached_typedef, bool noref, bool is_static, uint32_t qflags, char *vstring);
39 static ast_block* parse_block(parser_t *parser);
40 static bool parse_block_into(parser_t *parser, ast_block *block);
41 static bool parse_statement_or_block(parser_t *parser, ast_expression **out);
42 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases);
43 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma, bool truthvalue, bool with_labels);
44 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma, bool with_labels);
45 static ast_value* parser_create_array_setter_proto(parser_t *parser, ast_value *array, const char *funcname);
46 static ast_value* parser_create_array_getter_proto(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname);
47 static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef, bool *is_vararg);
48
49 static void parseerror(parser_t *parser, const char *fmt, ...)
50 {
51     va_list ap;
52     va_start(ap, fmt);
53     vcompile_error(parser->lex->tok.ctx, fmt, ap);
54     va_end(ap);
55 }
56
57 /* returns true if it counts as an error */
58 static bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, ...)
59 {
60     bool    r;
61     va_list ap;
62     va_start(ap, fmt);
63     r = vcompile_warning(parser->lex->tok.ctx, warntype, fmt, ap);
64     va_end(ap);
65     return r;
66 }
67
68 /**********************************************************************
69  * parsing
70  */
71
72 static bool parser_next(parser_t *parser)
73 {
74     /* lex_do kills the previous token */
75     parser->tok = lex_do(parser->lex);
76     if (parser->tok == TOKEN_EOF)
77         return true;
78     if (parser->tok >= TOKEN_ERROR) {
79         parseerror(parser, "lex error");
80         return false;
81     }
82     return true;
83 }
84
85 #define parser_tokval(p) ((p)->lex->tok.value)
86 #define parser_token(p)  (&((p)->lex->tok))
87
88 char *parser_strdup(const char *str)
89 {
90     if (str && !*str) {
91         /* actually dup empty strings */
92         char *out = (char*)mem_a(1);
93         *out = 0;
94         return out;
95     }
96     return util_strdup(str);
97 }
98
99 static ast_expression* parser_find_field(parser_t *parser, const char *name)
100 {
101     return ( ast_expression*)util_htget(parser->htfields, name);
102 }
103
104 static ast_expression* parser_find_label(parser_t *parser, const char *name)
105 {
106     size_t i;
107     for(i = 0; i < vec_size(parser->labels); i++)
108         if (!strcmp(parser->labels[i]->name, name))
109             return (ast_expression*)parser->labels[i];
110     return NULL;
111 }
112
113 ast_expression* parser_find_global(parser_t *parser, const char *name)
114 {
115     ast_expression *var = (ast_expression*)util_htget(parser->aliases, parser_tokval(parser));
116     if (var)
117         return var;
118     return (ast_expression*)util_htget(parser->htglobals, name);
119 }
120
121 static ast_expression* parser_find_param(parser_t *parser, const char *name)
122 {
123     size_t i;
124     ast_value *fun;
125     if (!parser->function)
126         return NULL;
127     fun = parser->function->vtype;
128     for (i = 0; i < vec_size(fun->expression.params); ++i) {
129         if (!strcmp(fun->expression.params[i]->name, name))
130             return (ast_expression*)(fun->expression.params[i]);
131     }
132     return NULL;
133 }
134
135 static ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto, bool *isparam)
136 {
137     size_t          i, hash;
138     ast_expression *e;
139
140     hash = util_hthash(parser->htglobals, name);
141
142     *isparam = false;
143     for (i = vec_size(parser->variables); i > upto;) {
144         --i;
145         if ( (e = (ast_expression*)util_htgeth(parser->variables[i], name, hash)) )
146             return e;
147     }
148     *isparam = true;
149     return parser_find_param(parser, name);
150 }
151
152 static ast_expression* parser_find_var(parser_t *parser, const char *name)
153 {
154     bool dummy;
155     ast_expression *v;
156     v         = parser_find_local(parser, name, 0, &dummy);
157     if (!v) v = parser_find_global(parser, name);
158     return v;
159 }
160
161 static ast_value* parser_find_typedef(parser_t *parser, const char *name, size_t upto)
162 {
163     size_t     i, hash;
164     ast_value *e;
165     hash = util_hthash(parser->typedefs[0], name);
166
167     for (i = vec_size(parser->typedefs); i > upto;) {
168         --i;
169         if ( (e = (ast_value*)util_htgeth(parser->typedefs[i], name, hash)) )
170             return e;
171     }
172     return NULL;
173 }
174
175 typedef struct
176 {
177     size_t etype; /* 0 = expression, others are operators */
178     bool            isparen;
179     size_t          off;
180     ast_expression *out;
181     ast_block      *block; /* for commas and function calls */
182     lex_ctx_t ctx;
183 } sy_elem;
184
185 enum {
186     PAREN_EXPR,
187     PAREN_FUNC,
188     PAREN_INDEX,
189     PAREN_TERNARY1,
190     PAREN_TERNARY2
191 };
192 typedef struct
193 {
194     sy_elem        *out;
195     sy_elem        *ops;
196     size_t         *argc;
197     unsigned int   *paren;
198 } shunt;
199
200 static sy_elem syexp(lex_ctx_t ctx, ast_expression *v) {
201     sy_elem e;
202     e.etype = 0;
203     e.off   = 0;
204     e.out   = v;
205     e.block = NULL;
206     e.ctx   = ctx;
207     e.isparen = false;
208     return e;
209 }
210
211 static sy_elem syblock(lex_ctx_t ctx, ast_block *v) {
212     sy_elem e;
213     e.etype = 0;
214     e.off   = 0;
215     e.out   = (ast_expression*)v;
216     e.block = v;
217     e.ctx   = ctx;
218     e.isparen = false;
219     return e;
220 }
221
222 static sy_elem syop(lex_ctx_t ctx, const oper_info *op) {
223     sy_elem e;
224     e.etype = 1 + (op - operators);
225     e.off   = 0;
226     e.out   = NULL;
227     e.block = NULL;
228     e.ctx   = ctx;
229     e.isparen = false;
230     return e;
231 }
232
233 static sy_elem syparen(lex_ctx_t ctx, size_t off) {
234     sy_elem e;
235     e.etype = 0;
236     e.off   = off;
237     e.out   = NULL;
238     e.block = NULL;
239     e.ctx   = ctx;
240     e.isparen = true;
241     return e;
242 }
243
244 /* With regular precedence rules, ent.foo[n] is the same as (ent.foo)[n],
245  * so we need to rotate it to become ent.(foo[n]).
246  */
247 static bool rotate_entfield_array_index_nodes(ast_expression **out)
248 {
249     ast_array_index *index, *oldindex;
250     ast_entfield    *entfield;
251
252     ast_value       *field;
253     ast_expression  *sub;
254     ast_expression  *entity;
255
256     lex_ctx_t ctx = ast_ctx(*out);
257
258     if (!ast_istype(*out, ast_array_index))
259         return false;
260     index = (ast_array_index*)*out;
261
262     if (!ast_istype(index->array, ast_entfield))
263         return false;
264     entfield = (ast_entfield*)index->array;
265
266     if (!ast_istype(entfield->field, ast_value))
267         return false;
268     field = (ast_value*)entfield->field;
269
270     sub    = index->index;
271     entity = entfield->entity;
272
273     oldindex = index;
274
275     index = ast_array_index_new(ctx, (ast_expression*)field, sub);
276     entfield = ast_entfield_new(ctx, entity, (ast_expression*)index);
277     *out = (ast_expression*)entfield;
278
279     oldindex->array = NULL;
280     oldindex->index = NULL;
281     ast_delete(oldindex);
282
283     return true;
284 }
285
286 static bool check_write_to(lex_ctx_t ctx, ast_expression *expr)
287 {
288     if (ast_istype(expr, ast_value)) {
289         ast_value *val = (ast_value*)expr;
290         if (val->cvq == CV_CONST) {
291             if (val->name[0] == '#') {
292                 compile_error(ctx, "invalid assignment to a literal constant");
293                 return false;
294             }
295             /*
296              * To work around quakeworld we must elide the error and make it
297              * a warning instead.
298              */
299             if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_QCC)
300                 compile_error(ctx, "assignment to constant `%s`", val->name);
301             else
302                 (void)!compile_warning(ctx, WARN_CONST_OVERWRITE, "assignment to constant `%s`", val->name);
303             return false;
304         }
305     }
306     return true;
307 }
308
309 static bool parser_sy_apply_operator(parser_t *parser, shunt *sy)
310 {
311     const oper_info *op;
312     lex_ctx_t ctx;
313     ast_expression *out = NULL;
314     ast_expression *exprs[3];
315     ast_block      *blocks[3];
316     ast_binstore   *asbinstore;
317     size_t i, assignop, addop, subop;
318     qcint_t  generated_op = 0;
319
320     char ty1[1024];
321     char ty2[1024];
322
323     if (!vec_size(sy->ops)) {
324         parseerror(parser, "internal error: missing operator");
325         return false;
326     }
327
328     if (vec_last(sy->ops).isparen) {
329         parseerror(parser, "unmatched parenthesis");
330         return false;
331     }
332
333     op = &operators[vec_last(sy->ops).etype - 1];
334     ctx = vec_last(sy->ops).ctx;
335
336     if (vec_size(sy->out) < op->operands) {
337         if (op->flags & OP_PREFIX)
338             compile_error(ctx, "expected expression after unary operator `%s`", op->op, (int)op->id);
339         else /* this should have errored previously already */
340             compile_error(ctx, "expected expression after operator `%s`", op->op, (int)op->id);
341         return false;
342     }
343
344     vec_shrinkby(sy->ops, 1);
345
346     /* op(:?) has no input and no output */
347     if (!op->operands)
348         return true;
349
350     vec_shrinkby(sy->out, op->operands);
351     for (i = 0; i < op->operands; ++i) {
352         exprs[i]  = sy->out[vec_size(sy->out)+i].out;
353         blocks[i] = sy->out[vec_size(sy->out)+i].block;
354
355         if (exprs[i]->vtype == TYPE_NOEXPR &&
356             !(i != 0 && op->id == opid2('?',':')) &&
357             !(i == 1 && op->id == opid1('.')))
358         {
359             if (ast_istype(exprs[i], ast_label))
360                 compile_error(ast_ctx(exprs[i]), "expected expression, got an unknown identifier");
361             else
362                 compile_error(ast_ctx(exprs[i]), "not an expression");
363             (void)!compile_warning(ast_ctx(exprs[i]), WARN_DEBUG, "expression %u\n", (unsigned int)i);
364         }
365     }
366
367     if (blocks[0] && !vec_size(blocks[0]->exprs) && op->id != opid1(',')) {
368         compile_error(ctx, "internal error: operator cannot be applied on empty blocks");
369         return false;
370     }
371
372 #define NotSameType(T) \
373              (exprs[0]->vtype != exprs[1]->vtype || \
374               exprs[0]->vtype != T)
375
376     switch (op->id)
377     {
378         default:
379             compile_error(ctx, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id);
380             return false;
381
382         case opid1('.'):
383             if (exprs[0]->vtype == TYPE_VECTOR &&
384                 exprs[1]->vtype == TYPE_NOEXPR)
385             {
386                 if      (exprs[1] == (ast_expression*)parser->const_vec[0])
387                     out = (ast_expression*)ast_member_new(ctx, exprs[0], 0, NULL);
388                 else if (exprs[1] == (ast_expression*)parser->const_vec[1])
389                     out = (ast_expression*)ast_member_new(ctx, exprs[0], 1, NULL);
390                 else if (exprs[1] == (ast_expression*)parser->const_vec[2])
391                     out = (ast_expression*)ast_member_new(ctx, exprs[0], 2, NULL);
392                 else {
393                     compile_error(ctx, "access to invalid vector component");
394                     return false;
395                 }
396             }
397             else if (exprs[0]->vtype == TYPE_ENTITY) {
398                 if (exprs[1]->vtype != TYPE_FIELD) {
399                     compile_error(ast_ctx(exprs[1]), "type error: right hand of member-operand should be an entity-field");
400                     return false;
401                 }
402                 out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]);
403             }
404             else if (exprs[0]->vtype == TYPE_VECTOR) {
405                 compile_error(ast_ctx(exprs[1]), "vectors cannot be accessed this way");
406                 return false;
407             }
408             else {
409                 compile_error(ast_ctx(exprs[1]), "type error: member-of operator on something that is not an entity or vector");
410                 return false;
411             }
412             break;
413
414         case opid1('['):
415             if (exprs[0]->vtype != TYPE_ARRAY &&
416                 !(exprs[0]->vtype == TYPE_FIELD &&
417                   exprs[0]->next->vtype == TYPE_ARRAY))
418             {
419                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
420                 compile_error(ast_ctx(exprs[0]), "cannot index value of type %s", ty1);
421                 return false;
422             }
423             if (exprs[1]->vtype != TYPE_FLOAT) {
424                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
425                 compile_error(ast_ctx(exprs[1]), "index must be of type float, not %s", ty1);
426                 return false;
427             }
428             out = (ast_expression*)ast_array_index_new(ctx, exprs[0], exprs[1]);
429             if (rotate_entfield_array_index_nodes(&out))
430             {
431 #if 0
432                 /* This is not broken in fteqcc anymore */
433                 if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_GMQCC) {
434                     /* this error doesn't need to make us bail out */
435                     (void)!parsewarning(parser, WARN_EXTENSIONS,
436                                         "accessing array-field members of an entity without parenthesis\n"
437                                         " -> this is an extension from -std=gmqcc");
438                 }
439 #endif
440             }
441             break;
442
443         case opid1(','):
444             if (vec_size(sy->paren) && vec_last(sy->paren) == PAREN_FUNC) {
445                 vec_push(sy->out, syexp(ctx, exprs[0]));
446                 vec_push(sy->out, syexp(ctx, exprs[1]));
447                 vec_last(sy->argc)++;
448                 return true;
449             }
450             if (blocks[0]) {
451                 if (!ast_block_add_expr(blocks[0], exprs[1]))
452                     return false;
453             } else {
454                 blocks[0] = ast_block_new(ctx);
455                 if (!ast_block_add_expr(blocks[0], exprs[0]) ||
456                     !ast_block_add_expr(blocks[0], exprs[1]))
457                 {
458                     return false;
459                 }
460             }
461             ast_block_set_type(blocks[0], exprs[1]);
462
463             vec_push(sy->out, syblock(ctx, blocks[0]));
464             return true;
465
466         case opid2('+','P'):
467             out = exprs[0];
468             break;
469         case opid2('-','P'):
470             if ((out = fold_op(parser->fold, op, exprs)))
471                 break;
472
473             if (exprs[0]->vtype != TYPE_FLOAT &&
474                 exprs[0]->vtype != TYPE_VECTOR) {
475                     compile_error(ctx, "invalid types used in unary expression: cannot negate type %s",
476                                   type_name[exprs[0]->vtype]);
477                 return false;
478             }
479             if (exprs[0]->vtype == TYPE_FLOAT)
480                 out = (ast_expression*)ast_unary_new(ctx, VINSTR_NEG_F, exprs[0]);
481             else
482                 out = (ast_expression*)ast_unary_new(ctx, VINSTR_NEG_V, exprs[0]);
483             break;
484
485         case opid2('!','P'):
486             if (!(out = fold_op(parser->fold, op, exprs))) {
487                 switch (exprs[0]->vtype) {
488                     case TYPE_FLOAT:
489                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
490                         break;
491                     case TYPE_VECTOR:
492                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]);
493                         break;
494                     case TYPE_STRING:
495                         if (OPTS_FLAG(TRUE_EMPTY_STRINGS))
496                             out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
497                         else
498                             out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[0]);
499                         break;
500                     /* we don't constant-fold NOT for these types */
501                     case TYPE_ENTITY:
502                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_ENT, exprs[0]);
503                         break;
504                     case TYPE_FUNCTION:
505                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_FNC, exprs[0]);
506                         break;
507                     default:
508                     compile_error(ctx, "invalid types used in expression: cannot logically negate type %s",
509                                   type_name[exprs[0]->vtype]);
510                     return false;
511                 }
512             }
513             break;
514
515         case opid1('+'):
516             if (exprs[0]->vtype != exprs[1]->vtype ||
517                (exprs[0]->vtype != TYPE_VECTOR && exprs[0]->vtype != TYPE_FLOAT) )
518             {
519                 compile_error(ctx, "invalid types used in expression: cannot add type %s and %s",
520                               type_name[exprs[0]->vtype],
521                               type_name[exprs[1]->vtype]);
522                 return false;
523             }
524             if (!(out = fold_op(parser->fold, op, exprs))) {
525                 switch (exprs[0]->vtype) {
526                     case TYPE_FLOAT:
527                         out = fold_binary(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
528                         break;
529                     case TYPE_VECTOR:
530                         out = fold_binary(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
531                         break;
532                     default:
533                         compile_error(ctx, "invalid types used in expression: cannot add type %s and %s",
534                                       type_name[exprs[0]->vtype],
535                                       type_name[exprs[1]->vtype]);
536                         return false;
537                 }
538             }
539             break;
540         case opid1('-'):
541             if  (exprs[0]->vtype != exprs[1]->vtype ||
542                 (exprs[0]->vtype != TYPE_VECTOR && exprs[0]->vtype != TYPE_FLOAT))
543             {
544                 compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s",
545                               type_name[exprs[1]->vtype],
546                               type_name[exprs[0]->vtype]);
547                 return false;
548             }
549             if (!(out = fold_op(parser->fold, op, exprs))) {
550                 switch (exprs[0]->vtype) {
551                     case TYPE_FLOAT:
552                         out = fold_binary(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
553                         break;
554                     case TYPE_VECTOR:
555                         out = fold_binary(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
556                         break;
557                     default:
558                         compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s",
559                                       type_name[exprs[1]->vtype],
560                                       type_name[exprs[0]->vtype]);
561                         return false;
562                 }
563             }
564             break;
565         case opid1('*'):
566             if (exprs[0]->vtype != exprs[1]->vtype &&
567                 !(exprs[0]->vtype == TYPE_VECTOR &&
568                   exprs[1]->vtype == TYPE_FLOAT) &&
569                 !(exprs[1]->vtype == TYPE_VECTOR &&
570                   exprs[0]->vtype == TYPE_FLOAT)
571                 )
572             {
573                 compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s",
574                               type_name[exprs[1]->vtype],
575                               type_name[exprs[0]->vtype]);
576                 return false;
577             }
578             if (!(out = fold_op(parser->fold, op, exprs))) {
579                 switch (exprs[0]->vtype) {
580                     case TYPE_FLOAT:
581                         if (exprs[1]->vtype == TYPE_VECTOR)
582                             out = fold_binary(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
583                         else
584                             out = fold_binary(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
585                         break;
586                     case TYPE_VECTOR:
587                         if (exprs[1]->vtype == TYPE_FLOAT)
588                             out = fold_binary(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
589                         else
590                             out = fold_binary(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
591                         break;
592                     default:
593                         compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s",
594                                       type_name[exprs[1]->vtype],
595                                       type_name[exprs[0]->vtype]);
596                         return false;
597                 }
598             }
599             break;
600
601         case opid1('/'):
602             if (exprs[1]->vtype != TYPE_FLOAT) {
603                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
604                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
605                 compile_error(ctx, "invalid types used in expression: cannot divide types %s and %s", ty1, ty2);
606                 return false;
607             }
608             if (!(out = fold_op(parser->fold, op, exprs))) {
609                 if (exprs[0]->vtype == TYPE_FLOAT)
610                     out = fold_binary(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
611                 else {
612                     ast_type_to_string(exprs[0], ty1, sizeof(ty1));
613                     ast_type_to_string(exprs[1], ty2, sizeof(ty2));
614                     compile_error(ctx, "invalid types used in expression: cannot divide types %s and %s", ty1, ty2);
615                     return false;
616                 }
617             }
618             break;
619
620         case opid1('%'):
621             if (NotSameType(TYPE_FLOAT)) {
622                 compile_error(ctx, "invalid types used in expression: cannot perform modulo operation between types %s and %s",
623                     type_name[exprs[0]->vtype],
624                     type_name[exprs[1]->vtype]);
625                 return false;
626             } else if (!(out = fold_op(parser->fold, op, exprs))) {
627                 /* generate a call to __builtin_mod */
628                 ast_expression *mod  = intrin_func(parser->intrin, "mod");
629                 ast_call       *call = NULL;
630                 if (!mod) return false; /* can return null for missing floor */
631
632                 call = ast_call_new(parser_ctx(parser), mod);
633                 vec_push(call->params, exprs[0]);
634                 vec_push(call->params, exprs[1]);
635
636                 out = (ast_expression*)call;
637             }
638             break;
639
640         case opid2('%','='):
641             compile_error(ctx, "%= is unimplemented");
642             return false;
643
644         case opid1('|'):
645         case opid1('&'):
646         case opid1('^'):
647             if ( !(exprs[0]->vtype == TYPE_FLOAT  && exprs[1]->vtype == TYPE_FLOAT) &&
648                  !(exprs[0]->vtype == TYPE_VECTOR && exprs[1]->vtype == TYPE_FLOAT) &&
649                  !(exprs[0]->vtype == TYPE_VECTOR && exprs[1]->vtype == TYPE_VECTOR))
650             {
651                 compile_error(ctx, "invalid types used in expression: cannot perform bit operations between types %s and %s",
652                               type_name[exprs[0]->vtype],
653                               type_name[exprs[1]->vtype]);
654                 return false;
655             }
656
657             if (!(out = fold_op(parser->fold, op, exprs))) {
658                 /*
659                  * IF the first expression is float, the following will be too
660                  * since scalar ^ vector is not allowed.
661                  */
662                 if (exprs[0]->vtype == TYPE_FLOAT) {
663                     out = fold_binary(ctx,
664                         (op->id == opid1('^') ? VINSTR_BITXOR : op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
665                         exprs[0], exprs[1]);
666                 } else {
667                     /*
668                      * The first is a vector: vector is allowed to bitop with vector and
669                      * with scalar, branch here for the second operand.
670                      */
671                     if (exprs[1]->vtype == TYPE_VECTOR) {
672                         /*
673                          * Bitop all the values of the vector components against the
674                          * vectors components in question.
675                          */
676                         out = fold_binary(ctx,
677                             (op->id == opid1('^') ? VINSTR_BITXOR_V : op->id == opid1('|') ? VINSTR_BITOR_V : VINSTR_BITAND_V),
678                             exprs[0], exprs[1]);
679                     } else {
680                         out = fold_binary(ctx,
681                             (op->id == opid1('^') ? VINSTR_BITXOR_VF : op->id == opid1('|') ? VINSTR_BITOR_VF : VINSTR_BITAND_VF),
682                             exprs[0], exprs[1]);
683                     }
684                 }
685             }
686             break;
687
688         case opid2('<','<'):
689         case opid2('>','>'):
690             if (NotSameType(TYPE_FLOAT)) {
691                 compile_error(ctx, "invalid types used in expression: cannot perform shift between types %s and %s",
692                     type_name[exprs[0]->vtype],
693                     type_name[exprs[1]->vtype]);
694                 return false;
695             }
696
697             if (!(out = fold_op(parser->fold, op, exprs))) {
698                 ast_expression *shift = intrin_func(parser->intrin, (op->id == opid2('<','<')) ? "__builtin_lshift" : "__builtin_rshift");
699                 ast_call       *call  = ast_call_new(parser_ctx(parser), shift);
700                 vec_push(call->params, exprs[0]);
701                 vec_push(call->params, exprs[1]);
702                 out = (ast_expression*)call;
703             }
704             break;
705
706         case opid3('<','<','='):
707         case opid3('>','>','='):
708             if (NotSameType(TYPE_FLOAT)) {
709                 compile_error(ctx, "invalid types used in expression: cannot perform shift operation between types %s and %s",
710                     type_name[exprs[0]->vtype],
711                     type_name[exprs[1]->vtype]);
712                 return false;
713             }
714
715             if(!(out = fold_op(parser->fold, op, exprs))) {
716                 ast_expression *shift = intrin_func(parser->intrin, (op->id == opid3('<','<','=')) ? "__builtin_lshift" : "__builtin_rshift");
717                 ast_call       *call  = ast_call_new(parser_ctx(parser), shift);
718                 vec_push(call->params, exprs[0]);
719                 vec_push(call->params, exprs[1]);
720                 out = (ast_expression*)ast_store_new(
721                     parser_ctx(parser),
722                     INSTR_STORE_F,
723                     exprs[0],
724                     (ast_expression*)call
725                 );
726             }
727
728             break;
729
730         case opid2('|','|'):
731             generated_op += 1; /* INSTR_OR */
732         case opid2('&','&'):
733             generated_op += INSTR_AND;
734             if (!(out = fold_op(parser->fold, op, exprs))) {
735                 if (OPTS_FLAG(PERL_LOGIC) && !ast_compare_type(exprs[0], exprs[1])) {
736                     ast_type_to_string(exprs[0], ty1, sizeof(ty1));
737                     ast_type_to_string(exprs[1], ty2, sizeof(ty2));
738                     compile_error(ctx, "invalid types for logical operation with -fperl-logic: %s and %s", ty1, ty2);
739                     return false;
740                 }
741                 for (i = 0; i < 2; ++i) {
742                     if (OPTS_FLAG(CORRECT_LOGIC) && exprs[i]->vtype == TYPE_VECTOR) {
743                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[i]);
744                         if (!out) break;
745                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, out);
746                         if (!out) break;
747                         exprs[i] = out; out = NULL;
748                         if (OPTS_FLAG(PERL_LOGIC)) {
749                             /* here we want to keep the right expressions' type */
750                             break;
751                         }
752                     }
753                     else if (OPTS_FLAG(FALSE_EMPTY_STRINGS) && exprs[i]->vtype == TYPE_STRING) {
754                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[i]);
755                         if (!out) break;
756                         out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, out);
757                         if (!out) break;
758                         exprs[i] = out; out = NULL;
759                         if (OPTS_FLAG(PERL_LOGIC)) {
760                             /* here we want to keep the right expressions' type */
761                             break;
762                         }
763                     }
764                 }
765                 out = fold_binary(ctx, generated_op, exprs[0], exprs[1]);
766             }
767             break;
768
769         case opid2('?',':'):
770             if (vec_last(sy->paren) != PAREN_TERNARY2) {
771                 compile_error(ctx, "mismatched parenthesis/ternary");
772                 return false;
773             }
774             vec_pop(sy->paren);
775             if (!ast_compare_type(exprs[1], exprs[2])) {
776                 ast_type_to_string(exprs[1], ty1, sizeof(ty1));
777                 ast_type_to_string(exprs[2], ty2, sizeof(ty2));
778                 compile_error(ctx, "operands of ternary expression must have the same type, got %s and %s", ty1, ty2);
779                 return false;
780             }
781             if (!(out = fold_op(parser->fold, op, exprs)))
782                 out = (ast_expression*)ast_ternary_new(ctx, exprs[0], exprs[1], exprs[2]);
783             break;
784
785         case opid2('*', '*'):
786             if (NotSameType(TYPE_FLOAT)) {
787                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
788                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
789                 compile_error(ctx, "invalid types used in exponentiation: %s and %s",
790                     ty1, ty2);
791                 return false;
792             }
793
794             if (!(out = fold_op(parser->fold, op, exprs))) {
795                 ast_call *gencall = ast_call_new(parser_ctx(parser), intrin_func(parser->intrin, "pow"));
796                 vec_push(gencall->params, exprs[0]);
797                 vec_push(gencall->params, exprs[1]);
798                 out = (ast_expression*)gencall;
799             }
800             break;
801
802         case opid2('>', '<'):
803             if (NotSameType(TYPE_VECTOR)) {
804                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
805                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
806                 compile_error(ctx, "invalid types used in cross product: %s and %s",
807                     ty1, ty2);
808                 return false;
809             }
810
811             if (!(out = fold_op(parser->fold, op, exprs))) {
812                 out = fold_binary(
813                         parser_ctx(parser),
814                         VINSTR_CROSS,
815                         exprs[0],
816                         exprs[1]
817                 );
818             }
819
820             break;
821
822         case opid3('<','=','>'): /* -1, 0, or 1 */
823             if (NotSameType(TYPE_FLOAT)) {
824                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
825                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
826                 compile_error(ctx, "invalid types used in comparision: %s and %s",
827                     ty1, ty2);
828
829                 return false;
830             }
831
832             if (!(out = fold_op(parser->fold, op, exprs))) {
833                 /* This whole block is NOT fold_binary safe */
834                 ast_binary *eq = ast_binary_new(ctx, INSTR_EQ_F, exprs[0], exprs[1]);
835
836                 eq->refs = AST_REF_NONE;
837
838                     /* if (lt) { */
839                 out = (ast_expression*)ast_ternary_new(ctx,
840                         (ast_expression*)ast_binary_new(ctx, INSTR_LT, exprs[0], exprs[1]),
841                         /* out = -1 */
842                         (ast_expression*)parser->fold->imm_float[2],
843                     /* } else { */
844                         /* if (eq) { */
845                         (ast_expression*)ast_ternary_new(ctx, (ast_expression*)eq,
846                             /* out = 0 */
847                             (ast_expression*)parser->fold->imm_float[0],
848                         /* } else { */
849                             /* out = 1 */
850                             (ast_expression*)parser->fold->imm_float[1]
851                         /* } */
852                         )
853                     /* } */
854                     );
855
856             }
857             break;
858
859         case opid1('>'):
860             generated_op += 1; /* INSTR_GT */
861         case opid1('<'):
862             generated_op += 1; /* INSTR_LT */
863         case opid2('>', '='):
864             generated_op += 1; /* INSTR_GE */
865         case opid2('<', '='):
866             generated_op += INSTR_LE;
867             if (NotSameType(TYPE_FLOAT)) {
868                 compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
869                               type_name[exprs[0]->vtype],
870                               type_name[exprs[1]->vtype]);
871                 return false;
872             }
873             if (!(out = fold_op(parser->fold, op, exprs)))
874                 out = fold_binary(ctx, generated_op, exprs[0], exprs[1]);
875             break;
876         case opid2('!', '='):
877             if (exprs[0]->vtype != exprs[1]->vtype) {
878                 compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
879                               type_name[exprs[0]->vtype],
880                               type_name[exprs[1]->vtype]);
881                 return false;
882             }
883             if (!(out = fold_op(parser->fold, op, exprs)))
884                 out = fold_binary(ctx, type_ne_instr[exprs[0]->vtype], exprs[0], exprs[1]);
885             break;
886         case opid2('=', '='):
887             if (exprs[0]->vtype != exprs[1]->vtype) {
888                 compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
889                               type_name[exprs[0]->vtype],
890                               type_name[exprs[1]->vtype]);
891                 return false;
892             }
893             if (!(out = fold_op(parser->fold, op, exprs)))
894                 out = fold_binary(ctx, type_eq_instr[exprs[0]->vtype], exprs[0], exprs[1]);
895             break;
896
897         case opid1('='):
898             if (ast_istype(exprs[0], ast_entfield)) {
899                 ast_expression *field = ((ast_entfield*)exprs[0])->field;
900                 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
901                     exprs[0]->vtype == TYPE_FIELD &&
902                     exprs[0]->next->vtype == TYPE_VECTOR)
903                 {
904                     assignop = type_storep_instr[TYPE_VECTOR];
905                 }
906                 else
907                     assignop = type_storep_instr[exprs[0]->vtype];
908                 if (assignop == VINSTR_END || !ast_compare_type(field->next, exprs[1]))
909                 {
910                     ast_type_to_string(field->next, ty1, sizeof(ty1));
911                     ast_type_to_string(exprs[1], ty2, sizeof(ty2));
912                     if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) &&
913                         field->next->vtype == TYPE_FUNCTION &&
914                         exprs[1]->vtype == TYPE_FUNCTION)
915                     {
916                         (void)!compile_warning(ctx, WARN_ASSIGN_FUNCTION_TYPES,
917                                                "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
918                     }
919                     else
920                         compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
921                 }
922             }
923             else
924             {
925                 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
926                     exprs[0]->vtype == TYPE_FIELD &&
927                     exprs[0]->next->vtype == TYPE_VECTOR)
928                 {
929                     assignop = type_store_instr[TYPE_VECTOR];
930                 }
931                 else {
932                     assignop = type_store_instr[exprs[0]->vtype];
933                 }
934
935                 if (assignop == VINSTR_END) {
936                     ast_type_to_string(exprs[0], ty1, sizeof(ty1));
937                     ast_type_to_string(exprs[1], ty2, sizeof(ty2));
938                     compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
939                 }
940                 else if (!ast_compare_type(exprs[0], exprs[1]))
941                 {
942                     ast_type_to_string(exprs[0], ty1, sizeof(ty1));
943                     ast_type_to_string(exprs[1], ty2, sizeof(ty2));
944                     if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) &&
945                         exprs[0]->vtype == TYPE_FUNCTION &&
946                         exprs[1]->vtype == TYPE_FUNCTION)
947                     {
948                         (void)!compile_warning(ctx, WARN_ASSIGN_FUNCTION_TYPES,
949                                                "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
950                     }
951                     else
952                         compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
953                 }
954             }
955             (void)check_write_to(ctx, exprs[0]);
956             /* When we're a vector of part of an entity field we use STOREP */
957             if (ast_istype(exprs[0], ast_member) && ast_istype(((ast_member*)exprs[0])->owner, ast_entfield))
958                 assignop = INSTR_STOREP_F;
959             out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
960             break;
961         case opid3('+','+','P'):
962         case opid3('-','-','P'):
963             /* prefix ++ */
964             if (exprs[0]->vtype != TYPE_FLOAT) {
965                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
966                 compile_error(ast_ctx(exprs[0]), "invalid type for prefix increment: %s", ty1);
967                 return false;
968             }
969             if (op->id == opid3('+','+','P'))
970                 addop = INSTR_ADD_F;
971             else
972                 addop = INSTR_SUB_F;
973             (void)check_write_to(ast_ctx(exprs[0]), exprs[0]);
974             if (ast_istype(exprs[0], ast_entfield)) {
975                 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
976                                                         exprs[0],
977                                                         (ast_expression*)parser->fold->imm_float[1]);
978             } else {
979                 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
980                                                         exprs[0],
981                                                         (ast_expression*)parser->fold->imm_float[1]);
982             }
983             break;
984         case opid3('S','+','+'):
985         case opid3('S','-','-'):
986             /* prefix ++ */
987             if (exprs[0]->vtype != TYPE_FLOAT) {
988                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
989                 compile_error(ast_ctx(exprs[0]), "invalid type for suffix increment: %s", ty1);
990                 return false;
991             }
992             if (op->id == opid3('S','+','+')) {
993                 addop = INSTR_ADD_F;
994                 subop = INSTR_SUB_F;
995             } else {
996                 addop = INSTR_SUB_F;
997                 subop = INSTR_ADD_F;
998             }
999             (void)check_write_to(ast_ctx(exprs[0]), exprs[0]);
1000             if (ast_istype(exprs[0], ast_entfield)) {
1001                 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
1002                                                         exprs[0],
1003                                                         (ast_expression*)parser->fold->imm_float[1]);
1004             } else {
1005                 out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
1006                                                         exprs[0],
1007                                                         (ast_expression*)parser->fold->imm_float[1]);
1008             }
1009             if (!out)
1010                 return false;
1011             out = fold_binary(ctx, subop,
1012                               out,
1013                               (ast_expression*)parser->fold->imm_float[1]);
1014
1015             break;
1016         case opid2('+','='):
1017         case opid2('-','='):
1018             if (exprs[0]->vtype != exprs[1]->vtype ||
1019                 (exprs[0]->vtype != TYPE_VECTOR && exprs[0]->vtype != TYPE_FLOAT) )
1020             {
1021                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1022                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1023                 compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s",
1024                               ty1, ty2);
1025                 return false;
1026             }
1027             (void)check_write_to(ctx, exprs[0]);
1028             if (ast_istype(exprs[0], ast_entfield))
1029                 assignop = type_storep_instr[exprs[0]->vtype];
1030             else
1031                 assignop = type_store_instr[exprs[0]->vtype];
1032             switch (exprs[0]->vtype) {
1033                 case TYPE_FLOAT:
1034                     out = (ast_expression*)ast_binstore_new(ctx, assignop,
1035                                                             (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F),
1036                                                             exprs[0], exprs[1]);
1037                     break;
1038                 case TYPE_VECTOR:
1039                     out = (ast_expression*)ast_binstore_new(ctx, assignop,
1040                                                             (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V),
1041                                                             exprs[0], exprs[1]);
1042                     break;
1043                 default:
1044                     compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s",
1045                                   type_name[exprs[0]->vtype],
1046                                   type_name[exprs[1]->vtype]);
1047                     return false;
1048             };
1049             break;
1050         case opid2('*','='):
1051         case opid2('/','='):
1052             if (exprs[1]->vtype != TYPE_FLOAT ||
1053                 !(exprs[0]->vtype == TYPE_FLOAT ||
1054                   exprs[0]->vtype == TYPE_VECTOR))
1055             {
1056                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1057                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1058                 compile_error(ctx, "invalid types used in expression: %s and %s",
1059                               ty1, ty2);
1060                 return false;
1061             }
1062             (void)check_write_to(ctx, exprs[0]);
1063             if (ast_istype(exprs[0], ast_entfield))
1064                 assignop = type_storep_instr[exprs[0]->vtype];
1065             else
1066                 assignop = type_store_instr[exprs[0]->vtype];
1067             switch (exprs[0]->vtype) {
1068                 case TYPE_FLOAT:
1069                     out = (ast_expression*)ast_binstore_new(ctx, assignop,
1070                                                             (op->id == opid2('*','=') ? INSTR_MUL_F : INSTR_DIV_F),
1071                                                             exprs[0], exprs[1]);
1072                     break;
1073                 case TYPE_VECTOR:
1074                     if (op->id == opid2('*','=')) {
1075                         out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_MUL_VF,
1076                                                                 exprs[0], exprs[1]);
1077                     } else {
1078                         out = fold_binary(ctx, INSTR_DIV_F,
1079                                          (ast_expression*)parser->fold->imm_float[1],
1080                                          exprs[1]);
1081                         if (!out) {
1082                             compile_error(ctx, "internal error: failed to generate division");
1083                             return false;
1084                         }
1085                         out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_MUL_VF,
1086                                                                 exprs[0], out);
1087                     }
1088                     break;
1089                 default:
1090                     compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s",
1091                                   type_name[exprs[0]->vtype],
1092                                   type_name[exprs[1]->vtype]);
1093                     return false;
1094             };
1095             break;
1096         case opid2('&','='):
1097         case opid2('|','='):
1098         case opid2('^','='):
1099             if (NotSameType(TYPE_FLOAT) && NotSameType(TYPE_VECTOR)) {
1100                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1101                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1102                 compile_error(ctx, "invalid types used in expression: %s and %s",
1103                               ty1, ty2);
1104                 return false;
1105             }
1106             (void)check_write_to(ctx, exprs[0]);
1107             if (ast_istype(exprs[0], ast_entfield))
1108                 assignop = type_storep_instr[exprs[0]->vtype];
1109             else
1110                 assignop = type_store_instr[exprs[0]->vtype];
1111             if (exprs[0]->vtype == TYPE_FLOAT)
1112                 out = (ast_expression*)ast_binstore_new(ctx, assignop,
1113                                                         (op->id == opid2('^','=') ? VINSTR_BITXOR : op->id == opid2('&','=') ? INSTR_BITAND : INSTR_BITOR),
1114                                                         exprs[0], exprs[1]);
1115             else
1116                 out = (ast_expression*)ast_binstore_new(ctx, assignop,
1117                                                         (op->id == opid2('^','=') ? VINSTR_BITXOR_V : op->id == opid2('&','=') ? VINSTR_BITAND_V : VINSTR_BITOR_V),
1118                                                         exprs[0], exprs[1]);
1119             break;
1120         case opid3('&','~','='):
1121             /* This is like: a &= ~(b);
1122              * But QC has no bitwise-not, so we implement it as
1123              * a -= a & (b);
1124              */
1125             if (NotSameType(TYPE_FLOAT) && NotSameType(TYPE_VECTOR)) {
1126                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1127                 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1128                 compile_error(ctx, "invalid types used in expression: %s and %s",
1129                               ty1, ty2);
1130                 return false;
1131             }
1132             if (ast_istype(exprs[0], ast_entfield))
1133                 assignop = type_storep_instr[exprs[0]->vtype];
1134             else
1135                 assignop = type_store_instr[exprs[0]->vtype];
1136             if (exprs[0]->vtype == TYPE_FLOAT)
1137                 out = fold_binary(ctx, INSTR_BITAND, exprs[0], exprs[1]);
1138             else
1139                 out = fold_binary(ctx, VINSTR_BITAND_V, exprs[0], exprs[1]);
1140             if (!out)
1141                 return false;
1142             (void)check_write_to(ctx, exprs[0]);
1143             if (exprs[0]->vtype == TYPE_FLOAT)
1144                 asbinstore = ast_binstore_new(ctx, assignop, INSTR_SUB_F, exprs[0], out);
1145             else
1146                 asbinstore = ast_binstore_new(ctx, assignop, INSTR_SUB_V, exprs[0], out);
1147             asbinstore->keep_dest = true;
1148             out = (ast_expression*)asbinstore;
1149             break;
1150
1151         case opid3('l', 'e', 'n'):
1152             if (exprs[0]->vtype != TYPE_STRING && exprs[0]->vtype != TYPE_ARRAY) {
1153                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1154                 compile_error(ast_ctx(exprs[0]), "invalid type for length operator: %s", ty1);
1155                 return false;
1156             }
1157             /* strings must be const, arrays are statically sized */
1158             if (exprs[0]->vtype == TYPE_STRING &&
1159                 !(((ast_value*)exprs[0])->hasvalue && ((ast_value*)exprs[0])->cvq == CV_CONST))
1160             {
1161                 compile_error(ast_ctx(exprs[0]), "operand of length operator not a valid constant expression");
1162                 return false;
1163             }
1164             out = fold_op(parser->fold, op, exprs);
1165             break;
1166
1167         case opid2('~', 'P'):
1168             if (exprs[0]->vtype != TYPE_FLOAT && exprs[0]->vtype != TYPE_VECTOR) {
1169                 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1170                 compile_error(ast_ctx(exprs[0]), "invalid type for bit not: %s", ty1);
1171                 return false;
1172             }
1173             if (!(out = fold_op(parser->fold, op, exprs))) {
1174                 if (exprs[0]->vtype == TYPE_FLOAT) {
1175                     out = fold_binary(ctx, INSTR_SUB_F, (ast_expression*)parser->fold->imm_float[2], exprs[0]);
1176                 } else {
1177                     out = fold_binary(ctx, INSTR_SUB_V, (ast_expression*)parser->fold->imm_vector[1], exprs[0]);
1178                 }
1179             }
1180             break;
1181     }
1182 #undef NotSameType
1183     if (!out) {
1184         compile_error(ctx, "failed to apply operator %s", op->op);
1185         return false;
1186     }
1187
1188     vec_push(sy->out, syexp(ctx, out));
1189     return true;
1190 }
1191
1192 static bool parser_close_call(parser_t *parser, shunt *sy)
1193 {
1194     /* was a function call */
1195     ast_expression *fun;
1196     ast_value      *funval = NULL;
1197     ast_call       *call;
1198
1199     size_t          fid;
1200     size_t          paramcount, i;
1201     bool            fold = true;
1202
1203     fid = vec_last(sy->ops).off;
1204     vec_shrinkby(sy->ops, 1);
1205
1206     /* out[fid] is the function
1207      * everything above is parameters...
1208      */
1209     if (!vec_size(sy->argc)) {
1210         parseerror(parser, "internal error: no argument counter available");
1211         return false;
1212     }
1213
1214     paramcount = vec_last(sy->argc);
1215     vec_pop(sy->argc);
1216
1217     if (vec_size(sy->out) < fid) {
1218         parseerror(parser, "internal error: broken function call%lu < %lu+%lu\n",
1219                    (unsigned long)vec_size(sy->out),
1220                    (unsigned long)fid,
1221                    (unsigned long)paramcount);
1222         return false;
1223     }
1224
1225     /*
1226      * TODO handle this at the intrinsic level with an ast_intrinsic
1227      * node and codegen.
1228      */
1229     if ((fun = sy->out[fid].out) == intrin_debug_typestring(parser->intrin)) {
1230         char ty[1024];
1231         if (fid+2 != vec_size(sy->out) ||
1232             vec_last(sy->out).block)
1233         {
1234             parseerror(parser, "intrinsic __builtin_debug_typestring requires exactly 1 parameter");
1235             return false;
1236         }
1237         ast_type_to_string(vec_last(sy->out).out, ty, sizeof(ty));
1238         ast_unref(vec_last(sy->out).out);
1239         sy->out[fid] = syexp(ast_ctx(vec_last(sy->out).out),
1240                              (ast_expression*)fold_constgen_string(parser->fold, ty, false));
1241         vec_shrinkby(sy->out, 1);
1242         return true;
1243     }
1244
1245     /*
1246      * Now we need to determine if the function that is being called is
1247      * an intrinsic so we can evaluate if the arguments to it are constant
1248      * and than fruitfully fold them.
1249      */
1250 #define fold_can_1(X)  \
1251     (ast_istype(((ast_expression*)(X)), ast_value) && (X)->hasvalue && ((X)->cvq == CV_CONST) && \
1252                 ((ast_expression*)(X))->vtype != TYPE_FUNCTION)
1253
1254     if (fid + 1 < vec_size(sy->out))
1255         ++paramcount;
1256
1257     for (i = 0; i < paramcount; ++i) {
1258         if (!fold_can_1((ast_value*)sy->out[fid + 1 + i].out)) {
1259             fold = false;
1260             break;
1261         }
1262     }
1263
1264     /*
1265      * All is well which ends well, if we make it into here we can ignore the
1266      * intrinsic call and just evaluate it i.e constant fold it.
1267      */
1268     if (fold && ast_istype(fun, ast_value) && ((ast_value*)fun)->intrinsic) {
1269         ast_expression **exprs  = NULL;
1270         ast_expression *foldval = NULL;
1271
1272         for (i = 0; i < paramcount; i++)
1273             vec_push(exprs, sy->out[fid+1 + i].out);
1274
1275         if (!(foldval = intrin_fold(parser->intrin, (ast_value*)fun, exprs))) {
1276             vec_free(exprs);
1277             goto fold_leave;
1278         }
1279
1280         /*
1281          * Blub: what sorts of unreffing and resizing of
1282          * sy->out should I be doing here?
1283          */
1284         sy->out[fid] = syexp(foldval->node.context, foldval);
1285         vec_shrinkby(sy->out, paramcount);
1286         vec_free(exprs);
1287
1288         return true;
1289     }
1290
1291     fold_leave:
1292     call = ast_call_new(sy->ops[vec_size(sy->ops)].ctx, fun);
1293
1294     if (!call)
1295         return false;
1296
1297     if (fid+1 + paramcount != vec_size(sy->out)) {
1298         parseerror(parser, "internal error: parameter count mismatch: (%lu+1+%lu), %lu",
1299                    (unsigned long)fid, (unsigned long)paramcount, (unsigned long)vec_size(sy->out));
1300         return false;
1301     }
1302
1303     for (i = 0; i < paramcount; ++i)
1304         vec_push(call->params, sy->out[fid+1 + i].out);
1305     vec_shrinkby(sy->out, paramcount);
1306     (void)!ast_call_check_types(call, parser->function->vtype->expression.varparam);
1307     if (parser->max_param_count < paramcount)
1308         parser->max_param_count = paramcount;
1309
1310     if (ast_istype(fun, ast_value)) {
1311         funval = (ast_value*)fun;
1312         if ((fun->flags & AST_FLAG_VARIADIC) &&
1313             !(/*funval->cvq == CV_CONST && */ funval->hasvalue && funval->constval.vfunc->builtin))
1314         {
1315             call->va_count = (ast_expression*)fold_constgen_float(parser->fold, (qcfloat_t)paramcount, false);
1316         }
1317     }
1318
1319     /* overwrite fid, the function, with a call */
1320     sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
1321
1322     if (fun->vtype != TYPE_FUNCTION) {
1323         parseerror(parser, "not a function (%s)", type_name[fun->vtype]);
1324         return false;
1325     }
1326
1327     if (!fun->next) {
1328         parseerror(parser, "could not determine function return type");
1329         return false;
1330     } else {
1331         ast_value *fval = (ast_istype(fun, ast_value) ? ((ast_value*)fun) : NULL);
1332
1333         if (fun->flags & AST_FLAG_DEPRECATED) {
1334             if (!fval) {
1335                 return !parsewarning(parser, WARN_DEPRECATED,
1336                         "call to function (which is marked deprecated)\n",
1337                         "-> it has been declared here: %s:%i",
1338                         ast_ctx(fun).file, ast_ctx(fun).line);
1339             }
1340             if (!fval->desc) {
1341                 return !parsewarning(parser, WARN_DEPRECATED,
1342                         "call to `%s` (which is marked deprecated)\n"
1343                         "-> `%s` declared here: %s:%i",
1344                         fval->name, fval->name, ast_ctx(fun).file, ast_ctx(fun).line);
1345             }
1346             return !parsewarning(parser, WARN_DEPRECATED,
1347                     "call to `%s` (deprecated: %s)\n"
1348                     "-> `%s` declared here: %s:%i",
1349                     fval->name, fval->desc, fval->name, ast_ctx(fun).file,
1350                     ast_ctx(fun).line);
1351         }
1352
1353         if (vec_size(fun->params) != paramcount &&
1354             !((fun->flags & AST_FLAG_VARIADIC) &&
1355               vec_size(fun->params) < paramcount))
1356         {
1357             const char *fewmany = (vec_size(fun->params) > paramcount) ? "few" : "many";
1358             if (fval)
1359                 return !parsewarning(parser, WARN_INVALID_PARAMETER_COUNT,
1360                                      "too %s parameters for call to %s: expected %i, got %i\n"
1361                                      " -> `%s` has been declared here: %s:%i",
1362                                      fewmany, fval->name, (int)vec_size(fun->params), (int)paramcount,
1363                                      fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line);
1364             else
1365                 return !parsewarning(parser, WARN_INVALID_PARAMETER_COUNT,
1366                                      "too %s parameters for function call: expected %i, got %i\n"
1367                                      " -> it has been declared here: %s:%i",
1368                                      fewmany, (int)vec_size(fun->params), (int)paramcount,
1369                                      ast_ctx(fun).file, (int)ast_ctx(fun).line);
1370         }
1371     }
1372
1373     return true;
1374 }
1375
1376 static bool parser_close_paren(parser_t *parser, shunt *sy)
1377 {
1378     if (!vec_size(sy->ops)) {
1379         parseerror(parser, "unmatched closing paren");
1380         return false;
1381     }
1382
1383     while (vec_size(sy->ops)) {
1384         if (vec_last(sy->ops).isparen) {
1385             if (vec_last(sy->paren) == PAREN_FUNC) {
1386                 vec_pop(sy->paren);
1387                 if (!parser_close_call(parser, sy))
1388                     return false;
1389                 break;
1390             }
1391             if (vec_last(sy->paren) == PAREN_EXPR) {
1392                 vec_pop(sy->paren);
1393                 if (!vec_size(sy->out)) {
1394                     compile_error(vec_last(sy->ops).ctx, "empty paren expression");
1395                     vec_shrinkby(sy->ops, 1);
1396                     return false;
1397                 }
1398                 vec_shrinkby(sy->ops, 1);
1399                 break;
1400             }
1401             if (vec_last(sy->paren) == PAREN_INDEX) {
1402                 vec_pop(sy->paren);
1403                 /* pop off the parenthesis */
1404                 vec_shrinkby(sy->ops, 1);
1405                 /* then apply the index operator */
1406                 if (!parser_sy_apply_operator(parser, sy))
1407                     return false;
1408                 break;
1409             }
1410             if (vec_last(sy->paren) == PAREN_TERNARY1) {
1411                 vec_last(sy->paren) = PAREN_TERNARY2;
1412                 /* pop off the parenthesis */
1413                 vec_shrinkby(sy->ops, 1);
1414                 break;
1415             }
1416             compile_error(vec_last(sy->ops).ctx, "invalid parenthesis");
1417             return false;
1418         }
1419         if (!parser_sy_apply_operator(parser, sy))
1420             return false;
1421     }
1422     return true;
1423 }
1424
1425 static void parser_reclassify_token(parser_t *parser)
1426 {
1427     size_t i;
1428     if (parser->tok >= TOKEN_START)
1429         return;
1430     for (i = 0; i < operator_count; ++i) {
1431         if (!strcmp(parser_tokval(parser), operators[i].op)) {
1432             parser->tok = TOKEN_OPERATOR;
1433             return;
1434         }
1435     }
1436 }
1437
1438 static ast_expression* parse_vararg_do(parser_t *parser)
1439 {
1440     ast_expression *idx, *out;
1441     ast_value      *typevar;
1442     ast_value      *funtype = parser->function->vtype;
1443     lex_ctx_t         ctx     = parser_ctx(parser);
1444
1445     if (!parser->function->varargs) {
1446         parseerror(parser, "function has no variable argument list");
1447         return NULL;
1448     }
1449
1450     if (!parser_next(parser) || parser->tok != '(') {
1451         parseerror(parser, "expected parameter index and type in parenthesis");
1452         return NULL;
1453     }
1454     if (!parser_next(parser)) {
1455         parseerror(parser, "error parsing parameter index");
1456         return NULL;
1457     }
1458
1459     idx = parse_expression_leave(parser, true, false, false);
1460     if (!idx)
1461         return NULL;
1462
1463     if (parser->tok != ',') {
1464         if (parser->tok != ')') {
1465             ast_unref(idx);
1466             parseerror(parser, "expected comma after parameter index");
1467             return NULL;
1468         }
1469         /* vararg piping: ...(start) */
1470         out = (ast_expression*)ast_argpipe_new(ctx, idx);
1471         return out;
1472     }
1473
1474     if (!parser_next(parser) || (parser->tok != TOKEN_IDENT && parser->tok != TOKEN_TYPENAME)) {
1475         ast_unref(idx);
1476         parseerror(parser, "expected typename for vararg");
1477         return NULL;
1478     }
1479
1480     typevar = parse_typename(parser, NULL, NULL, NULL);
1481     if (!typevar) {
1482         ast_unref(idx);
1483         return NULL;
1484     }
1485
1486     if (parser->tok != ')') {
1487         ast_unref(idx);
1488         ast_delete(typevar);
1489         parseerror(parser, "expected closing paren");
1490         return NULL;
1491     }
1492
1493     if (funtype->expression.varparam &&
1494         !ast_compare_type((ast_expression*)typevar, (ast_expression*)funtype->expression.varparam))
1495     {
1496         char ty1[1024];
1497         char ty2[1024];
1498         ast_type_to_string((ast_expression*)typevar, ty1, sizeof(ty1));
1499         ast_type_to_string((ast_expression*)funtype->expression.varparam, ty2, sizeof(ty2));
1500         compile_error(ast_ctx(typevar),
1501                       "function was declared to take varargs of type `%s`, requested type is: %s",
1502                       ty2, ty1);
1503     }
1504
1505     out = (ast_expression*)ast_array_index_new(ctx, (ast_expression*)(parser->function->varargs), idx);
1506     ast_type_adopt(out, typevar);
1507     ast_delete(typevar);
1508     return out;
1509 }
1510
1511 static ast_expression* parse_vararg(parser_t *parser)
1512 {
1513     bool           old_noops = parser->lex->flags.noops;
1514
1515     ast_expression *out;
1516
1517     parser->lex->flags.noops = true;
1518     out = parse_vararg_do(parser);
1519
1520     parser->lex->flags.noops = old_noops;
1521     return out;
1522 }
1523
1524 /* not to be exposed */
1525 bool ftepp_predef_exists(const char *name);
1526 static bool parse_sya_operand(parser_t *parser, shunt *sy, bool with_labels)
1527 {
1528     if (OPTS_FLAG(TRANSLATABLE_STRINGS) &&
1529         parser->tok == TOKEN_IDENT &&
1530         !strcmp(parser_tokval(parser), "_"))
1531     {
1532         /* a translatable string */
1533         ast_value *val;
1534
1535         parser->lex->flags.noops = true;
1536         if (!parser_next(parser) || parser->tok != '(') {
1537             parseerror(parser, "use _(\"string\") to create a translatable string constant");
1538             return false;
1539         }
1540         parser->lex->flags.noops = false;
1541         if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
1542             parseerror(parser, "expected a constant string in translatable-string extension");
1543             return false;
1544         }
1545         val = (ast_value*)fold_constgen_string(parser->fold, parser_tokval(parser), true);
1546         if (!val)
1547             return false;
1548         vec_push(sy->out, syexp(parser_ctx(parser), (ast_expression*)val));
1549
1550         if (!parser_next(parser) || parser->tok != ')') {
1551             parseerror(parser, "expected closing paren after translatable string");
1552             return false;
1553         }
1554         return true;
1555     }
1556     else if (parser->tok == TOKEN_DOTS)
1557     {
1558         ast_expression *va;
1559         if (!OPTS_FLAG(VARIADIC_ARGS)) {
1560             parseerror(parser, "cannot access varargs (try -fvariadic-args)");
1561             return false;
1562         }
1563         va = parse_vararg(parser);
1564         if (!va)
1565             return false;
1566         vec_push(sy->out, syexp(parser_ctx(parser), va));
1567         return true;
1568     }
1569     else if (parser->tok == TOKEN_FLOATCONST) {
1570         ast_expression *val = fold_constgen_float(parser->fold, (parser_token(parser)->constval.f), false);
1571         if (!val)
1572             return false;
1573         vec_push(sy->out, syexp(parser_ctx(parser), val));
1574         return true;
1575     }
1576     else if (parser->tok == TOKEN_INTCONST || parser->tok == TOKEN_CHARCONST) {
1577         ast_expression *val = fold_constgen_float(parser->fold, (qcfloat_t)(parser_token(parser)->constval.i), false);
1578         if (!val)
1579             return false;
1580         vec_push(sy->out, syexp(parser_ctx(parser), val));
1581         return true;
1582     }
1583     else if (parser->tok == TOKEN_STRINGCONST) {
1584         ast_expression *val = fold_constgen_string(parser->fold, parser_tokval(parser), false);
1585         if (!val)
1586             return false;
1587         vec_push(sy->out, syexp(parser_ctx(parser), val));
1588         return true;
1589     }
1590     else if (parser->tok == TOKEN_VECTORCONST) {
1591         ast_expression *val = fold_constgen_vector(parser->fold, parser_token(parser)->constval.v);
1592         if (!val)
1593             return false;
1594         vec_push(sy->out, syexp(parser_ctx(parser), val));
1595         return true;
1596     }
1597     else if (parser->tok == TOKEN_IDENT)
1598     {
1599         const char     *ctoken = parser_tokval(parser);
1600         ast_expression *prev = vec_size(sy->out) ? vec_last(sy->out).out : NULL;
1601         ast_expression *var;
1602         /* a_vector.{x,y,z} */
1603         if (!vec_size(sy->ops) ||
1604             !vec_last(sy->ops).etype ||
1605             operators[vec_last(sy->ops).etype-1].id != opid1('.'))
1606         {
1607             /* When adding more intrinsics, fix the above condition */
1608             prev = NULL;
1609         }
1610         if (prev && prev->vtype == TYPE_VECTOR && ctoken[0] >= 'x' && ctoken[0] <= 'z' && !ctoken[1])
1611         {
1612             var = (ast_expression*)parser->const_vec[ctoken[0]-'x'];
1613         } else {
1614             var = parser_find_var(parser, parser_tokval(parser));
1615             if (!var)
1616                 var = parser_find_field(parser, parser_tokval(parser));
1617         }
1618         if (!var && with_labels) {
1619             var = (ast_expression*)parser_find_label(parser, parser_tokval(parser));
1620             if (!with_labels) {
1621                 ast_label *lbl = ast_label_new(parser_ctx(parser), parser_tokval(parser), true);
1622                 var = (ast_expression*)lbl;
1623                 vec_push(parser->labels, lbl);
1624             }
1625         }
1626         if (!var && !strcmp(parser_tokval(parser), "__FUNC__"))
1627             var = (ast_expression*)fold_constgen_string(parser->fold, parser->function->name, false);
1628         if (!var) {
1629             /*
1630              * now we try for the real intrinsic hashtable. If the string
1631              * begins with __builtin, we simply skip past it, otherwise we
1632              * use the identifier as is.
1633              */
1634             if (!strncmp(parser_tokval(parser), "__builtin_", 10)) {
1635                 var = intrin_func(parser->intrin, parser_tokval(parser));
1636             }
1637
1638             /*
1639              * Try it again, intrin_func deals with the alias method as well
1640              * the first one masks for __builtin though, we emit warning here.
1641              */
1642             if (!var) {
1643                 if ((var = intrin_func(parser->intrin, parser_tokval(parser)))) {
1644                     (void)!compile_warning(
1645                         parser_ctx(parser),
1646                         WARN_BUILTINS,
1647                         "using implicitly defined builtin `__builtin_%s' for `%s'",
1648                         parser_tokval(parser),
1649                         parser_tokval(parser)
1650                     );
1651                 }
1652             }
1653
1654
1655             if (!var) {
1656                 char *correct = NULL;
1657                 size_t i;
1658
1659                 /*
1660                  * sometimes people use preprocessing predefs without enabling them
1661                  * i've done this thousands of times already myself.  Lets check for
1662                  * it in the predef table.  And diagnose it better :)
1663                  */
1664                 if (!OPTS_FLAG(FTEPP_PREDEFS) && ftepp_predef_exists(parser_tokval(parser))) {
1665                     parseerror(parser, "unexpected identifier: %s (use -fftepp-predef to enable pre-defined macros)", parser_tokval(parser));
1666                     return false;
1667                 }
1668
1669                 /*
1670                  * TODO: determine the best score for the identifier: be it
1671                  * a variable, a field.
1672                  *
1673                  * We should also consider adding correction tables for
1674                  * other things as well.
1675                  */
1676                 if (OPTS_OPTION_BOOL(OPTION_CORRECTION) && strlen(parser_tokval(parser)) <= 16) {
1677                     correction_t corr;
1678                     correct_init(&corr);
1679
1680                     for (i = 0; i < vec_size(parser->correct_variables); i++) {
1681                         correct = correct_str(&corr, parser->correct_variables[i], parser_tokval(parser));
1682                         if (strcmp(correct, parser_tokval(parser))) {
1683                             break;
1684                         } else  {
1685                             mem_d(correct);
1686                             correct = NULL;
1687                         }
1688                     }
1689                     correct_free(&corr);
1690
1691                     if (correct) {
1692                         parseerror(parser, "unexpected identifier: %s (did you mean %s?)", parser_tokval(parser), correct);
1693                         mem_d(correct);
1694                         return false;
1695                     }
1696                 }
1697                 parseerror(parser, "unexpected identifier: %s", parser_tokval(parser));
1698                 return false;
1699             }
1700         }
1701         else
1702         {
1703             if (ast_istype(var, ast_value)) {
1704                 ((ast_value*)var)->uses++;
1705             }
1706             else if (ast_istype(var, ast_member)) {
1707                 ast_member *mem = (ast_member*)var;
1708                 if (ast_istype(mem->owner, ast_value))
1709                     ((ast_value*)(mem->owner))->uses++;
1710             }
1711         }
1712         vec_push(sy->out, syexp(parser_ctx(parser), var));
1713         return true;
1714     }
1715     parseerror(parser, "unexpected token `%s`", parser_tokval(parser));
1716     return false;
1717 }
1718
1719 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma, bool truthvalue, bool with_labels)
1720 {
1721     ast_expression *expr = NULL;
1722     shunt sy;
1723     size_t i;
1724     bool wantop = false;
1725     /* only warn once about an assignment in a truth value because the current code
1726      * would trigger twice on: if(a = b && ...), once for the if-truth-value, once for the && part
1727      */
1728     bool warn_parenthesis = true;
1729
1730     /* count the parens because an if starts with one, so the
1731      * end of a condition is an unmatched closing paren
1732      */
1733     int ternaries = 0;
1734
1735     memset(&sy, 0, sizeof(sy));
1736
1737     parser->lex->flags.noops = false;
1738
1739     parser_reclassify_token(parser);
1740
1741     while (true)
1742     {
1743         if (parser->tok == TOKEN_TYPENAME) {
1744             parseerror(parser, "unexpected typename `%s`", parser_tokval(parser));
1745             goto onerr;
1746         }
1747
1748         if (parser->tok == TOKEN_OPERATOR)
1749         {
1750             /* classify the operator */
1751             const oper_info *op;
1752             const oper_info *olast = NULL;
1753             size_t o;
1754             for (o = 0; o < operator_count; ++o) {
1755                 if (((!(operators[o].flags & OP_PREFIX) == !!wantop)) &&
1756                     /* !(operators[o].flags & OP_SUFFIX) && / * remove this */
1757                     !strcmp(parser_tokval(parser), operators[o].op))
1758                 {
1759                     break;
1760                 }
1761             }
1762             if (o == operator_count) {
1763                 compile_error(parser_ctx(parser), "unexpected operator: %s", parser_tokval(parser));
1764                 goto onerr;
1765             }
1766             /* found an operator */
1767             op = &operators[o];
1768
1769             /* when declaring variables, a comma starts a new variable */
1770             if (op->id == opid1(',') && !vec_size(sy.paren) && stopatcomma) {
1771                 /* fixup the token */
1772                 parser->tok = ',';
1773                 break;
1774             }
1775
1776             /* a colon without a pervious question mark cannot be a ternary */
1777             if (!ternaries && op->id == opid2(':','?')) {
1778                 parser->tok = ':';
1779                 break;
1780             }
1781
1782             if (op->id == opid1(',')) {
1783                 if (vec_size(sy.paren) && vec_last(sy.paren) == PAREN_TERNARY2) {
1784                     (void)!parsewarning(parser, WARN_TERNARY_PRECEDENCE, "suggesting parenthesis around ternary expression");
1785                 }
1786             }
1787
1788             if (vec_size(sy.ops) && !vec_last(sy.ops).isparen)
1789                 olast = &operators[vec_last(sy.ops).etype-1];
1790
1791             /* first only apply higher precedences, assoc_left+equal comes after we warn about precedence rules */
1792             while (olast && op->prec < olast->prec)
1793             {
1794                 if (!parser_sy_apply_operator(parser, &sy))
1795                     goto onerr;
1796                 if (vec_size(sy.ops) && !vec_last(sy.ops).isparen)
1797                     olast = &operators[vec_last(sy.ops).etype-1];
1798                 else
1799                     olast = NULL;
1800             }
1801
1802 #define IsAssignOp(x) (\
1803                 (x) == opid1('=') || \
1804                 (x) == opid2('+','=') || \
1805                 (x) == opid2('-','=') || \
1806                 (x) == opid2('*','=') || \
1807                 (x) == opid2('/','=') || \
1808                 (x) == opid2('%','=') || \
1809                 (x) == opid2('&','=') || \
1810                 (x) == opid2('|','=') || \
1811                 (x) == opid3('&','~','=') \
1812                 )
1813             if (warn_parenthesis) {
1814                 if ( (olast && IsAssignOp(olast->id) && (op->id == opid2('&','&') || op->id == opid2('|','|'))) ||
1815                      (olast && IsAssignOp(op->id) && (olast->id == opid2('&','&') || olast->id == opid2('|','|'))) ||
1816                      (truthvalue && !vec_size(sy.paren) && IsAssignOp(op->id))
1817                    )
1818                 {
1819                     (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around assignment used as truth value");
1820                     warn_parenthesis = false;
1821                 }
1822
1823                 if (olast && olast->id != op->id) {
1824                     if ((op->id    == opid1('&') || op->id    == opid1('|') || op->id    == opid1('^')) &&
1825                         (olast->id == opid1('&') || olast->id == opid1('|') || olast->id == opid1('^')))
1826                     {
1827                         (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around bitwise operations");
1828                         warn_parenthesis = false;
1829                     }
1830                     else if ((op->id    == opid2('&','&') || op->id    == opid2('|','|')) &&
1831                              (olast->id == opid2('&','&') || olast->id == opid2('|','|')))
1832                     {
1833                         (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around logical operations");
1834                         warn_parenthesis = false;
1835                     }
1836                 }
1837             }
1838
1839             while (olast && (
1840                     (op->prec < olast->prec) ||
1841                     (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1842             {
1843                 if (!parser_sy_apply_operator(parser, &sy))
1844                     goto onerr;
1845                 if (vec_size(sy.ops) && !vec_last(sy.ops).isparen)
1846                     olast = &operators[vec_last(sy.ops).etype-1];
1847                 else
1848                     olast = NULL;
1849             }
1850
1851             if (op->id == opid1('(')) {
1852                 if (wantop) {
1853                     size_t sycount = vec_size(sy.out);
1854                     /* we expected an operator, this is the function-call operator */
1855                     vec_push(sy.paren, PAREN_FUNC);
1856                     vec_push(sy.ops, syparen(parser_ctx(parser), sycount-1));
1857                     vec_push(sy.argc, 0);
1858                 } else {
1859                     vec_push(sy.paren, PAREN_EXPR);
1860                     vec_push(sy.ops, syparen(parser_ctx(parser), 0));
1861                 }
1862                 wantop = false;
1863             } else if (op->id == opid1('[')) {
1864                 if (!wantop) {
1865                     parseerror(parser, "unexpected array subscript");
1866                     goto onerr;
1867                 }
1868                 vec_push(sy.paren, PAREN_INDEX);
1869                 /* push both the operator and the paren, this makes life easier */
1870                 vec_push(sy.ops, syop(parser_ctx(parser), op));
1871                 vec_push(sy.ops, syparen(parser_ctx(parser), 0));
1872                 wantop = false;
1873             } else if (op->id == opid2('?',':')) {
1874                 vec_push(sy.ops, syop(parser_ctx(parser), op));
1875                 vec_push(sy.ops, syparen(parser_ctx(parser), 0));
1876                 wantop = false;
1877                 ++ternaries;
1878                 vec_push(sy.paren, PAREN_TERNARY1);
1879             } else if (op->id == opid2(':','?')) {
1880                 if (!vec_size(sy.paren)) {
1881                     parseerror(parser, "unexpected colon outside ternary expression (missing parenthesis?)");
1882                     goto onerr;
1883                 }
1884                 if (vec_last(sy.paren) != PAREN_TERNARY1) {
1885                     parseerror(parser, "unexpected colon outside ternary expression (missing parenthesis?)");
1886                     goto onerr;
1887                 }
1888                 if (!parser_close_paren(parser, &sy))
1889                     goto onerr;
1890                 vec_push(sy.ops, syop(parser_ctx(parser), op));
1891                 wantop = false;
1892                 --ternaries;
1893             } else {
1894                 vec_push(sy.ops, syop(parser_ctx(parser), op));
1895                 wantop = !!(op->flags & OP_SUFFIX);
1896             }
1897         }
1898         else if (parser->tok == ')') {
1899             while (vec_size(sy.paren) && vec_last(sy.paren) == PAREN_TERNARY2) {
1900                 if (!parser_sy_apply_operator(parser, &sy))
1901                     goto onerr;
1902             }
1903             if (!vec_size(sy.paren))
1904                 break;
1905             if (wantop) {
1906                 if (vec_last(sy.paren) == PAREN_TERNARY1) {
1907                     parseerror(parser, "mismatched parentheses (closing paren in ternary expression?)");
1908                     goto onerr;
1909                 }
1910                 if (!parser_close_paren(parser, &sy))
1911                     goto onerr;
1912             } else {
1913                 /* must be a function call without parameters */
1914                 if (vec_last(sy.paren) != PAREN_FUNC) {
1915                     parseerror(parser, "closing paren in invalid position");
1916                     goto onerr;
1917                 }
1918                 if (!parser_close_paren(parser, &sy))
1919                     goto onerr;
1920             }
1921             wantop = true;
1922         }
1923         else if (parser->tok == '(') {
1924             parseerror(parser, "internal error: '(' should be classified as operator");
1925             goto onerr;
1926         }
1927         else if (parser->tok == '[') {
1928             parseerror(parser, "internal error: '[' should be classified as operator");
1929             goto onerr;
1930         }
1931         else if (parser->tok == ']') {
1932             while (vec_size(sy.paren) && vec_last(sy.paren) == PAREN_TERNARY2) {
1933                 if (!parser_sy_apply_operator(parser, &sy))
1934                     goto onerr;
1935             }
1936             if (!vec_size(sy.paren))
1937                 break;
1938             if (vec_last(sy.paren) != PAREN_INDEX) {
1939                 parseerror(parser, "mismatched parentheses, unexpected ']'");
1940                 goto onerr;
1941             }
1942             if (!parser_close_paren(parser, &sy))
1943                 goto onerr;
1944             wantop = true;
1945         }
1946         else if (!wantop) {
1947             if (!parse_sya_operand(parser, &sy, with_labels))
1948                 goto onerr;
1949 #if 0
1950             if (vec_size(sy.paren) && vec_last(sy.ops).isparen && vec_last(sy.paren) == PAREN_FUNC)
1951                 vec_last(sy.argc)++;
1952 #endif
1953             wantop = true;
1954         }
1955         else {
1956             /* in this case we might want to allow constant string concatenation */
1957             bool concatenated = false;
1958             if (parser->tok == TOKEN_STRINGCONST && vec_size(sy.out)) {
1959                 ast_expression *lexpr = vec_last(sy.out).out;
1960                 if (ast_istype(lexpr, ast_value)) {
1961                     ast_value *last = (ast_value*)lexpr;
1962                     if (last->isimm == true && last->cvq == CV_CONST &&
1963                         last->hasvalue && last->expression.vtype == TYPE_STRING)
1964                     {
1965                         char *newstr = NULL;
1966                         util_asprintf(&newstr, "%s%s", last->constval.vstring, parser_tokval(parser));
1967                         vec_last(sy.out).out = (ast_expression*)fold_constgen_string(parser->fold, newstr, false);
1968                         mem_d(newstr);
1969                         concatenated = true;
1970                     }
1971                 }
1972             }
1973             if (!concatenated) {
1974                 parseerror(parser, "expected operator or end of statement");
1975                 goto onerr;
1976             }
1977         }
1978
1979         if (!parser_next(parser)) {
1980             goto onerr;
1981         }
1982         if (parser->tok == ';' ||
1983             ((!vec_size(sy.paren) || (vec_size(sy.paren) == 1 && vec_last(sy.paren) == PAREN_TERNARY2)) &&
1984             (parser->tok == ']' || parser->tok == ')' || parser->tok == '}')))
1985         {
1986             break;
1987         }
1988     }
1989
1990     while (vec_size(sy.ops)) {
1991         if (!parser_sy_apply_operator(parser, &sy))
1992             goto onerr;
1993     }
1994
1995     parser->lex->flags.noops = true;
1996     if (vec_size(sy.out) != 1) {
1997         parseerror(parser, "expression expected");
1998         expr = NULL;
1999     } else
2000         expr = sy.out[0].out;
2001     vec_free(sy.out);
2002     vec_free(sy.ops);
2003     if (vec_size(sy.paren)) {
2004         parseerror(parser, "internal error: vec_size(sy.paren) = %lu", (unsigned long)vec_size(sy.paren));
2005         return NULL;
2006     }
2007     vec_free(sy.paren);
2008     vec_free(sy.argc);
2009     return expr;
2010
2011 onerr:
2012     parser->lex->flags.noops = true;
2013     for (i = 0; i < vec_size(sy.out); ++i) {
2014         if (sy.out[i].out)
2015             ast_unref(sy.out[i].out);
2016     }
2017     vec_free(sy.out);
2018     vec_free(sy.ops);
2019     vec_free(sy.paren);
2020     vec_free(sy.argc);
2021     return NULL;
2022 }
2023
2024 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma, bool with_labels)
2025 {
2026     ast_expression *e = parse_expression_leave(parser, stopatcomma, false, with_labels);
2027     if (!e)
2028         return NULL;
2029     if (parser->tok != ';') {
2030         parseerror(parser, "semicolon expected after expression");
2031         ast_unref(e);
2032         return NULL;
2033     }
2034     if (!parser_next(parser)) {
2035         ast_unref(e);
2036         return NULL;
2037     }
2038     return e;
2039 }
2040
2041 static void parser_enterblock(parser_t *parser)
2042 {
2043     vec_push(parser->variables, util_htnew(PARSER_HT_SIZE));
2044     vec_push(parser->_blocklocals, vec_size(parser->_locals));
2045     vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE));
2046     vec_push(parser->_blocktypedefs, vec_size(parser->_typedefs));
2047     vec_push(parser->_block_ctx, parser_ctx(parser));
2048
2049     /* corrector */
2050     vec_push(parser->correct_variables, correct_trie_new());
2051     vec_push(parser->correct_variables_score, NULL);
2052 }
2053
2054 static bool parser_leaveblock(parser_t *parser)
2055 {
2056     bool   rv = true;
2057     size_t locals, typedefs;
2058
2059     if (vec_size(parser->variables) <= PARSER_HT_LOCALS) {
2060         parseerror(parser, "internal error: parser_leaveblock with no block");
2061         return false;
2062     }
2063
2064     util_htdel(vec_last(parser->variables));
2065     correct_del(vec_last(parser->correct_variables), vec_last(parser->correct_variables_score));
2066
2067     vec_pop(parser->variables);
2068     vec_pop(parser->correct_variables);
2069     vec_pop(parser->correct_variables_score);
2070     if (!vec_size(parser->_blocklocals)) {
2071         parseerror(parser, "internal error: parser_leaveblock with no block (2)");
2072         return false;
2073     }
2074
2075     locals = vec_last(parser->_blocklocals);
2076     vec_pop(parser->_blocklocals);
2077     while (vec_size(parser->_locals) != locals) {
2078         ast_expression *e = vec_last(parser->_locals);
2079         ast_value      *v = (ast_value*)e;
2080         vec_pop(parser->_locals);
2081         if (ast_istype(e, ast_value) && !v->uses) {
2082             if (compile_warning(ast_ctx(v), WARN_UNUSED_VARIABLE, "unused variable: `%s`", v->name))
2083                 rv = false;
2084         }
2085     }
2086
2087     typedefs = vec_last(parser->_blocktypedefs);
2088     while (vec_size(parser->_typedefs) != typedefs) {
2089         ast_delete(vec_last(parser->_typedefs));
2090         vec_pop(parser->_typedefs);
2091     }
2092     util_htdel(vec_last(parser->typedefs));
2093     vec_pop(parser->typedefs);
2094
2095     vec_pop(parser->_block_ctx);
2096
2097     return rv;
2098 }
2099
2100 static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e)
2101 {
2102     vec_push(parser->_locals, e);
2103     util_htset(vec_last(parser->variables), name, (void*)e);
2104
2105     /* corrector */
2106     correct_add (
2107          vec_last(parser->correct_variables),
2108         &vec_last(parser->correct_variables_score),
2109         name
2110     );
2111 }
2112
2113 static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e)
2114 {
2115     vec_push(parser->globals, e);
2116     util_htset(parser->htglobals, name, e);
2117
2118     /* corrector */
2119     correct_add (
2120          parser->correct_variables[0],
2121         &parser->correct_variables_score[0],
2122         name
2123     );
2124 }
2125
2126 static ast_expression* process_condition(parser_t *parser, ast_expression *cond, bool *_ifnot)
2127 {
2128     bool       ifnot = false;
2129     ast_unary *unary;
2130     ast_expression *prev;
2131
2132     if (cond->vtype == TYPE_VOID || cond->vtype >= TYPE_VARIANT) {
2133         char ty[1024];
2134         ast_type_to_string(cond, ty, sizeof(ty));
2135         compile_error(ast_ctx(cond), "invalid type for if() condition: %s", ty);
2136     }
2137
2138     if (OPTS_FLAG(FALSE_EMPTY_STRINGS) && cond->vtype == TYPE_STRING)
2139     {
2140         prev = cond;
2141         cond = (ast_expression*)ast_unary_new(ast_ctx(cond), INSTR_NOT_S, cond);
2142         if (!cond) {
2143             ast_unref(prev);
2144             parseerror(parser, "internal error: failed to process condition");
2145             return NULL;
2146         }
2147         ifnot = !ifnot;
2148     }
2149     else if (OPTS_FLAG(CORRECT_LOGIC) && cond->vtype == TYPE_VECTOR)
2150     {
2151         /* vector types need to be cast to true booleans */
2152         ast_binary *bin = (ast_binary*)cond;
2153         if (!OPTS_FLAG(PERL_LOGIC) || !ast_istype(cond, ast_binary) || !(bin->op == INSTR_AND || bin->op == INSTR_OR))
2154         {
2155             /* in perl-logic, AND and OR take care of the -fcorrect-logic */
2156             prev = cond;
2157             cond = (ast_expression*)ast_unary_new(ast_ctx(cond), INSTR_NOT_V, cond);
2158             if (!cond) {
2159                 ast_unref(prev);
2160                 parseerror(parser, "internal error: failed to process condition");
2161                 return NULL;
2162             }
2163             ifnot = !ifnot;
2164         }
2165     }
2166
2167     unary = (ast_unary*)cond;
2168     /* ast_istype dereferences cond, should test here for safety */
2169     while (cond && ast_istype(cond, ast_unary) && unary->op == INSTR_NOT_F)
2170     {
2171         cond = unary->operand;
2172         unary->operand = NULL;
2173         ast_delete(unary);
2174         ifnot = !ifnot;
2175         unary = (ast_unary*)cond;
2176     }
2177
2178     if (!cond)
2179         parseerror(parser, "internal error: failed to process condition");
2180
2181     if (ifnot) *_ifnot = !*_ifnot;
2182     return cond;
2183 }
2184
2185 static bool parse_if(parser_t *parser, ast_block *block, ast_expression **out)
2186 {
2187     ast_ifthen *ifthen;
2188     ast_expression *cond, *ontrue = NULL, *onfalse = NULL;
2189     bool ifnot = false;
2190
2191     lex_ctx_t ctx = parser_ctx(parser);
2192
2193     (void)block; /* not touching */
2194
2195     /* skip the 'if', parse an optional 'not' and check for an opening paren */
2196     if (!parser_next(parser)) {
2197         parseerror(parser, "expected condition or 'not'");
2198         return false;
2199     }
2200     if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "not")) {
2201         ifnot = true;
2202         if (!parser_next(parser)) {
2203             parseerror(parser, "expected condition in parenthesis");
2204             return false;
2205         }
2206     }
2207     if (parser->tok != '(') {
2208         parseerror(parser, "expected 'if' condition in parenthesis");
2209         return false;
2210     }
2211     /* parse into the expression */
2212     if (!parser_next(parser)) {
2213         parseerror(parser, "expected 'if' condition after opening paren");
2214         return false;
2215     }
2216     /* parse the condition */
2217     cond = parse_expression_leave(parser, false, true, false);
2218     if (!cond)
2219         return false;
2220     /* closing paren */
2221     if (parser->tok != ')') {
2222         parseerror(parser, "expected closing paren after 'if' condition");
2223         ast_unref(cond);
2224         return false;
2225     }
2226     /* parse into the 'then' branch */
2227     if (!parser_next(parser)) {
2228         parseerror(parser, "expected statement for on-true branch of 'if'");
2229         ast_unref(cond);
2230         return false;
2231     }
2232     if (!parse_statement_or_block(parser, &ontrue)) {
2233         ast_unref(cond);
2234         return false;
2235     }
2236     if (!ontrue)
2237         ontrue = (ast_expression*)ast_block_new(parser_ctx(parser));
2238     /* check for an else */
2239     if (!strcmp(parser_tokval(parser), "else")) {
2240         /* parse into the 'else' branch */
2241         if (!parser_next(parser)) {
2242             parseerror(parser, "expected on-false branch after 'else'");
2243             ast_delete(ontrue);
2244             ast_unref(cond);
2245             return false;
2246         }
2247         if (!parse_statement_or_block(parser, &onfalse)) {
2248             ast_delete(ontrue);
2249             ast_unref(cond);
2250             return false;
2251         }
2252     }
2253
2254     cond = process_condition(parser, cond, &ifnot);
2255     if (!cond) {
2256         if (ontrue)  ast_delete(ontrue);
2257         if (onfalse) ast_delete(onfalse);
2258         return false;
2259     }
2260
2261     if (ifnot)
2262         ifthen = ast_ifthen_new(ctx, cond, onfalse, ontrue);
2263     else
2264         ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse);
2265     *out = (ast_expression*)ifthen;
2266     return true;
2267 }
2268
2269 static bool parse_while_go(parser_t *parser, ast_block *block, ast_expression **out);
2270 static bool parse_while(parser_t *parser, ast_block *block, ast_expression **out)
2271 {
2272     bool rv;
2273     char *label = NULL;
2274
2275     /* skip the 'while' and get the body */
2276     if (!parser_next(parser)) {
2277         if (OPTS_FLAG(LOOP_LABELS))
2278             parseerror(parser, "expected loop label or 'while' condition in parenthesis");
2279         else
2280             parseerror(parser, "expected 'while' condition in parenthesis");
2281         return false;
2282     }
2283
2284     if (parser->tok == ':') {
2285         if (!OPTS_FLAG(LOOP_LABELS))
2286             parseerror(parser, "labeled loops not activated, try using -floop-labels");
2287         if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
2288             parseerror(parser, "expected loop label");
2289             return false;
2290         }
2291         label = util_strdup(parser_tokval(parser));
2292         if (!parser_next(parser)) {
2293             mem_d(label);
2294             parseerror(parser, "expected 'while' condition in parenthesis");
2295             return false;
2296         }
2297     }
2298
2299     if (parser->tok != '(') {
2300         parseerror(parser, "expected 'while' condition in parenthesis");
2301         return false;
2302     }
2303
2304     vec_push(parser->breaks, label);
2305     vec_push(parser->continues, label);
2306
2307     rv = parse_while_go(parser, block, out);
2308     if (label)
2309         mem_d(label);
2310     if (vec_last(parser->breaks) != label || vec_last(parser->continues) != label) {
2311         parseerror(parser, "internal error: label stack corrupted");
2312         rv = false;
2313         ast_delete(*out);
2314         *out = NULL;
2315     }
2316     else {
2317         vec_pop(parser->breaks);
2318         vec_pop(parser->continues);
2319     }
2320     return rv;
2321 }
2322
2323 static bool parse_while_go(parser_t *parser, ast_block *block, ast_expression **out)
2324 {
2325     ast_loop *aloop;
2326     ast_expression *cond, *ontrue;
2327
2328     bool ifnot = false;
2329
2330     lex_ctx_t ctx = parser_ctx(parser);
2331
2332     (void)block; /* not touching */
2333
2334     /* parse into the expression */
2335     if (!parser_next(parser)) {
2336         parseerror(parser, "expected 'while' condition after opening paren");
2337         return false;
2338     }
2339     /* parse the condition */
2340     cond = parse_expression_leave(parser, false, true, false);
2341     if (!cond)
2342         return false;
2343     /* closing paren */
2344     if (parser->tok != ')') {
2345         parseerror(parser, "expected closing paren after 'while' condition");
2346         ast_unref(cond);
2347         return false;
2348     }
2349     /* parse into the 'then' branch */
2350     if (!parser_next(parser)) {
2351         parseerror(parser, "expected while-loop body");
2352         ast_unref(cond);
2353         return false;
2354     }
2355     if (!parse_statement_or_block(parser, &ontrue)) {
2356         ast_unref(cond);
2357         return false;
2358     }
2359
2360     cond = process_condition(parser, cond, &ifnot);
2361     if (!cond) {
2362         ast_unref(ontrue);
2363         return false;
2364     }
2365     aloop = ast_loop_new(ctx, NULL, cond, ifnot, NULL, false, NULL, ontrue);
2366     *out = (ast_expression*)aloop;
2367     return true;
2368 }
2369
2370 static bool parse_dowhile_go(parser_t *parser, ast_block *block, ast_expression **out);
2371 static bool parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
2372 {
2373     bool rv;
2374     char *label = NULL;
2375
2376     /* skip the 'do' and get the body */
2377     if (!parser_next(parser)) {
2378         if (OPTS_FLAG(LOOP_LABELS))
2379             parseerror(parser, "expected loop label or body");
2380         else
2381             parseerror(parser, "expected loop body");
2382         return false;
2383     }
2384
2385     if (parser->tok == ':') {
2386         if (!OPTS_FLAG(LOOP_LABELS))
2387             parseerror(parser, "labeled loops not activated, try using -floop-labels");
2388         if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
2389             parseerror(parser, "expected loop label");
2390             return false;
2391         }
2392         label = util_strdup(parser_tokval(parser));
2393         if (!parser_next(parser)) {
2394             mem_d(label);
2395             parseerror(parser, "expected loop body");
2396             return false;
2397         }
2398     }
2399
2400     vec_push(parser->breaks, label);
2401     vec_push(parser->continues, label);
2402
2403     rv = parse_dowhile_go(parser, block, out);
2404     if (label)
2405         mem_d(label);
2406     if (vec_last(parser->breaks) != label || vec_last(parser->continues) != label) {
2407         parseerror(parser, "internal error: label stack corrupted");
2408         rv = false;
2409         /*
2410          * Test for NULL otherwise ast_delete dereferences null pointer
2411          * and boom.
2412          */
2413         if (*out)
2414             ast_delete(*out);
2415         *out = NULL;
2416     }
2417     else {
2418         vec_pop(parser->breaks);
2419         vec_pop(parser->continues);
2420     }
2421     return rv;
2422 }
2423
2424 static bool parse_dowhile_go(parser_t *parser, ast_block *block, ast_expression **out)
2425 {
2426     ast_loop *aloop;
2427     ast_expression *cond, *ontrue;
2428
2429     bool ifnot = false;
2430
2431     lex_ctx_t ctx = parser_ctx(parser);
2432
2433     (void)block; /* not touching */
2434
2435     if (!parse_statement_or_block(parser, &ontrue))
2436         return false;
2437
2438     /* expect the "while" */
2439     if (parser->tok != TOKEN_KEYWORD ||
2440         strcmp(parser_tokval(parser), "while"))
2441     {
2442         parseerror(parser, "expected 'while' and condition");
2443         ast_delete(ontrue);
2444         return false;
2445     }
2446
2447     /* skip the 'while' and check for opening paren */
2448     if (!parser_next(parser) || parser->tok != '(') {
2449         parseerror(parser, "expected 'while' condition in parenthesis");
2450         ast_delete(ontrue);
2451         return false;
2452     }
2453     /* parse into the expression */
2454     if (!parser_next(parser)) {
2455         parseerror(parser, "expected 'while' condition after opening paren");
2456         ast_delete(ontrue);
2457         return false;
2458     }
2459     /* parse the condition */
2460     cond = parse_expression_leave(parser, false, true, false);
2461     if (!cond)
2462         return false;
2463     /* closing paren */
2464     if (parser->tok != ')') {
2465         parseerror(parser, "expected closing paren after 'while' condition");
2466         ast_delete(ontrue);
2467         ast_unref(cond);
2468         return false;
2469     }
2470     /* parse on */
2471     if (!parser_next(parser) || parser->tok != ';') {
2472         parseerror(parser, "expected semicolon after condition");
2473         ast_delete(ontrue);
2474         ast_unref(cond);
2475         return false;
2476     }
2477
2478     if (!parser_next(parser)) {
2479         parseerror(parser, "parse error");
2480         ast_delete(ontrue);
2481         ast_unref(cond);
2482         return false;
2483     }
2484
2485     cond = process_condition(parser, cond, &ifnot);
2486     if (!cond) {
2487         ast_delete(ontrue);
2488         return false;
2489     }
2490     aloop = ast_loop_new(ctx, NULL, NULL, false, cond, ifnot, NULL, ontrue);
2491     *out = (ast_expression*)aloop;
2492     return true;
2493 }
2494
2495 static bool parse_for_go(parser_t *parser, ast_block *block, ast_expression **out);
2496 static bool parse_for(parser_t *parser, ast_block *block, ast_expression **out)
2497 {
2498     bool rv;
2499     char *label = NULL;
2500
2501     /* skip the 'for' and check for opening paren */
2502     if (!parser_next(parser)) {
2503         if (OPTS_FLAG(LOOP_LABELS))
2504             parseerror(parser, "expected loop label or 'for' expressions in parenthesis");
2505         else
2506             parseerror(parser, "expected 'for' expressions in parenthesis");
2507         return false;
2508     }
2509
2510     if (parser->tok == ':') {
2511         if (!OPTS_FLAG(LOOP_LABELS))
2512             parseerror(parser, "labeled loops not activated, try using -floop-labels");
2513         if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
2514             parseerror(parser, "expected loop label");
2515             return false;
2516         }
2517         label = util_strdup(parser_tokval(parser));
2518         if (!parser_next(parser)) {
2519             mem_d(label);
2520             parseerror(parser, "expected 'for' expressions in parenthesis");
2521             return false;
2522         }
2523     }
2524
2525     if (parser->tok != '(') {
2526         parseerror(parser, "expected 'for' expressions in parenthesis");
2527         return false;
2528     }
2529
2530     vec_push(parser->breaks, label);
2531     vec_push(parser->continues, label);
2532
2533     rv = parse_for_go(parser, block, out);
2534     if (label)
2535         mem_d(label);
2536     if (vec_last(parser->breaks) != label || vec_last(parser->continues) != label) {
2537         parseerror(parser, "internal error: label stack corrupted");
2538         rv = false;
2539         ast_delete(*out);
2540         *out = NULL;
2541     }
2542     else {
2543         vec_pop(parser->breaks);
2544         vec_pop(parser->continues);
2545     }
2546     return rv;
2547 }
2548 static bool parse_for_go(parser_t *parser, ast_block *block, ast_expression **out)
2549 {
2550     ast_loop       *aloop;
2551     ast_expression *initexpr, *cond, *increment, *ontrue;
2552     ast_value      *typevar;
2553
2554     bool ifnot  = false;
2555
2556     lex_ctx_t ctx = parser_ctx(parser);
2557
2558     parser_enterblock(parser);
2559
2560     initexpr  = NULL;
2561     cond      = NULL;
2562     increment = NULL;
2563     ontrue    = NULL;
2564
2565     /* parse into the expression */
2566     if (!parser_next(parser)) {
2567         parseerror(parser, "expected 'for' initializer after opening paren");
2568         goto onerr;
2569     }
2570
2571     typevar = NULL;
2572     if (parser->tok == TOKEN_IDENT)
2573         typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
2574
2575     if (typevar || parser->tok == TOKEN_TYPENAME) {
2576 #if 0
2577         if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_GMQCC) {
2578             if (parsewarning(parser, WARN_EXTENSIONS,
2579                              "current standard does not allow variable declarations in for-loop initializers"))
2580                 goto onerr;
2581         }
2582 #endif
2583         if (!parse_variable(parser, block, true, CV_VAR, typevar, false, false, 0, NULL))
2584             goto onerr;
2585     }
2586     else if (parser->tok != ';')
2587     {
2588         initexpr = parse_expression_leave(parser, false, false, false);
2589         if (!initexpr)
2590             goto onerr;
2591
2592         /* move on to condition */
2593         if (parser->tok != ';') {
2594             parseerror(parser, "expected semicolon after for-loop initializer");
2595             goto onerr;
2596         }
2597
2598         if (!parser_next(parser)) {
2599             parseerror(parser, "expected for-loop condition");
2600             goto onerr;
2601         }
2602     }
2603
2604     /* parse the condition */
2605     if (parser->tok != ';') {
2606         cond = parse_expression_leave(parser, false, true, false);
2607         if (!cond)
2608             goto onerr;
2609     }
2610
2611     /* move on to incrementor */
2612     if (parser->tok != ';') {
2613         parseerror(parser, "expected semicolon after for-loop initializer");
2614         goto onerr;
2615     }
2616     if (!parser_next(parser)) {
2617         parseerror(parser, "expected for-loop condition");
2618         goto onerr;
2619     }
2620
2621     /* parse the incrementor */
2622     if (parser->tok != ')') {
2623         lex_ctx_t condctx = parser_ctx(parser);
2624         increment = parse_expression_leave(parser, false, false, false);
2625         if (!increment)
2626             goto onerr;
2627         if (!ast_side_effects(increment)) {
2628             if (compile_warning(condctx, WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
2629                 goto onerr;
2630         }
2631     }
2632
2633     /* closing paren */
2634     if (parser->tok != ')') {
2635         parseerror(parser, "expected closing paren after 'for-loop' incrementor");
2636         goto onerr;
2637     }
2638     /* parse into the 'then' branch */
2639     if (!parser_next(parser)) {
2640         parseerror(parser, "expected for-loop body");
2641         goto onerr;
2642     }
2643     if (!parse_statement_or_block(parser, &ontrue))
2644         goto onerr;
2645
2646     if (cond) {
2647         cond = process_condition(parser, cond, &ifnot);
2648         if (!cond)
2649             goto onerr;
2650     }
2651     aloop = ast_loop_new(ctx, initexpr, cond, ifnot, NULL, false, increment, ontrue);
2652     *out = (ast_expression*)aloop;
2653
2654     if (!parser_leaveblock(parser)) {
2655         ast_delete(aloop);
2656         return false;
2657     }
2658     return true;
2659 onerr:
2660     if (initexpr)  ast_unref(initexpr);
2661     if (cond)      ast_unref(cond);
2662     if (increment) ast_unref(increment);
2663     (void)!parser_leaveblock(parser);
2664     return false;
2665 }
2666
2667 static bool parse_return(parser_t *parser, ast_block *block, ast_expression **out)
2668 {
2669     ast_expression *exp      = NULL;
2670     ast_expression *var      = NULL;
2671     ast_return     *ret      = NULL;
2672     ast_value      *retval   = parser->function->return_value;
2673     ast_value      *expected = parser->function->vtype;
2674
2675     lex_ctx_t ctx = parser_ctx(parser);
2676
2677     (void)block; /* not touching */
2678
2679     if (!parser_next(parser)) {
2680         parseerror(parser, "expected return expression");
2681         return false;
2682     }
2683
2684     /* return assignments */
2685     if (parser->tok == '=') {
2686         if (!OPTS_FLAG(RETURN_ASSIGNMENTS)) {
2687             parseerror(parser, "return assignments not activated, try using -freturn-assigments");
2688             return false;
2689         }
2690
2691         if (type_store_instr[expected->expression.next->vtype] == VINSTR_END) {
2692             char ty1[1024];
2693             ast_type_to_string(expected->expression.next, ty1, sizeof(ty1));
2694             parseerror(parser, "invalid return type: `%s'", ty1);
2695             return false;
2696         }
2697
2698         if (!parser_next(parser)) {
2699             parseerror(parser, "expected return assignment expression");
2700             return false;
2701         }
2702
2703         if (!(exp = parse_expression_leave(parser, false, false, false)))
2704             return false;
2705
2706         /* prepare the return value */
2707         if (!retval) {
2708             retval = ast_value_new(ctx, "#LOCAL_RETURN", TYPE_VOID);
2709             ast_type_adopt(retval, expected->expression.next);
2710             parser->function->return_value = retval;
2711         }
2712
2713         if (!ast_compare_type(exp, (ast_expression*)retval)) {
2714             char ty1[1024], ty2[1024];
2715             ast_type_to_string(exp, ty1, sizeof(ty1));
2716             ast_type_to_string(&retval->expression, ty2, sizeof(ty2));
2717             parseerror(parser, "invalid type for return value: `%s', expected `%s'", ty1, ty2);
2718         }
2719
2720         /* store to 'return' local variable */
2721         var = (ast_expression*)ast_store_new(
2722             ctx,
2723             type_store_instr[expected->expression.next->vtype],
2724             (ast_expression*)retval, exp);
2725
2726         if (!var) {
2727             ast_unref(exp);
2728             return false;
2729         }
2730
2731         if (parser->tok != ';')
2732             parseerror(parser, "missing semicolon after return assignment");
2733         else if (!parser_next(parser))
2734             parseerror(parser, "parse error after return assignment");
2735
2736         *out = var;
2737         return true;
2738     }
2739
2740     if (parser->tok != ';') {
2741         exp = parse_expression(parser, false, false);
2742         if (!exp)
2743             return false;
2744
2745         if (exp->vtype != TYPE_NIL &&
2746             exp->vtype != ((ast_expression*)expected)->next->vtype)
2747         {
2748             parseerror(parser, "return with invalid expression");
2749         }
2750
2751         ret = ast_return_new(ctx, exp);
2752         if (!ret) {
2753             ast_unref(exp);
2754             return false;
2755         }
2756     } else {
2757         if (!parser_next(parser))
2758             parseerror(parser, "parse error");
2759
2760         if (!retval && expected->expression.next->vtype != TYPE_VOID)
2761         {
2762             (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
2763         }
2764         ret = ast_return_new(ctx, (ast_expression*)retval);
2765     }
2766     *out = (ast_expression*)ret;
2767     return true;
2768 }
2769
2770 static bool parse_break_continue(parser_t *parser, ast_block *block, ast_expression **out, bool is_continue)
2771 {
2772     size_t       i;
2773     unsigned int levels = 0;
2774     lex_ctx_t      ctx = parser_ctx(parser);
2775     const char **loops = (is_continue ? parser->continues : parser->breaks);
2776
2777     (void)block; /* not touching */
2778     if (!parser_next(parser)) {
2779         parseerror(parser, "expected semicolon or loop label");
2780         return false;
2781     }
2782
2783     if (!vec_size(loops)) {
2784         if (is_continue)
2785             parseerror(parser, "`continue` can only be used inside loops");
2786         else
2787             parseerror(parser, "`break` can only be used inside loops or switches");
2788     }
2789
2790     if (parser->tok == TOKEN_IDENT) {
2791         if (!OPTS_FLAG(LOOP_LABELS))
2792             parseerror(parser, "labeled loops not activated, try using -floop-labels");
2793         i = vec_size(loops);
2794         while (i--) {
2795             if (loops[i] && !strcmp(loops[i], parser_tokval(parser)))
2796                 break;
2797             if (!i) {
2798                 parseerror(parser, "no such loop to %s: `%s`",
2799                            (is_continue ? "continue" : "break out of"),
2800                            parser_tokval(parser));
2801                 return false;
2802             }
2803             ++levels;
2804         }
2805         if (!parser_next(parser)) {
2806             parseerror(parser, "expected semicolon");
2807             return false;
2808         }
2809     }
2810
2811     if (parser->tok != ';') {
2812         parseerror(parser, "expected semicolon");
2813         return false;
2814     }
2815
2816     if (!parser_next(parser))
2817         parseerror(parser, "parse error");
2818
2819     *out = (ast_expression*)ast_breakcont_new(ctx, is_continue, levels);
2820     return true;
2821 }
2822
2823 /* returns true when it was a variable qualifier, false otherwise!
2824  * on error, cvq is set to CV_WRONG
2825  */
2826 typedef struct {
2827     const char *name;
2828     size_t      flag;
2829 } attribute_t;
2830
2831 static bool parse_qualifiers(parser_t *parser, bool with_local, int *cvq, bool *noref, bool *is_static, uint32_t *_flags, char **message)
2832 {
2833     bool had_const    = false;
2834     bool had_var      = false;
2835     bool had_noref    = false;
2836     bool had_attrib   = false;
2837     bool had_static   = false;
2838     uint32_t flags    = 0;
2839
2840     static attribute_t attributes[] = {
2841         { "noreturn",   AST_FLAG_NORETURN   },
2842         { "inline",     AST_FLAG_INLINE     },
2843         { "eraseable",  AST_FLAG_ERASEABLE  },
2844         { "accumulate", AST_FLAG_ACCUMULATE },
2845         { "last",       AST_FLAG_FINAL_DECL }
2846     };
2847
2848    *cvq = CV_NONE;
2849
2850     for (;;) {
2851         size_t i;
2852         if (parser->tok == TOKEN_ATTRIBUTE_OPEN) {
2853             had_attrib = true;
2854             /* parse an attribute */
2855             if (!parser_next(parser)) {
2856                 parseerror(parser, "expected attribute after `[[`");
2857                 *cvq = CV_WRONG;
2858                 return false;
2859             }
2860
2861             for (i = 0; i < GMQCC_ARRAY_COUNT(attributes); i++) {
2862                 if (!strcmp(parser_tokval(parser), attributes[i].name)) {
2863                     flags |= attributes[i].flag;
2864                     if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2865                         parseerror(parser, "`%s` attribute has no parameters, expected `]]`",
2866                             attributes[i].name);
2867                         *cvq = CV_WRONG;
2868                         return false;
2869                     }
2870                     break;
2871                 }
2872             }
2873
2874             if (i != GMQCC_ARRAY_COUNT(attributes))
2875                 goto leave;
2876
2877
2878             if (!strcmp(parser_tokval(parser), "noref")) {
2879                 had_noref = true;
2880                 if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2881                     parseerror(parser, "`noref` attribute has no parameters, expected `]]`");
2882                     *cvq = CV_WRONG;
2883                     return false;
2884                 }
2885             }
2886             else if (!strcmp(parser_tokval(parser), "alias") && !(flags & AST_FLAG_ALIAS)) {
2887                 flags   |= AST_FLAG_ALIAS;
2888                 *message = NULL;
2889
2890                 if (!parser_next(parser)) {
2891                     parseerror(parser, "parse error in attribute");
2892                     goto argerr;
2893                 }
2894
2895                 if (parser->tok == '(') {
2896                     if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
2897                         parseerror(parser, "`alias` attribute missing parameter");
2898                         goto argerr;
2899                     }
2900
2901                     *message = util_strdup(parser_tokval(parser));
2902
2903                     if (!parser_next(parser)) {
2904                         parseerror(parser, "parse error in attribute");
2905                         goto argerr;
2906                     }
2907
2908                     if (parser->tok != ')') {
2909                         parseerror(parser, "`alias` attribute expected `)` after parameter");
2910                         goto argerr;
2911                     }
2912
2913                     if (!parser_next(parser)) {
2914                         parseerror(parser, "parse error in attribute");
2915                         goto argerr;
2916                     }
2917                 }
2918
2919                 if (parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2920                     parseerror(parser, "`alias` attribute expected `]]`");
2921                     goto argerr;
2922                 }
2923             }
2924             else if (!strcmp(parser_tokval(parser), "deprecated") && !(flags & AST_FLAG_DEPRECATED)) {
2925                 flags   |= AST_FLAG_DEPRECATED;
2926                 *message = NULL;
2927
2928                 if (!parser_next(parser)) {
2929                     parseerror(parser, "parse error in attribute");
2930                     goto argerr;
2931                 }
2932
2933                 if (parser->tok == '(') {
2934                     if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
2935                         parseerror(parser, "`deprecated` attribute missing parameter");
2936                         goto argerr;
2937                     }
2938
2939                     *message = util_strdup(parser_tokval(parser));
2940
2941                     if (!parser_next(parser)) {
2942                         parseerror(parser, "parse error in attribute");
2943                         goto argerr;
2944                     }
2945
2946                     if(parser->tok != ')') {
2947                         parseerror(parser, "`deprecated` attribute expected `)` after parameter");
2948                         goto argerr;
2949                     }
2950
2951                     if (!parser_next(parser)) {
2952                         parseerror(parser, "parse error in attribute");
2953                         goto argerr;
2954                     }
2955                 }
2956                 /* no message */
2957                 if (parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2958                     parseerror(parser, "`deprecated` attribute expected `]]`");
2959
2960                     argerr: /* ugly */
2961                     if (*message) mem_d(*message);
2962                     *message = NULL;
2963                     *cvq     = CV_WRONG;
2964                     return false;
2965                 }
2966             }
2967             else if (!strcmp(parser_tokval(parser), "coverage") && !(flags & AST_FLAG_COVERAGE)) {
2968                 flags |= AST_FLAG_COVERAGE;
2969                 if (!parser_next(parser)) {
2970                     error_in_coverage:
2971                     parseerror(parser, "parse error in coverage attribute");
2972                     *cvq = CV_WRONG;
2973                     return false;
2974                 }
2975                 if (parser->tok == '(') {
2976                     if (!parser_next(parser)) {
2977                         bad_coverage_arg:
2978                         parseerror(parser, "invalid parameter for coverage() attribute\n"
2979                                            "valid are: block");
2980                         *cvq = CV_WRONG;
2981                         return false;
2982                     }
2983                     if (parser->tok != ')') {
2984                         do {
2985                             if (parser->tok != TOKEN_IDENT)
2986                                 goto bad_coverage_arg;
2987                             if (!strcmp(parser_tokval(parser), "block"))
2988                                 flags |= AST_FLAG_BLOCK_COVERAGE;
2989                             else if (!strcmp(parser_tokval(parser), "none"))
2990                                 flags &= ~(AST_FLAG_COVERAGE_MASK);
2991                             else
2992                                 goto bad_coverage_arg;
2993                             if (!parser_next(parser))
2994                                 goto error_in_coverage;
2995                             if (parser->tok == ',') {
2996                                 if (!parser_next(parser))
2997                                     goto error_in_coverage;
2998                             }
2999                         } while (parser->tok != ')');
3000                     }
3001                     if (parser->tok != ')' || !parser_next(parser))
3002                         goto error_in_coverage;
3003                 } else {
3004                     /* without parameter [[coverage]] equals [[coverage(block)]] */
3005                     flags |= AST_FLAG_BLOCK_COVERAGE;
3006                 }
3007             }
3008             else
3009             {
3010                 /* Skip tokens until we hit a ]] */
3011                 (void)!parsewarning(parser, WARN_UNKNOWN_ATTRIBUTE, "unknown attribute starting with `%s`", parser_tokval(parser));
3012                 while (parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
3013                     if (!parser_next(parser)) {
3014                         parseerror(parser, "error inside attribute");
3015                         *cvq = CV_WRONG;
3016                         return false;
3017                     }
3018                 }
3019             }
3020         }
3021         else if (with_local && !strcmp(parser_tokval(parser), "static"))
3022             had_static = true;
3023         else if (!strcmp(parser_tokval(parser), "const"))
3024             had_const = true;
3025         else if (!strcmp(parser_tokval(parser), "var"))
3026             had_var = true;
3027         else if (with_local && !strcmp(parser_tokval(parser), "local"))
3028             had_var = true;
3029         else if (!strcmp(parser_tokval(parser), "noref"))
3030             had_noref = true;
3031         else if (!had_const && !had_var && !had_noref && !had_attrib && !had_static && !flags) {
3032             return false;
3033         }
3034         else
3035             break;
3036
3037         leave:
3038         if (!parser_next(parser))
3039             goto onerr;
3040     }
3041     if (had_const)
3042         *cvq = CV_CONST;
3043     else if (had_var)
3044         *cvq = CV_VAR;
3045     else
3046         *cvq = CV_NONE;
3047     *noref     = had_noref;
3048     *is_static = had_static;
3049     *_flags    = flags;
3050     return true;
3051 onerr:
3052     parseerror(parser, "parse error after variable qualifier");
3053     *cvq = CV_WRONG;
3054     return true;
3055 }
3056
3057 static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression **out);
3058 static bool parse_switch(parser_t *parser, ast_block *block, ast_expression **out)
3059 {
3060     bool rv;
3061     char *label = NULL;
3062
3063     /* skip the 'while' and get the body */
3064     if (!parser_next(parser)) {
3065         if (OPTS_FLAG(LOOP_LABELS))
3066             parseerror(parser, "expected loop label or 'switch' operand in parenthesis");
3067         else
3068             parseerror(parser, "expected 'switch' operand in parenthesis");
3069         return false;
3070     }
3071
3072     if (parser->tok == ':') {
3073         if (!OPTS_FLAG(LOOP_LABELS))
3074             parseerror(parser, "labeled loops not activated, try using -floop-labels");
3075         if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
3076             parseerror(parser, "expected loop label");
3077             return false;
3078         }
3079         label = util_strdup(parser_tokval(parser));
3080         if (!parser_next(parser)) {
3081             mem_d(label);
3082             parseerror(parser, "expected 'switch' operand in parenthesis");
3083             return false;
3084         }
3085     }
3086
3087     if (parser->tok != '(') {
3088         parseerror(parser, "expected 'switch' operand in parenthesis");
3089         return false;
3090     }
3091
3092     vec_push(parser->breaks, label);
3093
3094     rv = parse_switch_go(parser, block, out);
3095     if (label)
3096         mem_d(label);
3097     if (vec_last(parser->breaks) != label) {
3098         parseerror(parser, "internal error: label stack corrupted");
3099         rv = false;
3100         ast_delete(*out);
3101         *out = NULL;
3102     }
3103     else {
3104         vec_pop(parser->breaks);
3105     }
3106     return rv;
3107 }
3108
3109 static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression **out)
3110 {
3111     ast_expression *operand;
3112     ast_value      *opval;
3113     ast_value      *typevar;
3114     ast_switch     *switchnode;
3115     ast_switch_case swcase;
3116
3117     int  cvq;
3118     bool noref, is_static;
3119     uint32_t qflags = 0;
3120
3121     lex_ctx_t ctx = parser_ctx(parser);
3122
3123     (void)block; /* not touching */
3124     (void)opval;
3125
3126     /* parse into the expression */
3127     if (!parser_next(parser)) {
3128         parseerror(parser, "expected switch operand");
3129         return false;
3130     }
3131     /* parse the operand */
3132     operand = parse_expression_leave(parser, false, false, false);
3133     if (!operand)
3134         return false;
3135
3136     switchnode = ast_switch_new(ctx, operand);
3137
3138     /* closing paren */
3139     if (parser->tok != ')') {
3140         ast_delete(switchnode);
3141         parseerror(parser, "expected closing paren after 'switch' operand");
3142         return false;
3143     }
3144
3145     /* parse over the opening paren */
3146     if (!parser_next(parser) || parser->tok != '{') {
3147         ast_delete(switchnode);
3148         parseerror(parser, "expected list of cases");
3149         return false;
3150     }
3151
3152     if (!parser_next(parser)) {
3153         ast_delete(switchnode);
3154         parseerror(parser, "expected 'case' or 'default'");
3155         return false;
3156     }
3157
3158     /* new block; allow some variables to be declared here */
3159     parser_enterblock(parser);
3160     while (true) {
3161         typevar = NULL;
3162         if (parser->tok == TOKEN_IDENT)
3163             typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
3164         if (typevar || parser->tok == TOKEN_TYPENAME) {
3165             if (!parse_variable(parser, block, true, CV_NONE, typevar, false, false, 0, NULL)) {
3166                 ast_delete(switchnode);
3167                 return false;
3168             }
3169             continue;
3170         }
3171         if (parse_qualifiers(parser, true, &cvq, &noref, &is_static, &qflags, NULL))
3172         {
3173             if (cvq == CV_WRONG) {
3174                 ast_delete(switchnode);
3175                 return false;
3176             }
3177             if (!parse_variable(parser, block, true, cvq, NULL, noref, is_static, qflags, NULL)) {
3178                 ast_delete(switchnode);
3179                 return false;
3180             }
3181             continue;
3182         }
3183         break;
3184     }
3185
3186     /* case list! */
3187     while (parser->tok != '}') {
3188         ast_block *caseblock;
3189
3190         if (!strcmp(parser_tokval(parser), "case")) {
3191             if (!parser_next(parser)) {
3192                 ast_delete(switchnode);
3193                 parseerror(parser, "expected expression for case");
3194                 return false;
3195             }
3196             swcase.value = parse_expression_leave(parser, false, false, false);
3197             if (!swcase.value) {
3198                 ast_delete(switchnode);
3199                 parseerror(parser, "expected expression for case");
3200                 return false;
3201             }
3202             if (!OPTS_FLAG(RELAXED_SWITCH)) {
3203                 if (!ast_istype(swcase.value, ast_value)) { /* || ((ast_value*)swcase.value)->cvq != CV_CONST) { */
3204                     parseerror(parser, "case on non-constant values need to be explicitly enabled via -frelaxed-switch");
3205                     ast_unref(operand);
3206                     return false;
3207                 }
3208             }
3209         }
3210         else if (!strcmp(parser_tokval(parser), "default")) {
3211             swcase.value = NULL;
3212             if (!parser_next(parser)) {
3213                 ast_delete(switchnode);
3214                 parseerror(parser, "expected colon");
3215                 return false;
3216             }
3217         }
3218         else {
3219             ast_delete(switchnode);
3220             parseerror(parser, "expected 'case' or 'default'");
3221             return false;
3222         }
3223
3224         /* Now the colon and body */
3225         if (parser->tok != ':') {
3226             if (swcase.value) ast_unref(swcase.value);
3227             ast_delete(switchnode);
3228             parseerror(parser, "expected colon");
3229             return false;
3230         }
3231
3232         if (!parser_next(parser)) {
3233             if (swcase.value) ast_unref(swcase.value);
3234             ast_delete(switchnode);
3235             parseerror(parser, "expected statements or case");
3236             return false;
3237         }
3238         caseblock = ast_block_new(parser_ctx(parser));
3239         if (!caseblock) {
3240             if (swcase.value) ast_unref(swcase.value);
3241             ast_delete(switchnode);
3242             return false;
3243         }
3244         swcase.code = (ast_expression*)caseblock;
3245         vec_push(switchnode->cases, swcase);
3246         while (true) {
3247             ast_expression *expr;
3248             if (parser->tok == '}')
3249                 break;
3250             if (parser->tok == TOKEN_KEYWORD) {
3251                 if (!strcmp(parser_tokval(parser), "case") ||
3252                     !strcmp(parser_tokval(parser), "default"))
3253                 {
3254                     break;
3255                 }
3256             }
3257             if (!parse_statement(parser, caseblock, &expr, true)) {
3258                 ast_delete(switchnode);
3259                 return false;
3260             }
3261             if (!expr)
3262                 continue;
3263             if (!ast_block_add_expr(caseblock, expr)) {
3264                 ast_delete(switchnode);
3265                 return false;
3266             }
3267         }
3268     }
3269
3270     parser_leaveblock(parser);
3271
3272     /* closing paren */
3273     if (parser->tok != '}') {
3274         ast_delete(switchnode);
3275         parseerror(parser, "expected closing paren of case list");
3276         return false;
3277     }
3278     if (!parser_next(parser)) {
3279         ast_delete(switchnode);
3280         parseerror(parser, "parse error after switch");
3281         return false;
3282     }
3283     *out = (ast_expression*)switchnode;
3284     return true;
3285 }
3286
3287 /* parse computed goto sides */
3288 static ast_expression *parse_goto_computed(parser_t *parser, ast_expression **side) {
3289     ast_expression *on_true;
3290     ast_expression *on_false;
3291     ast_expression *cond;
3292
3293     if (!*side)
3294         return NULL;
3295
3296     if (ast_istype(*side, ast_ternary)) {
3297         ast_ternary *tern = (ast_ternary*)*side;
3298         on_true  = parse_goto_computed(parser, &tern->on_true);
3299         on_false = parse_goto_computed(parser, &tern->on_false);
3300
3301         if (!on_true || !on_false) {
3302             parseerror(parser, "expected label or expression in ternary");
3303             if (on_true) ast_unref(on_true);
3304             if (on_false) ast_unref(on_false);
3305             return NULL;
3306         }
3307
3308         cond = tern->cond;
3309         tern->cond = NULL;
3310         ast_delete(tern);
3311         *side = NULL;
3312         return (ast_expression*)ast_ifthen_new(parser_ctx(parser), cond, on_true, on_false);
3313     } else if (ast_istype(*side, ast_label)) {
3314         ast_goto *gt = ast_goto_new(parser_ctx(parser), ((ast_label*)*side)->name);
3315         ast_goto_set_label(gt, ((ast_label*)*side));
3316         *side = NULL;
3317         return (ast_expression*)gt;
3318     }
3319     return NULL;
3320 }
3321
3322 static bool parse_goto(parser_t *parser, ast_expression **out)
3323 {
3324     ast_goto       *gt = NULL;
3325     ast_expression *lbl;
3326
3327     if (!parser_next(parser))
3328         return false;
3329
3330     if (parser->tok != TOKEN_IDENT) {
3331         ast_expression *expression;
3332
3333         /* could be an expression i.e computed goto :-) */
3334         if (parser->tok != '(') {
3335             parseerror(parser, "expected label name after `goto`");
3336             return false;
3337         }
3338
3339         /* failed to parse expression for goto */
3340         if (!(expression = parse_expression(parser, false, true)) ||
3341             !(*out = parse_goto_computed(parser, &expression))) {
3342             parseerror(parser, "invalid goto expression");
3343             if(expression)
3344                 ast_unref(expression);
3345             return false;
3346         }
3347
3348         return true;
3349     }
3350
3351     /* not computed goto */
3352     gt = ast_goto_new(parser_ctx(parser), parser_tokval(parser));
3353     lbl = parser_find_label(parser, gt->name);
3354     if (lbl) {
3355         if (!ast_istype(lbl, ast_label)) {
3356             parseerror(parser, "internal error: label is not an ast_label");
3357             ast_delete(gt);
3358             return false;
3359         }
3360         ast_goto_set_label(gt, (ast_label*)lbl);
3361     }
3362     else
3363         vec_push(parser->gotos, gt);
3364
3365     if (!parser_next(parser) || parser->tok != ';') {
3366         parseerror(parser, "semicolon expected after goto label");
3367         return false;
3368     }
3369     if (!parser_next(parser)) {
3370         parseerror(parser, "parse error after goto");
3371         return false;
3372     }
3373
3374     *out = (ast_expression*)gt;
3375     return true;
3376 }
3377
3378 static bool parse_skipwhite(parser_t *parser)
3379 {
3380     do {
3381         if (!parser_next(parser))
3382             return false;
3383     } while (parser->tok == TOKEN_WHITE && parser->tok < TOKEN_ERROR);
3384     return parser->tok < TOKEN_ERROR;
3385 }
3386
3387 static bool parse_eol(parser_t *parser)
3388 {
3389     if (!parse_skipwhite(parser))
3390         return false;
3391     return parser->tok == TOKEN_EOL;
3392 }
3393
3394 static bool parse_pragma_do(parser_t *parser)
3395 {
3396     if (!parser_next(parser) ||
3397         parser->tok != TOKEN_IDENT ||
3398         strcmp(parser_tokval(parser), "pragma"))
3399     {
3400         parseerror(parser, "expected `pragma` keyword after `#`, got `%s`", parser_tokval(parser));
3401         return false;
3402     }
3403     if (!parse_skipwhite(parser) || parser->tok != TOKEN_IDENT) {
3404         parseerror(parser, "expected pragma, got `%s`", parser_tokval(parser));
3405         return false;
3406     }
3407
3408     if (!strcmp(parser_tokval(parser), "noref")) {
3409         if (!parse_skipwhite(parser) || parser->tok != TOKEN_INTCONST) {
3410             parseerror(parser, "`noref` pragma requires an argument: 0 or 1");
3411             return false;
3412         }
3413         parser->noref = !!parser_token(parser)->constval.i;
3414         if (!parse_eol(parser)) {
3415             parseerror(parser, "parse error after `noref` pragma");
3416             return false;
3417         }
3418     }
3419     else
3420     {
3421         (void)!parsewarning(parser, WARN_UNKNOWN_PRAGMAS, "ignoring #pragma %s", parser_tokval(parser));
3422
3423         /* skip to eol */
3424         while (!parse_eol(parser)) {
3425             parser_next(parser);
3426         }
3427
3428         return true;
3429     }
3430
3431     return true;
3432 }
3433
3434 static bool parse_pragma(parser_t *parser)
3435 {
3436     bool rv;
3437     parser->lex->flags.preprocessing = true;
3438     parser->lex->flags.mergelines = true;
3439     rv = parse_pragma_do(parser);
3440     if (parser->tok != TOKEN_EOL) {
3441         parseerror(parser, "junk after pragma");
3442         rv = false;
3443     }
3444     parser->lex->flags.preprocessing = false;
3445     parser->lex->flags.mergelines = false;
3446     if (!parser_next(parser)) {
3447         parseerror(parser, "parse error after pragma");
3448         rv = false;
3449     }
3450     return rv;
3451 }
3452
3453 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases)
3454 {
3455     bool       noref, is_static;
3456     int        cvq     = CV_NONE;
3457     uint32_t   qflags  = 0;
3458     ast_value *typevar = NULL;
3459     char      *vstring = NULL;
3460
3461     *out = NULL;
3462
3463     if (parser->tok == TOKEN_IDENT)
3464         typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
3465
3466     if (typevar || parser->tok == TOKEN_TYPENAME || parser->tok == '.' || parser->tok == TOKEN_DOTS)
3467     {
3468         /* local variable */
3469         if (!block) {
3470             parseerror(parser, "cannot declare a variable from here");
3471             return false;
3472         }
3473         if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) {
3474             if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
3475                 return false;
3476         }
3477         if (!parse_variable(parser, block, false, CV_NONE, typevar, false, false, 0, NULL))
3478