1 #include <string.h>
2 #include <math.h>
3
4 #include "intrin.h"
5 #include "fold.h"
6 #include "ast.h"
7 #include "parser.h"
8
9 #define PARSER_HT_LOCALS 2
10 #define PARSER_HT_SIZE 512
11 #define TYPEDEF_HT_SIZE 512
12
13 static void parser_enterblock(parser_t *parser);
14 static bool parser_leaveblock(parser_t *parser);
15 static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e);
16 static void parser_addlocal(parser_t *parser, const std::string &name, ast_expression *e);
17 static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e);
18 static void parser_addglobal(parser_t *parser, const std::string &name, ast_expression *e);
19 static bool parse_typedef(parser_t *parser);
20 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);
21 static ast_block* parse_block(parser_t *parser);
22 static bool parse_block_into(parser_t *parser, ast_block *block);
23 static bool parse_statement_or_block(parser_t *parser, ast_expression **out);
24 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases);
25 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma, bool truthvalue, bool with_labels);
26 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma, bool with_labels);
27 static ast_value* parser_create_array_setter_proto(parser_t *parser, ast_value *array, const char *funcname);
28 static ast_value* parser_create_array_getter_proto(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname);
29 static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef, bool *is_vararg);
30
parseerror_(parser_t * parser,const char * fmt,...)31 static void parseerror_(parser_t *parser, const char *fmt, ...)
32 {
33 va_list ap;
34 va_start(ap, fmt);
35 vcompile_error(parser->lex->tok.ctx, fmt, ap);
36 va_end(ap);
37 }
38
39 template<typename... Ts>
parseerror(parser_t * parser,const char * fmt,const Ts &...ts)40 static inline void parseerror(parser_t *parser, const char *fmt, const Ts&... ts) {
41 return parseerror_(parser, fmt, formatNormalize(ts)...);
42 }
43
44 // returns true if it counts as an error
parsewarning_(parser_t * parser,int warntype,const char * fmt,...)45 static bool GMQCC_WARN parsewarning_(parser_t *parser, int warntype, const char *fmt, ...)
46 {
47 bool r;
48 va_list ap;
49 va_start(ap, fmt);
50 r = vcompile_warning(parser->lex->tok.ctx, warntype, fmt, ap);
51 va_end(ap);
52 return r;
53 }
54
55 template<typename... Ts>
parsewarning(parser_t * parser,int warntype,const char * fmt,const Ts &...ts)56 static inline bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, const Ts&... ts) {
57 return parsewarning_(parser, warntype, fmt, formatNormalize(ts)...);
58 }
59
60 /**********************************************************************
61 * parsing
62 */
63
parser_next(parser_t * parser)64 static bool parser_next(parser_t *parser)
65 {
66 /* lex_do kills the previous token */
67 parser->tok = lex_do(parser->lex);
68 if (parser->tok == TOKEN_EOF)
69 return true;
70 if (parser->tok >= TOKEN_ERROR) {
71 parseerror(parser, "lex error");
72 return false;
73 }
74 return true;
75 }
76
77 #define parser_tokval(p) ((p)->lex->tok.value)
78 #define parser_token(p) (&((p)->lex->tok))
79
parser_strdup(const char * str)80 char *parser_strdup(const char *str)
81 {
82 if (str && !*str) {
83 /* actually dup empty strings */
84 char *out = (char*)mem_a(1);
85 *out = 0;
86 return out;
87 }
88 return util_strdup(str);
89 }
90
parser_find_field(parser_t * parser,const char * name)91 static ast_expression* parser_find_field(parser_t *parser, const char *name) {
92 return (ast_expression*)util_htget(parser->htfields, name);
93 }
parser_find_field(parser_t * parser,const std::string & name)94 static ast_expression* parser_find_field(parser_t *parser, const std::string &name) {
95 return parser_find_field(parser, name.c_str());
96 }
97
parser_find_label(parser_t * parser,const char * name)98 static ast_expression* parser_find_label(parser_t *parser, const char *name)
99 {
100 for (auto &it : parser->labels)
101 if (it->m_name == name)
102 return it;
103 return nullptr;
104 }
parser_find_label(parser_t * parser,const std::string & name)105 static inline ast_expression* parser_find_label(parser_t *parser, const std::string &name) {
106 return parser_find_label(parser, name.c_str());
107 }
108
parser_find_global(parser_t * parser,const char * name)109 ast_expression* parser_find_global(parser_t *parser, const char *name)
110 {
111 ast_expression *var = (ast_expression*)util_htget(parser->aliases, parser_tokval(parser));
112 if (var)
113 return var;
114 return (ast_expression*)util_htget(parser->htglobals, name);
115 }
116
parser_find_global(parser_t * parser,const std::string & name)117 ast_expression* parser_find_global(parser_t *parser, const std::string &name) {
118 return parser_find_global(parser, name.c_str());
119 }
120
parser_find_param(parser_t * parser,const char * name)121 static ast_expression* parser_find_param(parser_t *parser, const char *name)
122 {
123 ast_value *fun;
124 if (!parser->function)
125 return nullptr;
126 fun = parser->function->m_function_type;
127 for (auto &it : fun->m_type_params) {
128 if (it->m_name == name)
129 return it.get();
130 }
131 return nullptr;
132 }
133
parser_find_local(parser_t * parser,const char * name,size_t upto,bool * isparam)134 static ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto, bool *isparam)
135 {
136 size_t i, hash;
137 ast_expression *e;
138
139 hash = util_hthash(parser->htglobals, name);
140
141 *isparam = false;
142 for (i = vec_size(parser->variables); i > upto;) {
143 --i;
144 if ( (e = (ast_expression*)util_htgeth(parser->variables[i], name, hash)) )
145 return e;
146 }
147 *isparam = true;
148 return parser_find_param(parser, name);
149 }
150
parser_find_local(parser_t * parser,const std::string & name,size_t upto,bool * isparam)151 static ast_expression* parser_find_local(parser_t *parser, const std::string &name, size_t upto, bool *isparam) {
152 return parser_find_local(parser, name.c_str(), upto, isparam);
153 }
154
parser_find_var(parser_t * parser,const char * name)155 static ast_expression* parser_find_var(parser_t *parser, const char *name)
156 {
157 bool dummy;
158 ast_expression *v;
159 v = parser_find_local(parser, name, 0, &dummy);
160 if (!v) v = parser_find_global(parser, name);
161 return v;
162 }
163
parser_find_var(parser_t * parser,const std::string & name)164 static inline ast_expression* parser_find_var(parser_t *parser, const std::string &name) {
165 return parser_find_var(parser, name.c_str());
166 }
167
parser_find_typedef(parser_t * parser,const char * name,size_t upto)168 static ast_value* parser_find_typedef(parser_t *parser, const char *name, size_t upto)
169 {
170 size_t i, hash;
171 ast_value *e;
172 hash = util_hthash(parser->typedefs[0], name);
173
174 for (i = vec_size(parser->typedefs); i > upto;) {
175 --i;
176 if ( (e = (ast_value*)util_htgeth(parser->typedefs[i], name, hash)) )
177 return e;
178 }
179 return nullptr;
180 }
181
parser_find_typedef(parser_t * parser,const std::string & name,size_t upto)182 static ast_value* parser_find_typedef(parser_t *parser, const std::string &name, size_t upto) {
183 return parser_find_typedef(parser, name.c_str(), upto);
184 }
185
186 struct sy_elem {
187 size_t etype; /* 0 = expression, others are operators */
188 bool isparen;
189 size_t off;
190 ast_expression *out;
191 ast_block *block; /* for commas and function calls */
192 lex_ctx_t ctx;
193 };
194
195 enum {
196 PAREN_EXPR,
197 PAREN_FUNC,
198 PAREN_INDEX,
199 PAREN_TERNARY1,
200 PAREN_TERNARY2
201 };
202
203 struct shunt {
204 std::vector<sy_elem> out;
205 std::vector<sy_elem> ops;
206 std::vector<size_t> argc;
207 std::vector<unsigned int> paren;
208 };
209
syexp(lex_ctx_t ctx,ast_expression * v)210 static sy_elem syexp(lex_ctx_t ctx, ast_expression *v) {
211 sy_elem e;
212 e.etype = 0;
213 e.off = 0;
214 e.out = v;
215 e.block = nullptr;
216 e.ctx = ctx;
217 e.isparen = false;
218 return e;
219 }
220
syblock(lex_ctx_t ctx,ast_block * v)221 static sy_elem syblock(lex_ctx_t ctx, ast_block *v) {
222 sy_elem e;
223 e.etype = 0;
224 e.off = 0;
225 e.out = v;
226 e.block = v;
227 e.ctx = ctx;
228 e.isparen = false;
229 return e;
230 }
231
syop(lex_ctx_t ctx,const oper_info * op)232 static sy_elem syop(lex_ctx_t ctx, const oper_info *op) {
233 sy_elem e;
234 e.etype = 1 + (op - operators);
235 e.off = 0;
236 e.out = nullptr;
237 e.block = nullptr;
238 e.ctx = ctx;
239 e.isparen = false;
240 return e;
241 }
242
syparen(lex_ctx_t ctx,size_t off)243 static sy_elem syparen(lex_ctx_t ctx, size_t off) {
244 sy_elem e;
245 e.etype = 0;
246 e.off = off;
247 e.out = nullptr;
248 e.block = nullptr;
249 e.ctx = ctx;
250 e.isparen = true;
251 return e;
252 }
253
254 /* With regular precedence rules, ent.foo[n] is the same as (ent.foo)[n],
255 * so we need to rotate it to become ent.(foo[n]).
256 */
rotate_entfield_array_index_nodes(ast_expression ** out)257 static bool rotate_entfield_array_index_nodes(ast_expression **out)
258 {
259 ast_array_index *index, *oldindex;
260 ast_entfield *entfield;
261
262 ast_value *field;
263 ast_expression *sub;
264 ast_expression *entity;
265
266 lex_ctx_t ctx = (*out)->m_context;
267
268 if (!ast_istype(*out, ast_array_index))
269 return false;
270 index = (ast_array_index*)*out;
271
272 if (!ast_istype(index->m_array, ast_entfield))
273 return false;
274 entfield = (ast_entfield*)index->m_array;
275
276 if (!ast_istype(entfield->m_field, ast_value))
277 return false;
278 field = (ast_value*)entfield->m_field;
279
280 sub = index->m_index;
281 entity = entfield->m_entity;
282
283 oldindex = index;
284
285 index = ast_array_index::make(ctx, field, sub);
286 entfield = new ast_entfield(ctx, entity, index);
287 *out = entfield;
288
289 oldindex->m_array = nullptr;
290 oldindex->m_index = nullptr;
291 delete oldindex;
292
293 return true;
294 }
295
check_write_to(lex_ctx_t ctx,ast_expression * expr)296 static bool check_write_to(lex_ctx_t ctx, ast_expression *expr)
297 {
298 if (ast_istype(expr, ast_value)) {
299 ast_value *val = (ast_value*)expr;
300 if (val->m_cvq == CV_CONST) {
301 if (val->m_name[0] == '#') {
302 compile_error(ctx, "invalid assignment to a literal constant");
303 return false;
304 }
305 /*
306 * To work around quakeworld we must elide the error and make it
307 * a warning instead.
308 */
309 if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_QCC)
310 compile_error(ctx, "assignment to constant `%s`", val->m_name);
311 else
312 (void)!compile_warning(ctx, WARN_CONST_OVERWRITE, "assignment to constant `%s`", val->m_name);
313 return false;
314 }
315 }
316 return true;
317 }
318
parser_sy_apply_operator(parser_t * parser,shunt * sy)319 static bool parser_sy_apply_operator(parser_t *parser, shunt *sy)
320 {
321 const oper_info *op;
322 lex_ctx_t ctx;
323 ast_expression *out = nullptr;
324 ast_expression *exprs[3];
325 ast_block *blocks[3];
326 ast_binstore *asbinstore;
327 size_t i, assignop, addop, subop;
328 qcint_t generated_op = 0;
329
330 char ty1[1024];
331 char ty2[1024];
332
333 if (sy->ops.empty()) {
334 parseerror(parser, "internal error: missing operator");
335 return false;
336 }
337
338 if (sy->ops.back().isparen) {
339 parseerror(parser, "unmatched parenthesis");
340 return false;
341 }
342
343 op = &operators[sy->ops.back().etype - 1];
344 ctx = sy->ops.back().ctx;
345
346 if (sy->out.size() < op->operands) {
347 if (op->flags & OP_PREFIX)
348 compile_error(ctx, "expected expression after unary operator `%s`", op->op, (int)op->id);
349 else /* this should have errored previously already */
350 compile_error(ctx, "expected expression after operator `%s`", op->op, (int)op->id);
351 return false;
352 }
353
354 sy->ops.pop_back();
355
356 /* op(:?) has no input and no output */
357 if (!op->operands)
358 return true;
359
360 sy->out.erase(sy->out.end() - op->operands, sy->out.end());
361 for (i = 0; i < op->operands; ++i) {
362 exprs[i] = sy->out[sy->out.size()+i].out;
363 blocks[i] = sy->out[sy->out.size()+i].block;
364
365 if (exprs[i]->m_vtype == TYPE_NOEXPR &&
366 !(i != 0 && op->id == opid2('?',':')) &&
367 !(i == 1 && op->id == opid1('.')))
368 {
369 if (ast_istype(exprs[i], ast_label))
370 compile_error(exprs[i]->m_context, "expected expression, got an unknown identifier");
371 else
372 compile_error(exprs[i]->m_context, "not an expression");
373 (void)!compile_warning(exprs[i]->m_context, WARN_DEBUG, "expression %u\n", (unsigned int)i);
374 }
375 }
376
377 if (blocks[0] && blocks[0]->m_exprs.empty() && op->id != opid1(',')) {
378 compile_error(ctx, "internal error: operator cannot be applied on empty blocks");
379 return false;
380 }
381
382 #define NotSameType(T) \
383 (exprs[0]->m_vtype != exprs[1]->m_vtype || \
384 exprs[0]->m_vtype != T)
385
386 switch (op->id)
387 {
388 default:
389 compile_error(ctx, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id);
390 return false;
391
392 case opid1('.'):
393 if (exprs[0]->m_vtype == TYPE_VECTOR &&
394 exprs[1]->m_vtype == TYPE_NOEXPR)
395 {
396 if (exprs[1] == parser->const_vec[0])
397 out = ast_member::make(ctx, exprs[0], 0, "");
398 else if (exprs[1] == parser->const_vec[1])
399 out = ast_member::make(ctx, exprs[0], 1, "");
400 else if (exprs[1] == parser->const_vec[2])
401 out = ast_member::make(ctx, exprs[0], 2, "");
402 else {
403 compile_error(ctx, "access to invalid vector component");
404 return false;
405 }
406 }
407 else if (exprs[0]->m_vtype == TYPE_ENTITY) {
408 if (exprs[1]->m_vtype != TYPE_FIELD) {
409 compile_error(exprs[1]->m_context, "type error: right hand of member-operand should be an entity-field");
410 return false;
411 }
412 out = new ast_entfield(ctx, exprs[0], exprs[1]);
413 }
414 else if (exprs[0]->m_vtype == TYPE_VECTOR) {
415 compile_error(exprs[1]->m_context, "vectors cannot be accessed this way");
416 return false;
417 }
418 else {
419 compile_error(exprs[1]->m_context, "type error: member-of operator on something that is not an entity or vector");
420 return false;
421 }
422 break;
423
424 case opid1('['):
425 if (exprs[0]->m_vtype != TYPE_ARRAY &&
426 !(exprs[0]->m_vtype == TYPE_FIELD &&
427 exprs[0]->m_next->m_vtype == TYPE_ARRAY))
428 {
429 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
430 compile_error(exprs[0]->m_context, "cannot index value of type %s", ty1);
431 return false;
432 }
433 if (exprs[1]->m_vtype != TYPE_FLOAT) {
434 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
435 compile_error(exprs[1]->m_context, "index must be of type float, not %s", ty1);
436 return false;
437 }
438 out = ast_array_index::make(ctx, exprs[0], exprs[1]);
439 rotate_entfield_array_index_nodes(&out);
440 break;
441
442 case opid1(','):
443 if (sy->paren.size() && sy->paren.back() == PAREN_FUNC) {
444 sy->out.push_back(syexp(ctx, exprs[0]));
445 sy->out.push_back(syexp(ctx, exprs[1]));
446 sy->argc.back()++;
447 return true;
448 }
449 if (blocks[0]) {
450 if (!blocks[0]->addExpr(exprs[1]))
451 return false;
452 } else {
453 blocks[0] = new ast_block(ctx);
454 if (!blocks[0]->addExpr(exprs[0]) ||
455 !blocks[0]->addExpr(exprs[1]))
456 {
457 return false;
458 }
459 }
460 blocks[0]->setType(*exprs[1]);
461
462 sy->out.push_back(syblock(ctx, blocks[0]));
463 return true;
464
465 case opid2('+','P'):
466 out = exprs[0];
467 break;
468 case opid2('-','P'):
469 if ((out = parser->m_fold.op(op, exprs)))
470 break;
471
472 if (exprs[0]->m_vtype != TYPE_FLOAT &&
473 exprs[0]->m_vtype != TYPE_VECTOR) {
474 compile_error(ctx, "invalid types used in unary expression: cannot negate type %s",
475 type_name[exprs[0]->m_vtype]);
476 return false;
477 }
478 if (exprs[0]->m_vtype == TYPE_FLOAT)
479 out = ast_unary::make(ctx, VINSTR_NEG_F, exprs[0]);
480 else
481 out = ast_unary::make(ctx, VINSTR_NEG_V, exprs[0]);
482 break;
483
484 case opid2('!','P'):
485 if (!(out = parser->m_fold.op(op, exprs))) {
486 switch (exprs[0]->m_vtype) {
487 case TYPE_FLOAT:
488 out = ast_unary::make(ctx, INSTR_NOT_F, exprs[0]);
489 break;
490 case TYPE_VECTOR:
491 out = ast_unary::make(ctx, INSTR_NOT_V, exprs[0]);
492 break;
493 case TYPE_STRING:
494 if (OPTS_FLAG(TRUE_EMPTY_STRINGS))
495 out = ast_unary::make(ctx, INSTR_NOT_F, exprs[0]);
496 else
497 out = ast_unary::make(ctx, INSTR_NOT_S, exprs[0]);
498 break;
499 /* we don't constant-fold NOT for these types */
500 case TYPE_ENTITY:
501 out = ast_unary::make(ctx, INSTR_NOT_ENT, exprs[0]);
502 break;
503 case TYPE_FUNCTION:
504 out = ast_unary::make(ctx, INSTR_NOT_FNC, exprs[0]);
505 break;
506 default:
507 compile_error(ctx, "invalid types used in expression: cannot logically negate type %s",
508 type_name[exprs[0]->m_vtype]);
509 return false;
510 }
511 }
512 break;
513
514 case opid1('+'):
515 if (exprs[0]->m_vtype != exprs[1]->m_vtype ||
516 (exprs[0]->m_vtype != TYPE_VECTOR && exprs[0]->m_vtype != TYPE_FLOAT) )
517 {
518 compile_error(ctx, "invalid types used in expression: cannot add type %s and %s",
519 type_name[exprs[0]->m_vtype],
520 type_name[exprs[1]->m_vtype]);
521 return false;
522 }
523 if (!(out = parser->m_fold.op(op, exprs))) {
524 switch (exprs[0]->m_vtype) {
525 case TYPE_FLOAT:
526 out = fold::binary(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
527 break;
528 case TYPE_VECTOR:
529 out = fold::binary(ctx, INSTR_ADD_V, exprs[0], exprs[1]);
530 break;
531 default:
532 compile_error(ctx, "invalid types used in expression: cannot add type %s and %s",
533 type_name[exprs[0]->m_vtype],
534 type_name[exprs[1]->m_vtype]);
535 return false;
536 }
537 }
538 break;
539 case opid1('-'):
540 if (exprs[0]->m_vtype != exprs[1]->m_vtype ||
541 (exprs[0]->m_vtype != TYPE_VECTOR && exprs[0]->m_vtype != TYPE_FLOAT))
542 {
543 compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s",
544 type_name[exprs[1]->m_vtype],
545 type_name[exprs[0]->m_vtype]);
546 return false;
547 }
548 if (!(out = parser->m_fold.op(op, exprs))) {
549 switch (exprs[0]->m_vtype) {
550 case TYPE_FLOAT:
551 out = fold::binary(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
552 break;
553 case TYPE_VECTOR:
554 out = fold::binary(ctx, INSTR_SUB_V, exprs[0], exprs[1]);
555 break;
556 default:
557 compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s",
558 type_name[exprs[1]->m_vtype],
559 type_name[exprs[0]->m_vtype]);
560 return false;
561 }
562 }
563 break;
564 case opid1('*'):
565 if (exprs[0]->m_vtype != exprs[1]->m_vtype &&
566 !(exprs[0]->m_vtype == TYPE_VECTOR &&
567 exprs[1]->m_vtype == TYPE_FLOAT) &&
568 !(exprs[1]->m_vtype == TYPE_VECTOR &&
569 exprs[0]->m_vtype == TYPE_FLOAT)
570 )
571 {
572 compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s",
573 type_name[exprs[1]->m_vtype],
574 type_name[exprs[0]->m_vtype]);
575 return false;
576 }
577 if (!(out = parser->m_fold.op(op, exprs))) {
578 switch (exprs[0]->m_vtype) {
579 case TYPE_FLOAT:
580 if (exprs[1]->m_vtype == TYPE_VECTOR)
581 out = fold::binary(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
582 else
583 out = fold::binary(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
584 break;
585 case TYPE_VECTOR:
586 if (exprs[1]->m_vtype == TYPE_FLOAT)
587 out = fold::binary(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
588 else
589 out = fold::binary(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
590 break;
591 default:
592 compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s",
593 type_name[exprs[1]->m_vtype],
594 type_name[exprs[0]->m_vtype]);
595 return false;
596 }
597 }
598 break;
599
600 case opid1('/'):
601 if (exprs[1]->m_vtype != TYPE_FLOAT) {
602 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
603 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
604 compile_error(ctx, "invalid types used in expression: cannot divide types %s and %s", ty1, ty2);
605 return false;
606 }
607 if (!(out = parser->m_fold.op(op, exprs))) {
608 if (exprs[0]->m_vtype == TYPE_FLOAT)
609 out = fold::binary(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
610 else {
611 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
612 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
613 compile_error(ctx, "invalid types used in expression: cannot divide types %s and %s", ty1, ty2);
614 return false;
615 }
616 }
617 break;
618
619 case opid1('%'):
620 if (NotSameType(TYPE_FLOAT)) {
621 compile_error(ctx, "invalid types used in expression: cannot perform modulo operation between types %s and %s",
622 type_name[exprs[0]->m_vtype],
623 type_name[exprs[1]->m_vtype]);
624 return false;
625 } else if (!(out = parser->m_fold.op(op, exprs))) {
626 /* generate a call to __builtin_mod */
627 ast_expression *mod = parser->m_intrin.func("mod");
628 ast_call *call = nullptr;
629 if (!mod) return false; /* can return null for missing floor */
630
631 call = ast_call::make(parser_ctx(parser), mod);
632 call->m_params.push_back(exprs[0]);
633 call->m_params.push_back(exprs[1]);
634
635 out = call;
636 }
637 break;
638
639 case opid2('%','='):
640 compile_error(ctx, "%= is unimplemented");
641 return false;
642
643 case opid1('|'):
644 case opid1('&'):
645 case opid1('^'):
646 if ( !(exprs[0]->m_vtype == TYPE_FLOAT && exprs[1]->m_vtype == TYPE_FLOAT) &&
647 !(exprs[0]->m_vtype == TYPE_VECTOR && exprs[1]->m_vtype == TYPE_FLOAT) &&
648 !(exprs[0]->m_vtype == TYPE_VECTOR && exprs[1]->m_vtype == TYPE_VECTOR))
649 {
650 compile_error(ctx, "invalid types used in expression: cannot perform bit operations between types %s and %s",
651 type_name[exprs[0]->m_vtype],
652 type_name[exprs[1]->m_vtype]);
653 return false;
654 }
655
656 if (!(out = parser->m_fold.op(op, exprs))) {
657 /*
658 * IF the first expression is float, the following will be too
659 * since scalar ^ vector is not allowed.
660 */
661 if (exprs[0]->m_vtype == TYPE_FLOAT) {
662 out = fold::binary(ctx,
663 (op->id == opid1('^') ? VINSTR_BITXOR : op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
664 exprs[0], exprs[1]);
665 } else {
666 /*
667 * The first is a vector: vector is allowed to bitop with vector and
668 * with scalar, branch here for the second operand.
669 */
670 if (exprs[1]->m_vtype == TYPE_VECTOR) {
671 /*
672 * Bitop all the values of the vector components against the
673 * vectors components in question.
674 */
675 out = fold::binary(ctx,
676 (op->id == opid1('^') ? VINSTR_BITXOR_V : op->id == opid1('|') ? VINSTR_BITOR_V : VINSTR_BITAND_V),
677 exprs[0], exprs[1]);
678 } else {
679 out = fold::binary(ctx,
680 (op->id == opid1('^') ? VINSTR_BITXOR_VF : op->id == opid1('|') ? VINSTR_BITOR_VF : VINSTR_BITAND_VF),
681 exprs[0], exprs[1]);
682 }
683 }
684 }
685 break;
686
687 case opid2('<','<'):
688 case opid2('>','>'):
689 if (NotSameType(TYPE_FLOAT)) {
690 compile_error(ctx, "invalid types used in expression: cannot perform shift between types %s and %s",
691 type_name[exprs[0]->m_vtype],
692 type_name[exprs[1]->m_vtype]);
693 return false;
694 }
695
696 if (!(out = parser->m_fold.op(op, exprs))) {
697 ast_expression *shift = parser->m_intrin.func((op->id == opid2('<','<')) ? "__builtin_lshift" : "__builtin_rshift");
698 ast_call *call = ast_call::make(parser_ctx(parser), shift);
699 call->m_params.push_back(exprs[0]);
700 call->m_params.push_back(exprs[1]);
701 out = call;
702 }
703 break;
704
705 case opid3('<','<','='):
706 case opid3('>','>','='):
707 if (NotSameType(TYPE_FLOAT)) {
708 compile_error(ctx, "invalid types used in expression: cannot perform shift operation between types %s and %s",
709 type_name[exprs[0]->m_vtype],
710 type_name[exprs[1]->m_vtype]);
711 return false;
712 }
713
714 if(!(out = parser->m_fold.op(op, exprs))) {
715 ast_expression *shift = parser->m_intrin.func((op->id == opid3('<','<','=')) ? "__builtin_lshift" : "__builtin_rshift");
716 ast_call *call = ast_call::make(parser_ctx(parser), shift);
717 call->m_params.push_back(exprs[0]);
718 call->m_params.push_back(exprs[1]);
719 out = new ast_store(
720 parser_ctx(parser),
721 INSTR_STORE_F,
722 exprs[0],
723 call
724 );
725 }
726
727 break;
728
729 case opid2('|','|'):
730 generated_op += 1; /* INSTR_OR */
731 case opid2('&','&'):
732 generated_op += INSTR_AND;
733 if (!(out = parser->m_fold.op(op, exprs))) {
734 if (OPTS_FLAG(PERL_LOGIC) && !exprs[0]->compareType(*exprs[1])) {
735 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
736 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
737 compile_error(ctx, "invalid types for logical operation with -fperl-logic: %s and %s", ty1, ty2);
738 return false;
739 }
740 for (i = 0; i < 2; ++i) {
741 if (OPTS_FLAG(CORRECT_LOGIC) && exprs[i]->m_vtype == TYPE_VECTOR) {
742 out = ast_unary::make(ctx, INSTR_NOT_V, exprs[i]);
743 if (!out) break;
744 out = ast_unary::make(ctx, INSTR_NOT_F, out);
745 if (!out) break;
746 exprs[i] = out; out = nullptr;
747 if (OPTS_FLAG(PERL_LOGIC)) {
748 /* here we want to keep the right expressions' type */
749 break;
750 }
751 }
752 else if (OPTS_FLAG(FALSE_EMPTY_STRINGS) && exprs[i]->m_vtype == TYPE_STRING) {
753 out = ast_unary::make(ctx, INSTR_NOT_S, exprs[i]);
754 if (!out) break;
755 out = ast_unary::make(ctx, INSTR_NOT_F, out);
756 if (!out) break;
757 exprs[i] = out; out = nullptr;
758 if (OPTS_FLAG(PERL_LOGIC)) {
759 /* here we want to keep the right expressions' type */
760 break;
761 }
762 }
763 }
764 out = fold::binary(ctx, generated_op, exprs[0], exprs[1]);
765 }
766 break;
767
768 case opid2('?',':'):
769 if (sy->paren.back() != PAREN_TERNARY2) {
770 compile_error(ctx, "mismatched parenthesis/ternary");
771 return false;
772 }
773 sy->paren.pop_back();
774 if (!exprs[1]->compareType(*exprs[2])) {
775 ast_type_to_string(exprs[1], ty1, sizeof(ty1));
776 ast_type_to_string(exprs[2], ty2, sizeof(ty2));
777 compile_error(ctx, "operands of ternary expression must have the same type, got %s and %s", ty1, ty2);
778 return false;
779 }
780 if (!(out = parser->m_fold.op(op, exprs)))
781 out = new ast_ternary(ctx, exprs[0], exprs[1], exprs[2]);
782 break;
783
784 case opid2('*', '*'):
785 if (NotSameType(TYPE_FLOAT)) {
786 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
787 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
788 compile_error(ctx, "invalid types used in exponentiation: %s and %s",
789 ty1, ty2);
790 return false;
791 }
792
793 if (!(out = parser->m_fold.op(op, exprs))) {
794 ast_call *gencall = ast_call::make(parser_ctx(parser), parser->m_intrin.func("pow"));
795 gencall->m_params.push_back(exprs[0]);
796 gencall->m_params.push_back(exprs[1]);
797 out = gencall;
798 }
799 break;
800
801 case opid2('>', '<'):
802 if (NotSameType(TYPE_VECTOR)) {
803 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
804 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
805 compile_error(ctx, "invalid types used in cross product: %s and %s",
806 ty1, ty2);
807 return false;
808 }
809
810 if (!(out = parser->m_fold.op(op, exprs))) {
811 out = fold::binary(
812 parser_ctx(parser),
813 VINSTR_CROSS,
814 exprs[0],
815 exprs[1]
816 );
817 }
818
819 break;
820
821 case opid3('<','=','>'): /* -1, 0, or 1 */
822 if (NotSameType(TYPE_FLOAT)) {
823 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
824 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
825 compile_error(ctx, "invalid types used in comparision: %s and %s",
826 ty1, ty2);
827
828 return false;
829 }
830
831 if (!(out = parser->m_fold.op(op, exprs))) {
832 /* This whole block is NOT fold_binary safe */
833 ast_binary *eq = new ast_binary(ctx, INSTR_EQ_F, exprs[0], exprs[1]);
834
835 eq->m_refs = AST_REF_NONE;
836
837 /* if (lt) { */
838 out = new ast_ternary(ctx,
839 new ast_binary(ctx, INSTR_LT, exprs[0], exprs[1]),
840 /* out = -1 */
841 parser->m_fold.imm_float(2),
842 /* } else { */
843 /* if (eq) { */
844 new ast_ternary(ctx, eq,
845 /* out = 0 */
846 parser->m_fold.imm_float(0),
847 /* } else { */
848 /* out = 1 */
849 parser->m_fold.imm_float(1)
850 /* } */
851 )
852 /* } */
853 );
854
855 }
856 break;
857
858 case opid1('>'):
859 generated_op += 1; /* INSTR_GT */
860 case opid1('<'):
861 generated_op += 1; /* INSTR_LT */
862 case opid2('>', '='):
863 generated_op += 1; /* INSTR_GE */
864 case opid2('<', '='):
865 generated_op += INSTR_LE;
866 if (NotSameType(TYPE_FLOAT)) {
867 compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
868 type_name[exprs[0]->m_vtype],
869 type_name[exprs[1]->m_vtype]);
870 return false;
871 }
872 if (!(out = parser->m_fold.op(op, exprs)))
873 out = fold::binary(ctx, generated_op, exprs[0], exprs[1]);
874 break;
875 case opid2('!', '='):
876 if (exprs[0]->m_vtype != exprs[1]->m_vtype) {
877 compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
878 type_name[exprs[0]->m_vtype],
879 type_name[exprs[1]->m_vtype]);
880 return false;
881 }
882 if (!(out = parser->m_fold.op(op, exprs)))
883 out = fold::binary(ctx, type_ne_instr[exprs[0]->m_vtype], exprs[0], exprs[1]);
884 break;
885 case opid2('=', '='):
886 if (exprs[0]->m_vtype != exprs[1]->m_vtype) {
887 compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
888 type_name[exprs[0]->m_vtype],
889 type_name[exprs[1]->m_vtype]);
890 return false;
891 }
892 if (!(out = parser->m_fold.op(op, exprs)))
893 out = fold::binary(ctx, type_eq_instr[exprs[0]->m_vtype], exprs[0], exprs[1]);
894 break;
895
896 case opid1('='):
897 if (ast_istype(exprs[0], ast_entfield)) {
898 ast_expression *field = ((ast_entfield*)exprs[0])->m_field;
899 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
900 exprs[0]->m_vtype == TYPE_FIELD &&
901 exprs[0]->m_next->m_vtype == TYPE_VECTOR)
902 {
903 assignop = type_storep_instr[TYPE_VECTOR];
904 }
905 else
906 assignop = type_storep_instr[exprs[0]->m_vtype];
907 if (assignop == VINSTR_END || !field->m_next->compareType(*exprs[1]))
908 {
909 ast_type_to_string(field->m_next, ty1, sizeof(ty1));
910 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
911 if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) &&
912 field->m_next->m_vtype == TYPE_FUNCTION &&
913 exprs[1]->m_vtype == TYPE_FUNCTION)
914 {
915 (void)!compile_warning(ctx, WARN_ASSIGN_FUNCTION_TYPES,
916 "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
917 }
918 else
919 compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
920 }
921 }
922 else
923 {
924 if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) &&
925 exprs[0]->m_vtype == TYPE_FIELD &&
926 exprs[0]->m_next->m_vtype == TYPE_VECTOR)
927 {
928 assignop = type_store_instr[TYPE_VECTOR];
929 }
930 else {
931 assignop = type_store_instr[exprs[0]->m_vtype];
932 }
933
934 if (assignop == VINSTR_END) {
935 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
936 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
937 compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
938 }
939 else if (!exprs[0]->compareType(*exprs[1]))
940 {
941 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
942 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
943 if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) &&
944 exprs[0]->m_vtype == TYPE_FUNCTION &&
945 exprs[1]->m_vtype == TYPE_FUNCTION)
946 {
947 (void)!compile_warning(ctx, WARN_ASSIGN_FUNCTION_TYPES,
948 "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
949 }
950 else
951 compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1);
952 }
953 }
954 (void)check_write_to(ctx, exprs[0]);
955 /* When we're a vector of part of an entity field we use STOREP */
956 if (ast_istype(exprs[0], ast_member) && ast_istype(((ast_member*)exprs[0])->m_owner, ast_entfield))
957 assignop = INSTR_STOREP_F;
958 out = new ast_store(ctx, assignop, exprs[0], exprs[1]);
959 break;
960 case opid3('+','+','P'):
961 case opid3('-','-','P'):
962 /* prefix ++ */
963 if (exprs[0]->m_vtype != TYPE_FLOAT) {
964 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
965 compile_error(exprs[0]->m_context, "invalid type for prefix increment: %s", ty1);
966 return false;
967 }
968 if (op->id == opid3('+','+','P'))
969 addop = INSTR_ADD_F;
970 else
971 addop = INSTR_SUB_F;
972 (void)check_write_to(exprs[0]->m_context, exprs[0]);
973 if (ast_istype(exprs[0], ast_entfield)) {
974 out = new ast_binstore(ctx, INSTR_STOREP_F, addop,
975 exprs[0],
976 parser->m_fold.imm_float(1));
977 } else {
978 out = new ast_binstore(ctx, INSTR_STORE_F, addop,
979 exprs[0],
980 parser->m_fold.imm_float(1));
981 }
982 break;
983 case opid3('S','+','+'):
984 case opid3('S','-','-'):
985 /* prefix ++ */
986 if (exprs[0]->m_vtype != TYPE_FLOAT) {
987 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
988 compile_error(exprs[0]->m_context, "invalid type for suffix increment: %s", ty1);
989 return false;
990 }
991 if (op->id == opid3('S','+','+')) {
992 addop = INSTR_ADD_F;
993 subop = INSTR_SUB_F;
994 } else {
995 addop = INSTR_SUB_F;
996 subop = INSTR_ADD_F;
997 }
998 (void)check_write_to(exprs[0]->m_context, exprs[0]);
999 if (ast_istype(exprs[0], ast_entfield)) {
1000 out = new ast_binstore(ctx, INSTR_STOREP_F, addop,
1001 exprs[0],
1002 parser->m_fold.imm_float(1));
1003 } else {
1004 out = new ast_binstore(ctx, INSTR_STORE_F, addop,
1005 exprs[0],
1006 parser->m_fold.imm_float(1));
1007 }
1008 if (!out)
1009 return false;
1010 out = fold::binary(ctx, subop,
1011 out,
1012 parser->m_fold.imm_float(1));
1013
1014 break;
1015 case opid2('+','='):
1016 case opid2('-','='):
1017 if (exprs[0]->m_vtype != exprs[1]->m_vtype ||
1018 (exprs[0]->m_vtype != TYPE_VECTOR && exprs[0]->m_vtype != TYPE_FLOAT) )
1019 {
1020 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1021 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1022 compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s",
1023 ty1, ty2);
1024 return false;
1025 }
1026 (void)check_write_to(ctx, exprs[0]);
1027 if (ast_istype(exprs[0], ast_entfield))
1028 assignop = type_storep_instr[exprs[0]->m_vtype];
1029 else
1030 assignop = type_store_instr[exprs[0]->m_vtype];
1031 switch (exprs[0]->m_vtype) {
1032 case TYPE_FLOAT:
1033 out = new ast_binstore(ctx, assignop,
1034 (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F),
1035 exprs[0], exprs[1]);
1036 break;
1037 case TYPE_VECTOR:
1038 out = new ast_binstore(ctx, assignop,
1039 (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V),
1040 exprs[0], exprs[1]);
1041 break;
1042 default:
1043 compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s",
1044 type_name[exprs[0]->m_vtype],
1045 type_name[exprs[1]->m_vtype]);
1046 return false;
1047 };
1048 break;
1049 case opid2('*','='):
1050 case opid2('/','='):
1051 if (exprs[1]->m_vtype != TYPE_FLOAT ||
1052 !(exprs[0]->m_vtype == TYPE_FLOAT ||
1053 exprs[0]->m_vtype == TYPE_VECTOR))
1054 {
1055 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1056 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1057 compile_error(ctx, "invalid types used in expression: %s and %s",
1058 ty1, ty2);
1059 return false;
1060 }
1061 (void)check_write_to(ctx, exprs[0]);
1062 if (ast_istype(exprs[0], ast_entfield))
1063 assignop = type_storep_instr[exprs[0]->m_vtype];
1064 else
1065 assignop = type_store_instr[exprs[0]->m_vtype];
1066 switch (exprs[0]->m_vtype) {
1067 case TYPE_FLOAT:
1068 out = new ast_binstore(ctx, assignop,
1069 (op->id == opid2('*','=') ? INSTR_MUL_F : INSTR_DIV_F),
1070 exprs[0], exprs[1]);
1071 break;
1072 case TYPE_VECTOR:
1073 if (op->id == opid2('*','=')) {
1074 out = new ast_binstore(ctx, assignop, INSTR_MUL_VF,
1075 exprs[0], exprs[1]);
1076 } else {
1077 out = fold::binary(ctx, INSTR_DIV_F,
1078 parser->m_fold.imm_float(1),
1079 exprs[1]);
1080 if (!out) {
1081 compile_error(ctx, "internal error: failed to generate division");
1082 return false;
1083 }
1084 out = new ast_binstore(ctx, assignop, INSTR_MUL_VF,
1085 exprs[0], out);
1086 }
1087 break;
1088 default:
1089 compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s",
1090 type_name[exprs[0]->m_vtype],
1091 type_name[exprs[1]->m_vtype]);
1092 return false;
1093 };
1094 break;
1095 case opid2('&','='):
1096 case opid2('|','='):
1097 case opid2('^','='):
1098 if (NotSameType(TYPE_FLOAT) && NotSameType(TYPE_VECTOR)) {
1099 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1100 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1101 compile_error(ctx, "invalid types used in expression: %s and %s",
1102 ty1, ty2);
1103 return false;
1104 }
1105 (void)check_write_to(ctx, exprs[0]);
1106 if (ast_istype(exprs[0], ast_entfield))
1107 assignop = type_storep_instr[exprs[0]->m_vtype];
1108 else
1109 assignop = type_store_instr[exprs[0]->m_vtype];
1110 if (exprs[0]->m_vtype == TYPE_FLOAT)
1111 out = new ast_binstore(ctx, assignop,
1112 (op->id == opid2('^','=') ? VINSTR_BITXOR : op->id == opid2('&','=') ? INSTR_BITAND : INSTR_BITOR),
1113 exprs[0], exprs[1]);
1114 else
1115 out = new ast_binstore(ctx, assignop,
1116 (op->id == opid2('^','=') ? VINSTR_BITXOR_V : op->id == opid2('&','=') ? VINSTR_BITAND_V : VINSTR_BITOR_V),
1117 exprs[0], exprs[1]);
1118 break;
1119 case opid3('&','~','='):
1120 /* This is like: a &= ~(b);
1121 * But QC has no bitwise-not, so we implement it as
1122 * a -= a & (b);
1123 */
1124 if (NotSameType(TYPE_FLOAT) && NotSameType(TYPE_VECTOR)) {
1125 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1126 ast_type_to_string(exprs[1], ty2, sizeof(ty2));
1127 compile_error(ctx, "invalid types used in expression: %s and %s",
1128 ty1, ty2);
1129 return false;
1130 }
1131 if (ast_istype(exprs[0], ast_entfield))
1132 assignop = type_storep_instr[exprs[0]->m_vtype];
1133 else
1134 assignop = type_store_instr[exprs[0]->m_vtype];
1135 if (exprs[0]->m_vtype == TYPE_FLOAT)
1136 out = fold::binary(ctx, INSTR_BITAND, exprs[0], exprs[1]);
1137 else
1138 out = fold::binary(ctx, VINSTR_BITAND_V, exprs[0], exprs[1]);
1139 if (!out)
1140 return false;
1141 (void)check_write_to(ctx, exprs[0]);
1142 if (exprs[0]->m_vtype == TYPE_FLOAT)
1143 asbinstore = new ast_binstore(ctx, assignop, INSTR_SUB_F, exprs[0], out);
1144 else
1145 asbinstore = new ast_binstore(ctx, assignop, INSTR_SUB_V, exprs[0], out);
1146 asbinstore->m_keep_dest = true;
1147 out = asbinstore;
1148 break;
1149
1150 case opid3('l', 'e', 'n'):
1151 if (exprs[0]->m_vtype != TYPE_STRING && exprs[0]->m_vtype != TYPE_ARRAY) {
1152 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1153 compile_error(exprs[0]->m_context, "invalid type for length operator: %s", ty1);
1154 return false;
1155 }
1156 /* strings must be const, arrays are statically sized */
1157 if (exprs[0]->m_vtype == TYPE_STRING &&
1158 !(((ast_value*)exprs[0])->m_hasvalue && ((ast_value*)exprs[0])->m_cvq == CV_CONST))
1159 {
1160 compile_error(exprs[0]->m_context, "operand of length operator not a valid constant expression");
1161 return false;
1162 }
1163 out = parser->m_fold.op(op, exprs);
1164 break;
1165
1166 case opid2('~', 'P'):
1167 if (exprs[0]->m_vtype != TYPE_FLOAT && exprs[0]->m_vtype != TYPE_VECTOR) {
1168 ast_type_to_string(exprs[0], ty1, sizeof(ty1));
1169 compile_error(exprs[0]->m_context, "invalid type for bit not: %s", ty1);
1170 return false;
1171 }
1172 if (!(out = parser->m_fold.op(op, exprs))) {
1173 if (exprs[0]->m_vtype == TYPE_FLOAT) {
1174 out = fold::binary(ctx, INSTR_SUB_F, parser->m_fold.imm_float(2), exprs[0]);
1175 } else {
1176 out = fold::binary(ctx, INSTR_SUB_V, parser->m_fold.imm_vector(1), exprs[0]);
1177 }
1178 }
1179 break;
1180 }
1181 #undef NotSameType
1182 if (!out) {
1183 compile_error(ctx, "failed to apply operator %s", op->op);
1184 return false;
1185 }
1186
1187 sy->out.push_back(syexp(ctx, out));
1188 return true;
1189 }
1190
parser_close_call(parser_t * parser,shunt * sy)1191 static bool parser_close_call(parser_t *parser, shunt *sy)
1192 {
1193 /* was a function call */
1194 ast_expression *fun;
1195 ast_value *funval = nullptr;
1196 ast_call *call;
1197
1198 size_t fid;
1199 size_t paramcount, i;
1200 bool fold = true;
1201
1202 fid = sy->ops.back().off;
1203 sy->ops.pop_back();
1204
1205 /* out[fid] is the function
1206 * everything above is parameters...
1207 */
1208 if (sy->argc.empty()) {
1209 parseerror(parser, "internal error: no argument counter available");
1210 return false;
1211 }
1212
1213 paramcount = sy->argc.back();
1214 sy->argc.pop_back();
1215
1216 if (sy->out.size() < fid) {
1217 parseerror(parser, "internal error: broken function call %zu < %zu+%zu\n",
1218 sy->out.size(),
1219 fid,
1220 paramcount);
1221 return false;
1222 }
1223
1224 /*
1225 * TODO handle this at the intrinsic level with an ast_intrinsic
1226 * node and codegen.
1227 */
1228 if ((fun = sy->out[fid].out) == parser->m_intrin.debug_typestring()) {
1229 char ty[1024];
1230 if (fid+2 != sy->out.size() || sy->out.back().block) {
1231 parseerror(parser, "intrinsic __builtin_debug_typestring requires exactly 1 parameter");
1232 return false;
1233 }
1234 ast_type_to_string(sy->out.back().out, ty, sizeof(ty));
1235 ast_unref(sy->out.back().out);
1236 sy->out[fid] = syexp(sy->out.back().out->m_context,
1237 parser->m_fold.constgen_string(ty, false));
1238 sy->out.pop_back();
1239 return true;
1240 }
1241
1242 /*
1243 * Now we need to determine if the function that is being called is
1244 * an intrinsic so we can evaluate if the arguments to it are constant
1245 * and than fruitfully fold them.
1246 */
1247 #define fold_can_1(X) \
1248 (ast_istype(((X)), ast_value) && (X)->m_hasvalue && ((X)->m_cvq == CV_CONST) && \
1249 ((X))->m_vtype != TYPE_FUNCTION)
1250
1251 if (fid + 1 < sy->out.size())
1252 ++paramcount;
1253
1254 for (i = 0; i < paramcount; ++i) {
1255 if (!fold_can_1((ast_value*)sy->out[fid + 1 + i].out)) {
1256 fold = false;
1257 break;
1258 }
1259 }
1260
1261 /*
1262 * All is well which ends well, if we make it into here we can ignore the
1263 * intrinsic call and just evaluate it i.e constant fold it.
1264 */
1265 if (fold && ast_istype(fun, ast_value) && ((ast_value*)fun)->m_intrinsic) {
1266 ast_expression **exprs = nullptr;
1267 ast_expression *foldval = nullptr;
1268
1269 for (i = 0; i < paramcount; i++)
1270 vec_push(exprs, sy->out[fid+1 + i].out);
1271
1272 if (!(foldval = parser->m_intrin.do_fold((ast_value*)fun, exprs))) {
1273 vec_free(exprs);
1274 goto fold_leave;
1275 }
1276
1277 /*
1278 * Blub: what sorts of unreffing and resizing of
1279 * sy->out should I be doing here?
1280 */
1281 sy->out[fid] = syexp(foldval->m_context, foldval);
1282 sy->out.erase(sy->out.end() - paramcount, sy->out.end());
1283 vec_free(exprs);
1284
1285 return true;
1286 }
1287
1288 fold_leave:
1289 call = ast_call::make(sy->ops[sy->ops.size()].ctx, fun);
1290
1291 if (!call)
1292 return false;
1293
1294 if (fid+1 + paramcount != sy->out.size()) {
1295 parseerror(parser, "internal error: parameter count mismatch: (%zu+1+%zu), %zu",
1296 fid,
1297 paramcount,
1298 sy->out.size());
1299 return false;
1300 }
1301
1302 for (i = 0; i < paramcount; ++i)
1303 call->m_params.push_back(sy->out[fid+1 + i].out);
1304 sy->out.erase(sy->out.end() - paramcount, sy->out.end());
1305 (void)!call->checkTypes(parser->function->m_function_type->m_varparam);
1306 if (parser->max_param_count < paramcount)
1307 parser->max_param_count = paramcount;
1308
1309 if (ast_istype(fun, ast_value)) {
1310 funval = (ast_value*)fun;
1311 if ((fun->m_flags & AST_FLAG_VARIADIC) &&
1312 !(/*funval->m_cvq == CV_CONST && */ funval->m_hasvalue && funval->m_constval.vfunc->m_builtin))
1313 {
1314 call->m_va_count = parser->m_fold.constgen_float((qcfloat_t)paramcount, false);
1315 }
1316 }
1317
1318 /* overwrite fid, the function, with a call */
1319 sy->out[fid] = syexp(call->m_context, call);
1320
1321 if (fun->m_vtype != TYPE_FUNCTION) {
1322 parseerror(parser, "not a function (%s)", type_name[fun->m_vtype]);
1323 return false;
1324 }
1325
1326 if (!fun->m_next) {
1327 parseerror(parser, "could not determine function return type");
1328 return false;
1329 } else {
1330 ast_value *fval = (ast_istype(fun, ast_value) ? ((ast_value*)fun) : nullptr);
1331
1332 if (fun->m_flags & AST_FLAG_DEPRECATED) {
1333 if (!fval) {
1334 return !parsewarning(parser, WARN_DEPRECATED,
1335 "call to function (which is marked deprecated)\n",
1336 "-> it has been declared here: %s:%i",
1337 fun->m_context.file, fun->m_context.line);
1338 }
1339 if (!fval->m_desc.length()) {
1340 return !parsewarning(parser, WARN_DEPRECATED,
1341 "call to `%s` (which is marked deprecated)\n"
1342 "-> `%s` declared here: %s:%i",
1343 fval->m_name, fval->m_name, fun->m_context.file, fun->m_context.line);
1344 }
1345 return !parsewarning(parser, WARN_DEPRECATED,
1346 "call to `%s` (deprecated: %s)\n"
1347 "-> `%s` declared here: %s:%i",
1348 fval->m_name, fval->m_desc, fval->m_name, fun->m_context.file,
1349 fun->m_context.line);
1350 }
1351
1352 if (fun->m_type_params.size() != paramcount &&
1353 !((fun->m_flags & AST_FLAG_VARIADIC) &&
1354 fun->m_type_params.size() < paramcount))
1355 {
1356 const char *fewmany = (fun->m_type_params.size() > paramcount) ? "few" : "many";
1357 if (fval)
1358 return !parsewarning(parser, WARN_INVALID_PARAMETER_COUNT,
1359 "too %s parameters for call to %s: expected %i, got %i\n"
1360 " -> `%s` has been declared here: %s:%i",
1361 fewmany, fval->m_name, (int)fun->m_type_params.size(), (int)paramcount,
1362 fval->m_name, fun->m_context.file, (int)fun->m_context.line);
1363 else
1364 return !parsewarning(parser, WARN_INVALID_PARAMETER_COUNT,
1365 "too %s parameters for function call: expected %i, got %i\n"
1366 " -> it has been declared here: %s:%i",
1367 fewmany, (int)fun->m_type_params.size(), (int)paramcount,
1368 fun->m_context.file, (int)fun->m_context.line);
1369 }
1370 }
1371
1372 return true;
1373 }
1374
parser_close_paren(parser_t * parser,shunt * sy)1375 static bool parser_close_paren(parser_t *parser, shunt *sy)
1376 {
1377 if (sy->ops.empty()) {
1378 parseerror(parser, "unmatched closing paren");
1379 return false;
1380 }
1381
1382 while (sy->ops.size()) {
1383 if (sy->ops.back().isparen) {
1384 if (sy->paren.back() == PAREN_FUNC) {
1385 sy->paren.pop_back();
1386 if (!parser_close_call(parser, sy))
1387 return false;
1388 break;
1389 }
1390 if (sy->paren.back() == PAREN_EXPR) {
1391 sy->paren.pop_back();
1392 if (sy->out.empty()) {
1393 compile_error(sy->ops.back().ctx, "empty paren expression");
1394 sy->ops.pop_back();
1395 return false;
1396 }
1397 sy->ops.pop_back();
1398 break;
1399 }
1400 if (sy->paren.back() == PAREN_INDEX) {
1401 sy->paren.pop_back();
1402 // pop off the parenthesis
1403 sy->ops.pop_back();
1404 /* then apply the index operator */
1405 if (!parser_sy_apply_operator(parser, sy))
1406 return false;
1407 break;
1408 }
1409 if (sy->paren.back() == PAREN_TERNARY1) {
1410 sy->paren.back() = PAREN_TERNARY2;
1411 // pop off the parenthesis
1412 sy->ops.pop_back();
1413 break;
1414 }
1415 compile_error(sy->ops.back().ctx, "invalid parenthesis");
1416 return false;
1417 }
1418 if (!parser_sy_apply_operator(parser, sy))
1419 return false;
1420 }
1421 return true;
1422 }
1423
parser_reclassify_token(parser_t * parser)1424 static void parser_reclassify_token(parser_t *parser)
1425 {
1426 size_t i;
1427 if (parser->tok >= TOKEN_START)
1428 return;
1429 for (i = 0; i < operator_count; ++i) {
1430 if (!strcmp(parser_tokval(parser), operators[i].op)) {
1431 parser->tok = TOKEN_OPERATOR;
1432 return;
1433 }
1434 }
1435 }
1436
parse_vararg_do(parser_t * parser)1437 static ast_expression* parse_vararg_do(parser_t *parser)
1438 {
1439 ast_expression *idx, *out;
1440 ast_value *typevar;
1441 ast_value *funtype = parser->function->m_function_type;
1442 lex_ctx_t ctx = parser_ctx(parser);
1443
1444 if (!parser->function->m_varargs) {
1445 parseerror(parser, "function has no variable argument list");
1446 return nullptr;
1447 }
1448
1449 if (!parser_next(parser) || parser->tok != '(') {
1450 parseerror(parser, "expected parameter index and type in parenthesis");
1451 return nullptr;
1452 }
1453 if (!parser_next(parser)) {
1454 parseerror(parser, "error parsing parameter index");
1455 return nullptr;
1456 }
1457
1458 idx = parse_expression_leave(parser, true, false, false);
1459 if (!idx)
1460 return nullptr;
1461
1462 if (parser->tok != ',') {
1463 if (parser->tok != ')') {
1464 ast_unref(idx);
1465 parseerror(parser, "expected comma after parameter index");
1466 return nullptr;
1467 }
1468 // vararg piping: ...(start)
1469 out = new ast_argpipe(ctx, idx);
1470 return out;
1471 }
1472
1473 if (!parser_next(parser) || (parser->tok != TOKEN_IDENT && parser->tok != TOKEN_TYPENAME)) {
1474 ast_unref(idx);
1475 parseerror(parser, "expected typename for vararg");
1476 return nullptr;
1477 }
1478
1479 typevar = parse_typename(parser, nullptr, nullptr, nullptr);
1480 if (!typevar) {
1481 ast_unref(idx);
1482 return nullptr;
1483 }
1484
1485 if (parser->tok != ')') {
1486 ast_unref(idx);
1487 delete typevar;
1488 parseerror(parser, "expected closing paren");
1489 return nullptr;
1490 }
1491
1492 if (funtype->m_varparam &&
1493 !typevar->compareType(*funtype->m_varparam))
1494 {
1495 char ty1[1024];
1496 char ty2[1024];
1497 ast_type_to_string(typevar, ty1, sizeof(ty1));
1498 ast_type_to_string(funtype->m_varparam, ty2, sizeof(ty2));
1499 compile_error(typevar->m_context,
1500 "function was declared to take varargs of type `%s`, requested type is: %s",
1501 ty2, ty1);
1502 }
1503
1504 out = ast_array_index::make(ctx, parser->function->m_varargs.get(), idx);
1505 out->adoptType(*typevar);
1506 delete typevar;
1507 return out;
1508 }
1509
parse_vararg(parser_t * parser)1510 static ast_expression* parse_vararg(parser_t *parser)
1511 {
1512 bool old_noops = parser->lex->flags.noops;
1513
1514 ast_expression *out;
1515
1516 parser->lex->flags.noops = true;
1517 out = parse_vararg_do(parser);
1518
1519 parser->lex->flags.noops = old_noops;
1520 return out;
1521 }
1522
1523 /* not to be exposed */
1524 bool ftepp_predef_exists(const char *name);
parse_sya_operand(parser_t * parser,shunt * sy,bool with_labels)1525 static bool parse_sya_operand(parser_t *parser, shunt *sy, bool with_labels)
1526 {
1527 if (OPTS_FLAG(TRANSLATABLE_STRINGS) &&
1528 parser->tok == TOKEN_IDENT &&
1529 !strcmp(parser_tokval(parser), "_"))
1530 {
1531 /* a translatable string */
1532 ast_value *val;
1533
1534 parser->lex->flags.noops = true;
1535 if (!parser_next(parser) || parser->tok != '(') {
1536 parseerror(parser, "use _(\"string\") to create a translatable string constant");
1537 return false;
1538 }
1539 parser->lex->flags.noops = false;
1540 if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
1541 parseerror(parser, "expected a constant string in translatable-string extension");
1542 return false;
1543 }
1544 val = (ast_value*)parser->m_fold.constgen_string(parser_tokval(parser), true);
1545 if (!val)
1546 return false;
1547 sy->out.push_back(syexp(parser_ctx(parser), val));
1548
1549 if (!parser_next(parser) || parser->tok != ')') {
1550 parseerror(parser, "expected closing paren after translatable string");
1551 return false;
1552 }
1553 return true;
1554 }
1555 else if (parser->tok == TOKEN_DOTS)
1556 {
1557 ast_expression *va;
1558 if (!OPTS_FLAG(VARIADIC_ARGS)) {
1559 parseerror(parser, "cannot access varargs (try -fvariadic-args)");
1560 return false;
1561 }
1562 va = parse_vararg(parser);
1563 if (!va)
1564 return false;
1565 sy->out.push_back(syexp(parser_ctx(parser), va));
1566 return true;
1567 }
1568 else if (parser->tok == TOKEN_FLOATCONST) {
1569 ast_expression *val = parser->m_fold.constgen_float((parser_token(parser)->constval.f), false);
1570 if (!val)
1571 return false;
1572 sy->out.push_back(syexp(parser_ctx(parser), val));
1573 return true;
1574 }
1575 else if (parser->tok == TOKEN_INTCONST || parser->tok == TOKEN_CHARCONST) {
1576 ast_expression *val = parser->m_fold.constgen_float((qcfloat_t)(parser_token(parser)->constval.i), false);
1577 if (!val)
1578 return false;
1579 sy->out.push_back(syexp(parser_ctx(parser), val));
1580 return true;
1581 }
1582 else if (parser->tok == TOKEN_STRINGCONST) {
1583 ast_expression *val = parser->m_fold.constgen_string(parser_tokval(parser), false);
1584 if (!val)
1585 return false;
1586 sy->out.push_back(syexp(parser_ctx(parser), val));
1587 return true;
1588 }
1589 else if (parser->tok == TOKEN_VECTORCONST) {
1590 ast_expression *val = parser->m_fold.constgen_vector(parser_token(parser)->constval.v);
1591 if (!val)
1592 return false;
1593 sy->out.push_back(syexp(parser_ctx(parser), val));
1594 return true;
1595 }
1596 else if (parser->tok == TOKEN_IDENT)
1597 {
1598 const char *ctoken = parser_tokval(parser);
1599 ast_expression *prev = sy->out.size() ? sy->out.back().out : nullptr;
1600 ast_expression *var;
1601 /* a_vector.{x,y,z} */
1602 if (sy->ops.empty() ||
1603 !sy->ops.back().etype ||
1604 operators[sy->ops.back().etype-1].id != opid1('.'))
1605 {
1606 /* When adding more intrinsics, fix the above condition */
1607 prev = nullptr;
1608 }
1609 if (prev && prev->m_vtype == TYPE_VECTOR && ctoken[0] >= 'x' && ctoken[0] <= 'z' && !ctoken[1])
1610 {
1611 var = parser->const_vec[ctoken[0]-'x'];
1612 } else {
1613 var = parser_find_var(parser, parser_tokval(parser));
1614 if (!var)
1615 var = parser_find_field(parser, parser_tokval(parser));
1616 }
1617 if (!var && with_labels) {
1618 var = parser_find_label(parser, parser_tokval(parser));
1619 if (!with_labels) {
1620 ast_label *lbl = new ast_label(parser_ctx(parser), parser_tokval(parser), true);
1621 var = lbl;
1622 parser->labels.push_back(lbl);
1623 }
1624 }
1625 if (!var && !strcmp(parser_tokval(parser), "__FUNC__"))
1626 var = parser->m_fold.constgen_string(parser->function->m_name, false);
1627 if (!var) {
1628 /*
1629 * now we try for the real intrinsic hashtable. If the string
1630 * begins with __builtin, we simply skip past it, otherwise we
1631 * use the identifier as is.
1632 */
1633 if (!strncmp(parser_tokval(parser), "__builtin_", 10)) {
1634 var = parser->m_intrin.func(parser_tokval(parser));
1635 }
1636
1637 /*
1638 * Try it again, intrin_func deals with the alias method as well
1639 * the first one masks for __builtin though, we emit warning here.
1640 */
1641 if (!var) {
1642 if ((var = parser->m_intrin.func(parser_tokval(parser)))) {
1643 (void)!compile_warning(
1644 parser_ctx(parser),
1645 WARN_BUILTINS,
1646 "using implicitly defined builtin `__builtin_%s' for `%s'",
1647 parser_tokval(parser),
1648 parser_tokval(parser)
1649 );
1650 }
1651 }
1652
1653
1654 if (!var) {
1655 /*
1656 * sometimes people use preprocessing predefs without enabling them
1657 * i've done this thousands of times already myself. Lets check for
1658 * it in the predef table. And diagnose it better :)
1659 */
1660 if (!OPTS_FLAG(FTEPP_PREDEFS) && ftepp_predef_exists(parser_tokval(parser))) {
1661 parseerror(parser, "unexpected identifier: %s (use -fftepp-predef to enable pre-defined macros)", parser_tokval(parser));
1662 return false;
1663 }
1664
1665 parseerror(parser, "unexpected identifier: %s", parser_tokval(parser));
1666 return false;
1667 }
1668 }
1669 else
1670 {
1671 if (ast_istype(var, ast_value)) {
1672 ((ast_value*)var)->m_uses++;
1673 }
1674 else if (ast_istype(var, ast_member)) {
1675 ast_member *mem = (ast_member*)var;
1676 if (ast_istype(mem->m_owner, ast_value))
1677 ((ast_value*)(mem->m_owner))->m_uses++;
1678 }
1679 }
1680 sy->out.push_back(syexp(parser_ctx(parser), var));
1681 return true;
1682 }
1683 parseerror(parser, "unexpected token `%s`", parser_tokval(parser));
1684 return false;
1685 }
1686
parse_expression_leave(parser_t * parser,bool stopatcomma,bool truthvalue,bool with_labels)1687 static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma, bool truthvalue, bool with_labels)
1688 {
1689 ast_expression *expr = nullptr;
1690 shunt sy;
1691 bool wantop = false;
1692 /* only warn once about an assignment in a truth value because the current code
1693 * would trigger twice on: if(a = b && ...), once for the if-truth-value, once for the && part
1694 */
1695 bool warn_parenthesis = true;
1696
1697 /* count the parens because an if starts with one, so the
1698 * end of a condition is an unmatched closing paren
1699 */
1700 int ternaries = 0;
1701
1702 memset(&sy, 0, sizeof(sy));
1703
1704 parser->lex->flags.noops = false;
1705
1706 parser_reclassify_token(parser);
1707
1708 while (true)
1709 {
1710 if (parser->tok == TOKEN_TYPENAME) {
1711 parseerror(parser, "unexpected typename `%s`", parser_tokval(parser));
1712 goto onerr;
1713 }
1714
1715 if (parser->tok == TOKEN_OPERATOR)
1716 {
1717 /* classify the operator */
1718 const oper_info *op;
1719 const oper_info *olast = nullptr;
1720 size_t o;
1721 for (o = 0; o < operator_count; ++o) {
1722 if (((!(operators[o].flags & OP_PREFIX) == !!wantop)) &&
1723 /* !(operators[o].flags & OP_SUFFIX) && / * remove this */
1724 !strcmp(parser_tokval(parser), operators[o].op))
1725 {
1726 break;
1727 }
1728 }
1729 if (o == operator_count) {
1730 compile_error(parser_ctx(parser), "unexpected operator: %s", parser_tokval(parser));
1731 goto onerr;
1732 }
1733 /* found an operator */
1734 op = &operators[o];
1735
1736 /* when declaring variables, a comma starts a new variable */
1737 if (op->id == opid1(',') && sy.paren.empty() && stopatcomma) {
1738 /* fixup the token */
1739 parser->tok = ',';
1740 break;
1741 }
1742
1743 /* a colon without a pervious question mark cannot be a ternary */
1744 if (!ternaries && op->id == opid2(':','?')) {
1745 parser->tok = ':';
1746 break;
1747 }
1748
1749 if (op->id == opid1(',')) {
1750 if (sy.paren.size() && sy.paren.back() == PAREN_TERNARY2) {
1751 (void)!parsewarning(parser, WARN_TERNARY_PRECEDENCE, "suggesting parenthesis around ternary expression");
1752 }
1753 }
1754
1755 if (sy.ops.size() && !sy.ops.back().isparen)
1756 olast = &operators[sy.ops.back().etype-1];
1757
1758 /* first only apply higher precedences, assoc_left+equal comes after we warn about precedence rules */
1759 while (olast && op->prec < olast->prec)
1760 {
1761 if (!parser_sy_apply_operator(parser, &sy))
1762 goto onerr;
1763 if (sy.ops.size() && !sy.ops.back().isparen)
1764 olast = &operators[sy.ops.back().etype-1];
1765 else
1766 olast = nullptr;
1767 }
1768
1769 #define IsAssignOp(x) (\
1770 (x) == opid1('=') || \
1771 (x) == opid2('+','=') || \
1772 (x) == opid2('-','=') || \
1773 (x) == opid2('*','=') || \
1774 (x) == opid2('/','=') || \
1775 (x) == opid2('%','=') || \
1776 (x) == opid2('&','=') || \
1777 (x) == opid2('|','=') || \
1778 (x) == opid3('&','~','=') \
1779 )
1780 if (warn_parenthesis) {
1781 if ( (olast && IsAssignOp(olast->id) && (op->id == opid2('&','&') || op->id == opid2('|','|'))) ||
1782 (olast && IsAssignOp(op->id) && (olast->id == opid2('&','&') || olast->id == opid2('|','|'))) ||
1783 (truthvalue && sy.paren.empty() && IsAssignOp(op->id))
1784 )
1785 {
1786 (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around assignment used as truth value");
1787 warn_parenthesis = false;
1788 }
1789
1790 if (olast && olast->id != op->id) {
1791 if ((op->id == opid1('&') || op->id == opid1('|') || op->id == opid1('^')) &&
1792 (olast->id == opid1('&') || olast->id == opid1('|') || olast->id == opid1('^')))
1793 {
1794 (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around bitwise operations");
1795 warn_parenthesis = false;
1796 }
1797 else if ((op->id == opid2('&','&') || op->id == opid2('|','|')) &&
1798 (olast->id == opid2('&','&') || olast->id == opid2('|','|')))
1799 {
1800 (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around logical operations");
1801 warn_parenthesis = false;
1802 }
1803 }
1804 }
1805
1806 while (olast && (
1807 (op->prec < olast->prec) ||
1808 (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) )
1809 {
1810 if (!parser_sy_apply_operator(parser, &sy))
1811 goto onerr;
1812 if (sy.ops.size() && !sy.ops.back().isparen)
1813 olast = &operators[sy.ops.back().etype-1];
1814 else
1815 olast = nullptr;
1816 }
1817
1818 if (op->id == opid1('(')) {
1819 if (wantop) {
1820 size_t sycount = sy.out.size();
1821 /* we expected an operator, this is the function-call operator */
1822 sy.paren.push_back(PAREN_FUNC);
1823 sy.ops.push_back(syparen(parser_ctx(parser), sycount-1));
1824 sy.argc.push_back(0);
1825 } else {
1826 sy.paren.push_back(PAREN_EXPR);
1827 sy.ops.push_back(syparen(parser_ctx(parser), 0));
1828 }
1829 wantop = false;
1830 } else if (op->id == opid1('[')) {
1831 if (!wantop) {
1832 parseerror(parser, "unexpected array subscript");
1833 goto onerr;
1834 }
1835 sy.paren.push_back(PAREN_INDEX);
1836 /* push both the operator and the paren, this makes life easier */
1837 sy.ops.push_back(syop(parser_ctx(parser), op));
1838 sy.ops.push_back(syparen(parser_ctx(parser), 0));
1839 wantop = false;
1840 } else if (op->id == opid2('?',':')) {
1841 sy.ops.push_back(syop(parser_ctx(parser), op));
1842 sy.ops.push_back(syparen(parser_ctx(parser), 0));
1843 wantop = false;
1844 ++ternaries;
1845 sy.paren.push_back(PAREN_TERNARY1);
1846 } else if (op->id == opid2(':','?')) {
1847 if (sy.paren.empty()) {
1848 parseerror(parser, "unexpected colon outside ternary expression (missing parenthesis?)");
1849 goto onerr;
1850 }
1851 if (sy.paren.back() != PAREN_TERNARY1) {
1852 parseerror(parser, "unexpected colon outside ternary expression (missing parenthesis?)");
1853 goto onerr;
1854 }
1855 if (!parser_close_paren(parser, &sy))
1856 goto onerr;
1857 sy.ops.push_back(syop(parser_ctx(parser), op));
1858 wantop = false;
1859 --ternaries;
1860 } else {
1861 sy.ops.push_back(syop(parser_ctx(parser), op));
1862 wantop = !!(op->flags & OP_SUFFIX);
1863 }
1864 }
1865 else if (parser->tok == ')') {
1866 while (sy.paren.size() && sy.paren.back() == PAREN_TERNARY2) {
1867 if (!parser_sy_apply_operator(parser, &sy))
1868 goto onerr;
1869 }
1870 if (sy.paren.empty())
1871 break;
1872 if (wantop) {
1873 if (sy.paren.back() == PAREN_TERNARY1) {
1874 parseerror(parser, "mismatched parentheses (closing paren in ternary expression?)");
1875 goto onerr;
1876 }
1877 if (!parser_close_paren(parser, &sy))
1878 goto onerr;
1879 } else {
1880 /* must be a function call without parameters */
1881 if (sy.paren.back() != PAREN_FUNC) {
1882 parseerror(parser, "closing paren in invalid position");
1883 goto onerr;
1884 }
1885 if (!parser_close_paren(parser, &sy))
1886 goto onerr;
1887 }
1888 wantop = true;
1889 }
1890 else if (parser->tok == '(') {
1891 parseerror(parser, "internal error: '(' should be classified as operator");
1892 goto onerr;
1893 }
1894 else if (parser->tok == '[') {
1895 parseerror(parser, "internal error: '[' should be classified as operator");
1896 goto onerr;
1897 }
1898 else if (parser->tok == ']') {
1899 while (sy.paren.size() && sy.paren.back() == PAREN_TERNARY2) {
1900 if (!parser_sy_apply_operator(parser, &sy))
1901 goto onerr;
1902 }
1903 if (sy.paren.empty())
1904 break;
1905 if (sy.paren.back() != PAREN_INDEX) {
1906 parseerror(parser, "mismatched parentheses, unexpected ']'");
1907 goto onerr;
1908 }
1909 if (!parser_close_paren(parser, &sy))
1910 goto onerr;
1911 wantop = true;
1912 }
1913 else if (!wantop) {
1914 if (!parse_sya_operand(parser, &sy, with_labels))
1915 goto onerr;
1916 wantop = true;
1917 }
1918 else {
1919 /* in this case we might want to allow constant string concatenation */
1920 bool concatenated = false;
1921 if (parser->tok == TOKEN_STRINGCONST && sy.out.size()) {
1922 ast_expression *lexpr = sy.out.back().out;
1923 if (ast_istype(lexpr, ast_value)) {
1924 ast_value *last = (ast_value*)lexpr;
1925 if (last->m_isimm == true && last->m_cvq == CV_CONST &&
1926 last->m_hasvalue && last->m_vtype == TYPE_STRING)
1927 {
1928 char *newstr = nullptr;
1929 util_asprintf(&newstr, "%s%s", last->m_constval.vstring, parser_tokval(parser));
1930 sy.out.back().out = parser->m_fold.constgen_string(newstr, false);
1931 mem_d(newstr);
1932 concatenated = true;
1933 }
1934 }
1935 }
1936 if (!concatenated) {
1937 parseerror(parser, "expected operator or end of statement");
1938 goto onerr;
1939 }
1940 }
1941
1942 if (!parser_next(parser)) {
1943 goto onerr;
1944 }
1945 if (parser->tok == ';' ||
1946 ((sy.paren.empty() || (sy.paren.size() == 1 && sy.paren.back() == PAREN_TERNARY2)) &&
1947 (parser->tok == ']' || parser->tok == ')' || parser->tok == '}')))
1948 {
1949 break;
1950 }
1951 }
1952
1953 while (sy.ops.size()) {
1954 if (!parser_sy_apply_operator(parser, &sy))
1955 goto onerr;
1956 }
1957
1958 parser->lex->flags.noops = true;
1959 if (sy.out.size() != 1) {
1960 parseerror(parser, "expression expected");
1961 expr = nullptr;
1962 } else
1963 expr = sy.out[0].out;
1964 if (sy.paren.size()) {
1965 parseerror(parser, "internal error: sy.paren.size() = %zu", sy.paren.size());
1966 return nullptr;
1967 }
1968 return expr;
1969
1970 onerr:
1971 parser->lex->flags.noops = true;
1972 for (auto &it : sy.out)
1973 if (it.out) ast_unref(it.out);
1974 return nullptr;
1975 }
1976
parse_expression(parser_t * parser,bool stopatcomma,bool with_labels)1977 static ast_expression* parse_expression(parser_t *parser, bool stopatcomma, bool with_labels)
1978 {
1979 ast_expression *e = parse_expression_leave(parser, stopatcomma, false, with_labels);
1980 if (!e)
1981 return nullptr;
1982 if (parser->tok != ';') {
1983 parseerror(parser, "semicolon expected after expression");
1984 ast_unref(e);
1985 return nullptr;
1986 }
1987 if (!parser_next(parser)) {
1988 ast_unref(e);
1989 return nullptr;
1990 }
1991 return e;
1992 }
1993
parser_enterblock(parser_t * parser)1994 static void parser_enterblock(parser_t *parser)
1995 {
1996 vec_push(parser->variables, util_htnew(PARSER_HT_SIZE));
1997 vec_push(parser->_blocklocals, vec_size(parser->_locals));
1998 vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE));
1999 vec_push(parser->_blocktypedefs, vec_size(parser->_typedefs));
2000 vec_push(parser->_block_ctx, parser_ctx(parser));
2001 }
2002
parser_leaveblock(parser_t * parser)2003 static bool parser_leaveblock(parser_t *parser)
2004 {
2005 bool rv = true;
2006 size_t locals, typedefs;
2007
2008 if (vec_size(parser->variables) <= PARSER_HT_LOCALS) {
2009 parseerror(parser, "internal error: parser_leaveblock with no block");
2010 return false;
2011 }
2012
2013 util_htdel(vec_last(parser->variables));
2014
2015 vec_pop(parser->variables);
2016 if (!vec_size(parser->_blocklocals)) {
2017 parseerror(parser, "internal error: parser_leaveblock with no block (2)");
2018 return false;
2019 }
2020
2021 locals = vec_last(parser->_blocklocals);
2022 vec_pop(parser->_blocklocals);
2023 while (vec_size(parser->_locals) != locals) {
2024 ast_expression *e = vec_last(parser->_locals);
2025 ast_value *v = (ast_value*)e;
2026 vec_pop(parser->_locals);
2027 if (ast_istype(e, ast_value) && !v->m_uses) {
2028 if (compile_warning(v->m_context, WARN_UNUSED_VARIABLE, "unused variable: `%s`", v->m_name))
2029 rv = false;
2030 }
2031 }
2032
2033 typedefs = vec_last(parser->_blocktypedefs);
2034 while (vec_size(parser->_typedefs) != typedefs) {
2035 delete vec_last(parser->_typedefs);
2036 vec_pop(parser->_typedefs);
2037 }
2038 util_htdel(vec_last(parser->typedefs));
2039 vec_pop(parser->typedefs);
2040
2041 vec_pop(parser->_block_ctx);
2042
2043 return rv;
2044 }
2045
parser_addlocal(parser_t * parser,const char * name,ast_expression * e)2046 static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e)
2047 {
2048 vec_push(parser->_locals, e);
2049 util_htset(vec_last(parser->variables), name, (void*)e);
2050 }
parser_addlocal(parser_t * parser,const std::string & name,ast_expression * e)2051 static void parser_addlocal(parser_t *parser, const std::string &name, ast_expression *e) {
2052 return parser_addlocal(parser, name.c_str(), e);
2053 }
2054
parser_addglobal(parser_t * parser,const char * name,ast_expression * e)2055 static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e)
2056 {
2057 parser->globals.push_back(e);
2058 util_htset(parser->htglobals, name, e);
2059 }
parser_addglobal(parser_t * parser,const std::string & name,ast_expression * e)2060 static void parser_addglobal(parser_t *parser, const std::string &name, ast_expression *e) {
2061 return parser_addglobal(parser, name.c_str(), e);
2062 }
2063
process_condition(parser_t * parser,ast_expression * cond,bool * _ifnot)2064 static ast_expression* process_condition(parser_t *parser, ast_expression *cond, bool *_ifnot)
2065 {
2066 bool ifnot = false;
2067 ast_unary *unary;
2068 ast_expression *prev;
2069
2070 if (cond->m_vtype == TYPE_VOID || cond->m_vtype >= TYPE_VARIANT) {
2071 char ty[1024];
2072 ast_type_to_string(cond, ty, sizeof(ty));
2073 compile_error(cond->m_context, "invalid type for if() condition: %s", ty);
2074 }
2075
2076 if (OPTS_FLAG(FALSE_EMPTY_STRINGS) && cond->m_vtype == TYPE_STRING)
2077 {
2078 prev = cond;
2079 cond = ast_unary::make(cond->m_context, INSTR_NOT_S, cond);
2080 if (!cond) {
2081 ast_unref(prev);
2082 parseerror(parser, "internal error: failed to process condition");
2083 return nullptr;
2084 }
2085 ifnot = !ifnot;
2086 }
2087 else if (OPTS_FLAG(CORRECT_LOGIC) && cond->m_vtype == TYPE_VECTOR)
2088 {
2089 /* vector types need to be cast to true booleans */
2090 ast_binary *bin = (ast_binary*)cond;
2091 if (!OPTS_FLAG(PERL_LOGIC) || !ast_istype(cond, ast_binary) || !(bin->m_op == INSTR_AND || bin->m_op == INSTR_OR))
2092 {
2093 /* in perl-logic, AND and OR take care of the -fcorrect-logic */
2094 prev = cond;
2095 cond = ast_unary::make(cond->m_context, INSTR_NOT_V, cond);
2096 if (!cond) {
2097 ast_unref(prev);
2098 parseerror(parser, "internal error: failed to process condition");
2099 return nullptr;
2100 }
2101 ifnot = !ifnot;
2102 }
2103 }
2104
2105 unary = (ast_unary*)cond;
2106 /* ast_istype dereferences cond, should test here for safety */
2107 while (cond && ast_istype(cond, ast_unary) && unary->m_op == INSTR_NOT_F)
2108 {
2109 cond = unary->m_operand;
2110 unary->m_operand = nullptr;
2111 delete unary;
2112 ifnot = !ifnot;
2113 unary = (ast_unary*)cond;
2114 }
2115
2116 if (!cond)
2117 parseerror(parser, "internal error: failed to process condition");
2118
2119 if (ifnot) *_ifnot = !*_ifnot;
2120 return cond;
2121 }
2122
parse_if(parser_t * parser,ast_block * block,ast_expression ** out)2123 static bool parse_if(parser_t *parser, ast_block *block, ast_expression **out)
2124 {
2125 ast_ifthen *ifthen;
2126 ast_expression *cond, *ontrue = nullptr, *onfalse = nullptr;
2127 bool ifnot = false;
2128
2129 lex_ctx_t ctx = parser_ctx(parser);
2130
2131 (void)block; /* not touching */
2132
2133 /* skip the 'if', parse an optional 'not' and check for an opening paren */
2134 if (!parser_next(parser)) {
2135 parseerror(parser, "expected condition or 'not'");
2136 return false;
2137 }
2138 if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "not")) {
2139 ifnot = true;
2140 if (!parser_next(parser)) {
2141 parseerror(parser, "expected condition in parenthesis");
2142 return false;
2143 }
2144 }
2145 if (parser->tok != '(') {
2146 parseerror(parser, "expected 'if' condition in parenthesis");
2147 return false;
2148 }
2149 /* parse into the expression */
2150 if (!parser_next(parser)) {
2151 parseerror(parser, "expected 'if' condition after opening paren");
2152 return false;
2153 }
2154 /* parse the condition */
2155 cond = parse_expression_leave(parser, false, true, false);
2156 if (!cond)
2157 return false;
2158 /* closing paren */
2159 if (parser->tok != ')') {
2160 parseerror(parser, "expected closing paren after 'if' condition");
2161 ast_unref(cond);
2162 return false;
2163 }
2164 /* parse into the 'then' branch */
2165 if (!parser_next(parser)) {
2166 parseerror(parser, "expected statement for on-true branch of 'if'");
2167 ast_unref(cond);
2168 return false;
2169 }
2170 if (!parse_statement_or_block(parser, &ontrue)) {
2171 ast_unref(cond);
2172 return false;
2173 }
2174 if (!ontrue)
2175 ontrue = new ast_block(parser_ctx(parser));
2176 /* check for an else */
2177 if (!strcmp(parser_tokval(parser), "else")) {
2178 /* parse into the 'else' branch */
2179 if (!parser_next(parser)) {
2180 parseerror(parser, "expected on-false branch after 'else'");
2181 delete ontrue;
2182 ast_unref(cond);
2183 return false;
2184 }
2185 if (!parse_statement_or_block(parser, &onfalse)) {
2186 delete ontrue;
2187 ast_unref(cond);
2188 return false;
2189 }
2190 }
2191
2192 cond = process_condition(parser, cond, &ifnot);
2193 if (!cond) {
2194 if (ontrue) delete ontrue;
2195 if (onfalse) delete onfalse;
2196 return false;
2197 }
2198
2199 if (ifnot)
2200 ifthen = new ast_ifthen(ctx, cond, onfalse, ontrue);
2201 else
2202 ifthen = new ast_ifthen(ctx, cond, ontrue, onfalse);
2203 *out = ifthen;
2204 return true;
2205 }
2206
2207 static bool parse_while_go(parser_t *parser, ast_block *block, ast_expression **out);
parse_while(parser_t * parser,ast_block * block,ast_expression ** out)2208 static bool parse_while(parser_t *parser, ast_block *block, ast_expression **out)
2209 {
2210 bool rv;
2211 char *label = nullptr;
2212
2213 /* skip the 'while' and get the body */
2214 if (!parser_next(parser)) {
2215 if (OPTS_FLAG(LOOP_LABELS))
2216 parseerror(parser, "expected loop label or 'while' condition in parenthesis");
2217 else
2218 parseerror(parser, "expected 'while' condition in parenthesis");
2219 return false;
2220 }
2221
2222 if (parser->tok == ':') {
2223 if (!OPTS_FLAG(LOOP_LABELS))
2224 parseerror(parser, "labeled loops not activated, try using -floop-labels");
2225 if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
2226 parseerror(parser, "expected loop label");
2227 return false;
2228 }
2229 label = util_strdup(parser_tokval(parser));
2230 if (!parser_next(parser)) {
2231 mem_d(label);
2232 parseerror(parser, "expected 'while' condition in parenthesis");
2233 return false;
2234 }
2235 }
2236
2237 if (parser->tok != '(') {
2238 parseerror(parser, "expected 'while' condition in parenthesis");
2239 return false;
2240 }
2241
2242 parser->breaks.push_back(label);
2243 parser->continues.push_back(label);
2244
2245 rv = parse_while_go(parser, block, out);
2246 if (label)
2247 mem_d(label);
2248 if (parser->breaks.back() != label || parser->continues.back() != label) {
2249 parseerror(parser, "internal error: label stack corrupted");
2250 rv = false;
2251 delete *out;
2252 *out = nullptr;
2253 }
2254 else {
2255 parser->breaks.pop_back();
2256 parser->continues.pop_back();
2257 }
2258 return rv;
2259 }
2260
parse_while_go(parser_t * parser,ast_block * block,ast_expression ** out)2261 static bool parse_while_go(parser_t *parser, ast_block *block, ast_expression **out)
2262 {
2263 ast_loop *aloop;
2264 ast_expression *cond, *ontrue;
2265
2266 bool ifnot = false;
2267
2268 lex_ctx_t ctx = parser_ctx(parser);
2269
2270 (void)block; /* not touching */
2271
2272 /* parse into the expression */
2273 if (!parser_next(parser)) {
2274 parseerror(parser, "expected 'while' condition after opening paren");
2275 return false;
2276 }
2277 /* parse the condition */
2278 cond = parse_expression_leave(parser, false, true, false);
2279 if (!cond)
2280 return false;
2281 /* closing paren */
2282 if (parser->tok != ')') {
2283 parseerror(parser, "expected closing paren after 'while' condition");
2284 ast_unref(cond);
2285 return false;
2286 }
2287 /* parse into the 'then' branch */
2288 if (!parser_next(parser)) {
2289 parseerror(parser, "expected while-loop body");
2290 ast_unref(cond);
2291 return false;
2292 }
2293 if (!parse_statement_or_block(parser, &ontrue)) {
2294 ast_unref(cond);
2295 return false;
2296 }
2297
2298 cond = process_condition(parser, cond, &ifnot);
2299 if (!cond) {
2300 ast_unref(ontrue);
2301 return false;
2302 }
2303 aloop = new ast_loop(ctx, nullptr, cond, ifnot, nullptr, false, nullptr, ontrue);
2304 *out = aloop;
2305 return true;
2306 }
2307
2308 static bool parse_dowhile_go(parser_t *parser, ast_block *block, ast_expression **out);
parse_dowhile(parser_t * parser,ast_block * block,ast_expression ** out)2309 static bool parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out)
2310 {
2311 bool rv;
2312 char *label = nullptr;
2313
2314 /* skip the 'do' and get the body */
2315 if (!parser_next(parser)) {
2316 if (OPTS_FLAG(LOOP_LABELS))
2317 parseerror(parser, "expected loop label or body");
2318 else
2319 parseerror(parser, "expected loop body");
2320 return false;
2321 }
2322
2323 if (parser->tok == ':') {
2324 if (!OPTS_FLAG(LOOP_LABELS))
2325 parseerror(parser, "labeled loops not activated, try using -floop-labels");
2326 if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
2327 parseerror(parser, "expected loop label");
2328 return false;
2329 }
2330 label = util_strdup(parser_tokval(parser));
2331 if (!parser_next(parser)) {
2332 mem_d(label);
2333 parseerror(parser, "expected loop body");
2334 return false;
2335 }
2336 }
2337
2338 parser->breaks.push_back(label);
2339 parser->continues.push_back(label);
2340
2341 rv = parse_dowhile_go(parser, block, out);
2342 if (label)
2343 mem_d(label);
2344 if (parser->breaks.back() != label || parser->continues.back() != label) {
2345 parseerror(parser, "internal error: label stack corrupted");
2346 rv = false;
2347 delete *out;
2348 *out = nullptr;
2349 }
2350 else {
2351 parser->breaks.pop_back();
2352 parser->continues.pop_back();
2353 }
2354 return rv;
2355 }
2356
parse_dowhile_go(parser_t * parser,ast_block * block,ast_expression ** out)2357 static bool parse_dowhile_go(parser_t *parser, ast_block *block, ast_expression **out)
2358 {
2359 ast_loop *aloop;
2360 ast_expression *cond, *ontrue;
2361
2362 bool ifnot = false;
2363
2364 lex_ctx_t ctx = parser_ctx(parser);
2365
2366 (void)block; /* not touching */
2367
2368 if (!parse_statement_or_block(parser, &ontrue))
2369 return false;
2370
2371 /* expect the "while" */
2372 if (parser->tok != TOKEN_KEYWORD ||
2373 strcmp(parser_tokval(parser), "while"))
2374 {
2375 parseerror(parser, "expected 'while' and condition");
2376 delete ontrue;
2377 return false;
2378 }
2379
2380 /* skip the 'while' and check for opening paren */
2381 if (!parser_next(parser) || parser->tok != '(') {
2382 parseerror(parser, "expected 'while' condition in parenthesis");
2383 delete ontrue;
2384 return false;
2385 }
2386 /* parse into the expression */
2387 if (!parser_next(parser)) {
2388 parseerror(parser, "expected 'while' condition after opening paren");
2389 delete ontrue;
2390 return false;
2391 }
2392 /* parse the condition */
2393 cond = parse_expression_leave(parser, false, true, false);
2394 if (!cond)
2395 return false;
2396 /* closing paren */
2397 if (parser->tok != ')') {
2398 parseerror(parser, "expected closing paren after 'while' condition");
2399 delete ontrue;
2400 ast_unref(cond);
2401 return false;
2402 }
2403 /* parse on */
2404 if (!parser_next(parser) || parser->tok != ';') {
2405 parseerror(parser, "expected semicolon after condition");
2406 delete ontrue;
2407 ast_unref(cond);
2408 return false;
2409 }
2410
2411 if (!parser_next(parser)) {
2412 parseerror(parser, "parse error");
2413 delete ontrue;
2414 ast_unref(cond);
2415 return false;
2416 }
2417
2418 cond = process_condition(parser, cond, &ifnot);
2419 if (!cond) {
2420 delete ontrue;
2421 return false;
2422 }
2423 aloop = new ast_loop(ctx, nullptr, nullptr, false, cond, ifnot, nullptr, ontrue);
2424 *out = aloop;
2425 return true;
2426 }
2427
2428 static bool parse_for_go(parser_t *parser, ast_block *block, ast_expression **out);
parse_for(parser_t * parser,ast_block * block,ast_expression ** out)2429 static bool parse_for(parser_t *parser, ast_block *block, ast_expression **out)
2430 {
2431 bool rv;
2432 char *label = nullptr;
2433
2434 /* skip the 'for' and check for opening paren */
2435 if (!parser_next(parser)) {
2436 if (OPTS_FLAG(LOOP_LABELS))
2437 parseerror(parser, "expected loop label or 'for' expressions in parenthesis");
2438 else
2439 parseerror(parser, "expected 'for' expressions in parenthesis");
2440 return false;
2441 }
2442
2443 if (parser->tok == ':') {
2444 if (!OPTS_FLAG(LOOP_LABELS))
2445 parseerror(parser, "labeled loops not activated, try using -floop-labels");
2446 if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
2447 parseerror(parser, "expected loop label");
2448 return false;
2449 }
2450 label = util_strdup(parser_tokval(parser));
2451 if (!parser_next(parser)) {
2452 mem_d(label);
2453 parseerror(parser, "expected 'for' expressions in parenthesis");
2454 return false;
2455 }
2456 }
2457
2458 if (parser->tok != '(') {
2459 parseerror(parser, "expected 'for' expressions in parenthesis");
2460 return false;
2461 }
2462
2463 parser->breaks.push_back(label);
2464 parser->continues.push_back(label);
2465
2466 rv = parse_for_go(parser, block, out);
2467 if (label)
2468 mem_d(label);
2469 if (parser->breaks.back() != label || parser->continues.back() != label) {
2470 parseerror(parser, "internal error: label stack corrupted");
2471 rv = false;
2472 delete *out;
2473 *out = nullptr;
2474 }
2475 else {
2476 parser->breaks.pop_back();
2477 parser->continues.pop_back();
2478 }
2479 return rv;
2480 }
parse_for_go(parser_t * parser,ast_block * block,ast_expression ** out)2481 static bool parse_for_go(parser_t *parser, ast_block *block, ast_expression **out)
2482 {
2483 ast_loop *aloop;
2484 ast_expression *initexpr, *cond, *increment, *ontrue;
2485 ast_value *typevar;
2486
2487 bool ifnot = false;
2488
2489 lex_ctx_t ctx = parser_ctx(parser);
2490
2491 parser_enterblock(parser);
2492
2493 initexpr = nullptr;
2494 cond = nullptr;
2495 increment = nullptr;
2496 ontrue = nullptr;
2497
2498 /* parse into the expression */
2499 if (!parser_next(parser)) {
2500 parseerror(parser, "expected 'for' initializer after opening paren");
2501 goto onerr;
2502 }
2503
2504 typevar = nullptr;
2505 if (parser->tok == TOKEN_IDENT)
2506 typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
2507
2508 if (typevar || parser->tok == TOKEN_TYPENAME) {
2509 if (!parse_variable(parser, block, true, CV_VAR, typevar, false, false, 0, nullptr))
2510 goto onerr;
2511 }
2512 else if (parser->tok != ';')
2513 {
2514 initexpr = parse_expression_leave(parser, false, false, false);
2515 if (!initexpr)
2516 goto onerr;
2517 /* move on to condition */
2518 if (parser->tok != ';') {
2519 parseerror(parser, "expected semicolon after for-loop initializer");
2520 goto onerr;
2521 }
2522 if (!parser_next(parser)) {
2523 parseerror(parser, "expected for-loop condition");
2524 goto onerr;
2525 }
2526 } else if (!parser_next(parser)) {
2527 parseerror(parser, "expected for-loop condition");
2528 goto onerr;
2529 }
2530
2531 /* parse the condition */
2532 if (parser->tok != ';') {
2533 cond = parse_expression_leave(parser, false, true, false);
2534 if (!cond)
2535 goto onerr;
2536 }
2537 /* move on to incrementor */
2538 if (parser->tok != ';') {
2539 parseerror(parser, "expected semicolon after for-loop initializer");
2540 goto onerr;
2541 }
2542 if (!parser_next(parser)) {
2543 parseerror(parser, "expected for-loop condition");
2544 goto onerr;
2545 }
2546
2547 /* parse the incrementor */
2548 if (parser->tok != ')') {
2549 lex_ctx_t condctx = parser_ctx(parser);
2550 increment = parse_expression_leave(parser, false, false, false);
2551 if (!increment)
2552 goto onerr;
2553 if (!increment->m_side_effects) {
2554 if (compile_warning(condctx, WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
2555 goto onerr;
2556 }
2557 }
2558
2559 /* closing paren */
2560 if (parser->tok != ')') {
2561 parseerror(parser, "expected closing paren after 'for-loop' incrementor");
2562 goto onerr;
2563 }
2564 /* parse into the 'then' branch */
2565 if (!parser_next(parser)) {
2566 parseerror(parser, "expected for-loop body");
2567 goto onerr;
2568 }
2569 if (!parse_statement_or_block(parser, &ontrue))
2570 goto onerr;
2571
2572 if (cond) {
2573 cond = process_condition(parser, cond, &ifnot);
2574 if (!cond)
2575 goto onerr;
2576 }
2577 aloop = new ast_loop(ctx, initexpr, cond, ifnot, nullptr, false, increment, ontrue);
2578 *out = aloop;
2579
2580 if (!parser_leaveblock(parser)) {
2581 delete aloop;
2582 return false;
2583 }
2584 return true;
2585 onerr:
2586 if (initexpr) ast_unref(initexpr);
2587 if (cond) ast_unref(cond);
2588 if (increment) ast_unref(increment);
2589 (void)!parser_leaveblock(parser);
2590 return false;
2591 }
2592
parse_return(parser_t * parser,ast_block * block,ast_expression ** out)2593 static bool parse_return(parser_t *parser, ast_block *block, ast_expression **out)
2594 {
2595 ast_expression *exp = nullptr;
2596 ast_expression *var = nullptr;
2597 ast_return *ret = nullptr;
2598 ast_value *retval = parser->function->m_return_value;
2599 ast_value *expected = parser->function->m_function_type;
2600
2601 lex_ctx_t ctx = parser_ctx(parser);
2602
2603 (void)block; /* not touching */
2604
2605 if (!parser_next(parser)) {
2606 parseerror(parser, "expected return expression");
2607 return false;
2608 }
2609
2610 /* return assignments */
2611 if (parser->tok == '=') {
2612 if (!OPTS_FLAG(RETURN_ASSIGNMENTS)) {
2613 parseerror(parser, "return assignments not activated, try using -freturn-assigments");
2614 return false;
2615 }
2616
2617 if (type_store_instr[expected->m_next->m_vtype] == VINSTR_END) {
2618 char ty1[1024];
2619 ast_type_to_string(expected->m_next, ty1, sizeof(ty1));
2620 parseerror(parser, "invalid return type: `%s'", ty1);
2621 return false;
2622 }
2623
2624 if (!parser_next(parser)) {
2625 parseerror(parser, "expected return assignment expression");
2626 return false;
2627 }
2628
2629 if (!(exp = parse_expression_leave(parser, false, false, false)))
2630 return false;
2631
2632 /* prepare the return value */
2633 if (!retval) {
2634 retval = new ast_value(ctx, "#LOCAL_RETURN", TYPE_VOID);
2635 retval->adoptType(*expected->m_next);
2636 parser->function->m_return_value = retval;
2637 }
2638
2639 if (!exp->compareType(*retval)) {
2640 char ty1[1024], ty2[1024];
2641 ast_type_to_string(exp, ty1, sizeof(ty1));
2642 ast_type_to_string(retval, ty2, sizeof(ty2));
2643 parseerror(parser, "invalid type for return value: `%s', expected `%s'", ty1, ty2);
2644 }
2645
2646 /* store to 'return' local variable */
2647 var = new ast_store(
2648 ctx,
2649 type_store_instr[expected->m_next->m_vtype],
2650 retval, exp);
2651
2652 if (!var) {
2653 ast_unref(exp);
2654 return false;
2655 }
2656
2657 if (parser->tok != ';')
2658 parseerror(parser, "missing semicolon after return assignment");
2659 else if (!parser_next(parser))
2660 parseerror(parser, "parse error after return assignment");
2661
2662 *out = var;
2663 return true;
2664 }
2665
2666 if (parser->tok != ';') {
2667 exp = parse_expression(parser, false, false);
2668 if (!exp)
2669 return false;
2670
2671 if (exp->m_vtype != TYPE_NIL &&
2672 exp->m_vtype != (expected)->m_next->m_vtype)
2673 {
2674 parseerror(parser, "return with invalid expression");
2675 }
2676
2677 ret = new ast_return(ctx, exp);
2678 if (!ret) {
2679 ast_unref(exp);
2680 return false;
2681 }
2682 } else {
2683 if (!parser_next(parser))
2684 parseerror(parser, "parse error");
2685
2686 if (!retval && expected->m_next->m_vtype != TYPE_VOID)
2687 {
2688 (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
2689 }
2690 ret = new ast_return(ctx, retval);
2691 }
2692 *out = ret;
2693 return true;
2694 }
2695
parse_break_continue(parser_t * parser,ast_block * block,ast_expression ** out,bool is_continue)2696 static bool parse_break_continue(parser_t *parser, ast_block *block, ast_expression **out, bool is_continue)
2697 {
2698 size_t i;
2699 unsigned int levels = 0;
2700 lex_ctx_t ctx = parser_ctx(parser);
2701 auto &loops = (is_continue ? parser->continues : parser->breaks);
2702
2703 (void)block; /* not touching */
2704 if (!parser_next(parser)) {
2705 parseerror(parser, "expected semicolon or loop label");
2706 return false;
2707 }
2708
2709 if (loops.empty()) {
2710 if (is_continue)
2711 parseerror(parser, "`continue` can only be used inside loops");
2712 else
2713 parseerror(parser, "`break` can only be used inside loops or switches");
2714 }
2715
2716 if (parser->tok == TOKEN_IDENT) {
2717 if (!OPTS_FLAG(LOOP_LABELS))
2718 parseerror(parser, "labeled loops not activated, try using -floop-labels");
2719 i = loops.size();
2720 while (i--) {
2721 if (loops[i] && !strcmp(loops[i], parser_tokval(parser)))
2722 break;
2723 if (!i) {
2724 parseerror(parser, "no such loop to %s: `%s`",
2725 (is_continue ? "continue" : "break out of"),
2726 parser_tokval(parser));
2727 return false;
2728 }
2729 ++levels;
2730 }
2731 if (!parser_next(parser)) {
2732 parseerror(parser, "expected semicolon");
2733 return false;
2734 }
2735 }
2736
2737 if (parser->tok != ';') {
2738 parseerror(parser, "expected semicolon");
2739 return false;
2740 }
2741
2742 if (!parser_next(parser))
2743 parseerror(parser, "parse error");
2744
2745 *out = new ast_breakcont(ctx, is_continue, levels);
2746 return true;
2747 }
2748
2749 /* returns true when it was a variable qualifier, false otherwise!
2750 * on error, cvq is set to CV_WRONG
2751 */
2752 struct attribute_t {
2753 const char *name;
2754 size_t flag;
2755 };
2756
parse_qualifiers(parser_t * parser,bool with_local,int * cvq,bool * noref,bool * is_static,uint32_t * _flags,char ** message)2757 static bool parse_qualifiers(parser_t *parser, bool with_local, int *cvq, bool *noref, bool *is_static, uint32_t *_flags, char **message)
2758 {
2759 bool had_const = false;
2760 bool had_var = false;
2761 bool had_noref = false;
2762 bool had_attrib = false;
2763 bool had_static = false;
2764 uint32_t flags = 0;
2765
2766 static attribute_t attributes[] = {
2767 { "noreturn", AST_FLAG_NORETURN },
2768 { "inline", AST_FLAG_INLINE },
2769 { "eraseable", AST_FLAG_ERASEABLE },
2770 { "accumulate", AST_FLAG_ACCUMULATE },
2771 { "last", AST_FLAG_FINAL_DECL }
2772 };
2773
2774 *cvq = CV_NONE;
2775
2776 for (;;) {
2777 size_t i;
2778 if (parser->tok == TOKEN_ATTRIBUTE_OPEN) {
2779 had_attrib = true;
2780 /* parse an attribute */
2781 if (!parser_next(parser)) {
2782 parseerror(parser, "expected attribute after `[[`");
2783 *cvq = CV_WRONG;
2784 return false;
2785 }
2786
2787 for (i = 0; i < GMQCC_ARRAY_COUNT(attributes); i++) {
2788 if (!strcmp(parser_tokval(parser), attributes[i].name)) {
2789 flags |= attributes[i].flag;
2790 if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2791 parseerror(parser, "`%s` attribute has no parameters, expected `]]`",
2792 attributes[i].name);
2793 *cvq = CV_WRONG;
2794 return false;
2795 }
2796 break;
2797 }
2798 }
2799
2800 if (i != GMQCC_ARRAY_COUNT(attributes))
2801 goto leave;
2802
2803
2804 if (!strcmp(parser_tokval(parser), "noref")) {
2805 had_noref = true;
2806 if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2807 parseerror(parser, "`noref` attribute has no parameters, expected `]]`");
2808 *cvq = CV_WRONG;
2809 return false;
2810 }
2811 }
2812 else if (!strcmp(parser_tokval(parser), "alias") && !(flags & AST_FLAG_ALIAS)) {
2813 flags |= AST_FLAG_ALIAS;
2814 *message = nullptr;
2815
2816 if (!parser_next(parser)) {
2817 parseerror(parser, "parse error in attribute");
2818 goto argerr;
2819 }
2820
2821 if (parser->tok == '(') {
2822 if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
2823 parseerror(parser, "`alias` attribute missing parameter");
2824 goto argerr;
2825 }
2826
2827 *message = util_strdup(parser_tokval(parser));
2828
2829 if (!parser_next(parser)) {
2830 parseerror(parser, "parse error in attribute");
2831 goto argerr;
2832 }
2833
2834 if (parser->tok != ')') {
2835 parseerror(parser, "`alias` attribute expected `)` after parameter");
2836 goto argerr;
2837 }
2838
2839 if (!parser_next(parser)) {
2840 parseerror(parser, "parse error in attribute");
2841 goto argerr;
2842 }
2843 }
2844
2845 if (parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2846 parseerror(parser, "`alias` attribute expected `]]`");
2847 goto argerr;
2848 }
2849 }
2850 else if (!strcmp(parser_tokval(parser), "deprecated") && !(flags & AST_FLAG_DEPRECATED)) {
2851 flags |= AST_FLAG_DEPRECATED;
2852 *message = nullptr;
2853
2854 if (!parser_next(parser)) {
2855 parseerror(parser, "parse error in attribute");
2856 goto argerr;
2857 }
2858
2859 if (parser->tok == '(') {
2860 if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) {
2861 parseerror(parser, "`deprecated` attribute missing parameter");
2862 goto argerr;
2863 }
2864
2865 *message = util_strdup(parser_tokval(parser));
2866
2867 if (!parser_next(parser)) {
2868 parseerror(parser, "parse error in attribute");
2869 goto argerr;
2870 }
2871
2872 if(parser->tok != ')') {
2873 parseerror(parser, "`deprecated` attribute expected `)` after parameter");
2874 goto argerr;
2875 }
2876
2877 if (!parser_next(parser)) {
2878 parseerror(parser, "parse error in attribute");
2879 goto argerr;
2880 }
2881 }
2882 /* no message */
2883 if (parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2884 parseerror(parser, "`deprecated` attribute expected `]]`");
2885
2886 argerr: /* ugly */
2887 if (*message) mem_d(*message);
2888 *message = nullptr;
2889 *cvq = CV_WRONG;
2890 return false;
2891 }
2892 }
2893 else if (!strcmp(parser_tokval(parser), "coverage") && !(flags & AST_FLAG_COVERAGE)) {
2894 flags |= AST_FLAG_COVERAGE;
2895 if (!parser_next(parser)) {
2896 error_in_coverage:
2897 parseerror(parser, "parse error in coverage attribute");
2898 *cvq = CV_WRONG;
2899 return false;
2900 }
2901 if (parser->tok == '(') {
2902 if (!parser_next(parser)) {
2903 bad_coverage_arg:
2904 parseerror(parser, "invalid parameter for coverage() attribute\n"
2905 "valid are: block");
2906 *cvq = CV_WRONG;
2907 return false;
2908 }
2909 if (parser->tok != ')') {
2910 do {
2911 if (parser->tok != TOKEN_IDENT)
2912 goto bad_coverage_arg;
2913 if (!strcmp(parser_tokval(parser), "block"))
2914 flags |= AST_FLAG_BLOCK_COVERAGE;
2915 else if (!strcmp(parser_tokval(parser), "none"))
2916 flags &= ~(AST_FLAG_COVERAGE_MASK);
2917 else
2918 goto bad_coverage_arg;
2919 if (!parser_next(parser))
2920 goto error_in_coverage;
2921 if (parser->tok == ',') {
2922 if (!parser_next(parser))
2923 goto error_in_coverage;
2924 }
2925 } while (parser->tok != ')');
2926 }
2927 if (parser->tok != ')' || !parser_next(parser))
2928 goto error_in_coverage;
2929 } else {
2930 /* without parameter [[coverage]] equals [[coverage(block)]] */
2931 flags |= AST_FLAG_BLOCK_COVERAGE;
2932 }
2933 }
2934 else
2935 {
2936 /* Skip tokens until we hit a ]] */
2937 (void)!parsewarning(parser, WARN_UNKNOWN_ATTRIBUTE, "unknown attribute starting with `%s`", parser_tokval(parser));
2938 while (parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
2939 if (!parser_next(parser)) {
2940 parseerror(parser, "error inside attribute");
2941 *cvq = CV_WRONG;
2942 return false;
2943 }
2944 }
2945 }
2946 }
2947 else if (with_local && !strcmp(parser_tokval(parser), "static"))
2948 had_static = true;
2949 else if (!strcmp(parser_tokval(parser), "const"))
2950 had_const = true;
2951 else if (!strcmp(parser_tokval(parser), "var"))
2952 had_var = true;
2953 else if (with_local && !strcmp(parser_tokval(parser), "local"))
2954 had_var = true;
2955 else if (!strcmp(parser_tokval(parser), "noref"))
2956 had_noref = true;
2957 else if (!had_const && !had_var && !had_noref && !had_attrib && !had_static && !flags) {
2958 return false;
2959 }
2960 else
2961 break;
2962
2963 leave:
2964 if (!parser_next(parser))
2965 goto onerr;
2966 }
2967 if (had_const)
2968 *cvq = CV_CONST;
2969 else if (had_var)
2970 *cvq = CV_VAR;
2971 else
2972 *cvq = CV_NONE;
2973 *noref = had_noref;
2974 *is_static = had_static;
2975 *_flags = flags;
2976 return true;
2977 onerr:
2978 parseerror(parser, "parse error after variable qualifier");
2979 *cvq = CV_WRONG;
2980 return true;
2981 }
2982
2983 static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression **out);
parse_switch(parser_t * parser,ast_block * block,ast_expression ** out)2984 static bool parse_switch(parser_t *parser, ast_block *block, ast_expression **out)
2985 {
2986 bool rv;
2987 char *label = nullptr;
2988
2989 /* skip the 'while' and get the body */
2990 if (!parser_next(parser)) {
2991 if (OPTS_FLAG(LOOP_LABELS))
2992 parseerror(parser, "expected loop label or 'switch' operand in parenthesis");
2993 else
2994 parseerror(parser, "expected 'switch' operand in parenthesis");
2995 return false;
2996 }
2997
2998 if (parser->tok == ':') {
2999 if (!OPTS_FLAG(LOOP_LABELS))
3000 parseerror(parser, "labeled loops not activated, try using -floop-labels");
3001 if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
3002 parseerror(parser, "expected loop label");
3003 return false;
3004 }
3005 label = util_strdup(parser_tokval(parser));
3006 if (!parser_next(parser)) {
3007 mem_d(label);
3008 parseerror(parser, "expected 'switch' operand in parenthesis");
3009 return false;
3010 }
3011 }
3012
3013 if (parser->tok != '(') {
3014 parseerror(parser, "expected 'switch' operand in parenthesis");
3015 return false;
3016 }
3017
3018 parser->breaks.push_back(label);
3019
3020 rv = parse_switch_go(parser, block, out);
3021 if (label)
3022 mem_d(label);
3023 if (parser->breaks.back() != label) {
3024 parseerror(parser, "internal error: label stack corrupted");
3025 rv = false;
3026 delete *out;
3027 *out = nullptr;
3028 }
3029 else {
3030 parser->breaks.pop_back();
3031 }
3032 return rv;
3033 }
3034
parse_switch_go(parser_t * parser,ast_block * block,ast_expression ** out)3035 static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression **out)
3036 {
3037 ast_expression *operand;
3038 ast_value *opval;
3039 ast_value *typevar;
3040 ast_switch *switchnode;
3041 ast_switch_case swcase;
3042
3043 int cvq;
3044 bool noref, is_static;
3045 uint32_t qflags = 0;
3046
3047 lex_ctx_t ctx = parser_ctx(parser);
3048
3049 (void)block; /* not touching */
3050 (void)opval;
3051
3052 /* parse into the expression */
3053 if (!parser_next(parser)) {
3054 parseerror(parser, "expected switch operand");
3055 return false;
3056 }
3057 /* parse the operand */
3058 operand = parse_expression_leave(parser, false, false, false);
3059 if (!operand)
3060 return false;
3061
3062 switchnode = new ast_switch(ctx, operand);
3063
3064 /* closing paren */
3065 if (parser->tok != ')') {
3066 delete switchnode;
3067 parseerror(parser, "expected closing paren after 'switch' operand");
3068 return false;
3069 }
3070
3071 /* parse over the opening paren */
3072 if (!parser_next(parser) || parser->tok != '{') {
3073 delete switchnode;
3074 parseerror(parser, "expected list of cases");
3075 return false;
3076 }
3077
3078 if (!parser_next(parser)) {
3079 delete switchnode;
3080 parseerror(parser, "expected 'case' or 'default'");
3081 return false;
3082 }
3083
3084 /* new block; allow some variables to be declared here */
3085 parser_enterblock(parser);
3086 while (true) {
3087 typevar = nullptr;
3088 if (parser->tok == TOKEN_IDENT)
3089 typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
3090 if (typevar || parser->tok == TOKEN_TYPENAME) {
3091 if (!parse_variable(parser, block, true, CV_NONE, typevar, false, false, 0, nullptr)) {
3092 delete switchnode;
3093 return false;
3094 }
3095 continue;
3096 }
3097 if (parse_qualifiers(parser, true, &cvq, &noref, &is_static, &qflags, nullptr))
3098 {
3099 if (cvq == CV_WRONG) {
3100 delete switchnode;
3101 return false;
3102 }
3103 if (!parse_variable(parser, block, true, cvq, nullptr, noref, is_static, qflags, nullptr)) {
3104 delete switchnode;
3105 return false;
3106 }
3107 continue;
3108 }
3109 break;
3110 }
3111
3112 /* case list! */
3113 while (parser->tok != '}') {
3114 ast_block *caseblock;
3115
3116 if (!strcmp(parser_tokval(parser), "case")) {
3117 if (!parser_next(parser)) {
3118 delete switchnode;
3119 parseerror(parser, "expected expression for case");
3120 return false;
3121 }
3122 swcase.m_value = parse_expression_leave(parser, false, false, false);
3123 if (!swcase.m_value) {
3124 delete switchnode;
3125 parseerror(parser, "expected expression for case");
3126 return false;
3127 }
3128 if (!OPTS_FLAG(RELAXED_SWITCH)) {
3129 if (!ast_istype(swcase.m_value, ast_value)) { /* || ((ast_value*)swcase.m_value)->m_cvq != CV_CONST) { */
3130 parseerror(parser, "case on non-constant values need to be explicitly enabled via -frelaxed-switch");
3131 ast_unref(operand);
3132 return false;
3133 }
3134 }
3135 }
3136 else if (!strcmp(parser_tokval(parser), "default")) {
3137 swcase.m_value = nullptr;
3138 if (!parser_next(parser)) {
3139 delete switchnode;
3140 parseerror(parser, "expected colon");
3141 return false;
3142 }
3143 }
3144 else {
3145 delete switchnode;
3146 parseerror(parser, "expected 'case' or 'default'");
3147 return false;
3148 }
3149
3150 /* Now the colon and body */
3151 if (parser->tok != ':') {
3152 if (swcase.m_value) ast_unref(swcase.m_value);
3153 delete switchnode;
3154 parseerror(parser, "expected colon");
3155 return false;
3156 }
3157
3158 if (!parser_next(parser)) {
3159 if (swcase.m_value) ast_unref(swcase.m_value);
3160 delete switchnode;
3161 parseerror(parser, "expected statements or case");
3162 return false;
3163 }
3164 caseblock = new ast_block(parser_ctx(parser));
3165 if (!caseblock) {
3166 if (swcase.m_value) ast_unref(swcase.m_value);
3167 delete switchnode;
3168 return false;
3169 }
3170 swcase.m_code = caseblock;
3171 switchnode->m_cases.push_back(swcase);
3172 while (true) {
3173 ast_expression *expr;
3174 if (parser->tok == '}')
3175 break;
3176 if (parser->tok == TOKEN_KEYWORD) {
3177 if (!strcmp(parser_tokval(parser), "case") ||
3178 !strcmp(parser_tokval(parser), "default"))
3179 {
3180 break;
3181 }
3182 }
3183 if (!parse_statement(parser, caseblock, &expr, true)) {
3184 delete switchnode;
3185 return false;
3186 }
3187 if (!expr)
3188 continue;
3189 if (!caseblock->addExpr(expr)) {
3190 delete switchnode;
3191 return false;
3192 }
3193 }
3194 }
3195
3196 parser_leaveblock(parser);
3197
3198 /* closing paren */
3199 if (parser->tok != '}') {
3200 delete switchnode;
3201 parseerror(parser, "expected closing paren of case list");
3202 return false;
3203 }
3204 if (!parser_next(parser)) {
3205 delete switchnode;
3206 parseerror(parser, "parse error after switch");
3207 return false;
3208 }
3209 *out = switchnode;
3210 return true;
3211 }
3212
3213 /* parse computed goto sides */
parse_goto_computed(parser_t * parser,ast_expression ** side)3214 static ast_expression *parse_goto_computed(parser_t *parser, ast_expression **side) {
3215 ast_expression *on_true;
3216 ast_expression *on_false;
3217 ast_expression *cond;
3218
3219 if (!*side)
3220 return nullptr;
3221
3222 if (ast_istype(*side, ast_ternary)) {
3223 ast_ternary *tern = (ast_ternary*)*side;
3224 on_true = parse_goto_computed(parser, &tern->m_on_true);
3225 on_false = parse_goto_computed(parser, &tern->m_on_false);
3226
3227 if (!on_true || !on_false) {
3228 parseerror(parser, "expected label or expression in ternary");
3229 if (on_true) ast_unref(on_true);
3230 if (on_false) ast_unref(on_false);
3231 return nullptr;
3232 }
3233
3234 cond = tern->m_cond;
3235 tern->m_cond = nullptr;
3236 delete tern;
3237 *side = nullptr;
3238 return new ast_ifthen(parser_ctx(parser), cond, on_true, on_false);
3239 } else if (ast_istype(*side, ast_label)) {
3240 ast_goto *gt = new ast_goto(parser_ctx(parser), ((ast_label*)*side)->m_name);
3241 gt->setLabel(reinterpret_cast<ast_label*>(*side));
3242 *side = nullptr;
3243 return gt;
3244 }
3245 return nullptr;
3246 }
3247
parse_goto(parser_t * parser,ast_expression ** out)3248 static bool parse_goto(parser_t *parser, ast_expression **out)
3249 {
3250 ast_goto *gt = nullptr;
3251 ast_expression *lbl;
3252
3253 if (!parser_next(parser))
3254 return false;
3255
3256 if (parser->tok != TOKEN_IDENT) {
3257 ast_expression *expression;
3258
3259 /* could be an expression i.e computed goto :-) */
3260 if (parser->tok != '(') {
3261 parseerror(parser, "expected label name after `goto`");
3262 return false;
3263 }
3264
3265 /* failed to parse expression for goto */
3266 if (!(expression = parse_expression(parser, false, true)) ||
3267 !(*out = parse_goto_computed(parser, &expression))) {
3268 parseerror(parser, "invalid goto expression");
3269 if(expression)
3270 ast_unref(expression);
3271 return false;
3272 }
3273
3274 return true;
3275 }
3276
3277 /* not computed goto */
3278 gt = new ast_goto(parser_ctx(parser), parser_tokval(parser));
3279 lbl = parser_find_label(parser, gt->m_name);
3280 if (lbl) {
3281 if (!ast_istype(lbl, ast_label)) {
3282 parseerror(parser, "internal error: label is not an ast_label");
3283 delete gt;
3284 return false;
3285 }
3286 gt->setLabel(reinterpret_cast<ast_label*>(lbl));
3287 }
3288 else
3289 parser->gotos.push_back(gt);
3290
3291 if (!parser_next(parser) || parser->tok != ';') {
3292 parseerror(parser, "semicolon expected after goto label");
3293 return false;
3294 }
3295 if (!parser_next(parser)) {
3296 parseerror(parser, "parse error after goto");
3297 return false;
3298 }
3299
3300 *out = gt;
3301 return true;
3302 }
3303
parse_skipwhite(parser_t * parser)3304 static bool parse_skipwhite(parser_t *parser)
3305 {
3306 do {
3307 if (!parser_next(parser))
3308 return false;
3309 } while (parser->tok == TOKEN_WHITE && parser->tok < TOKEN_ERROR);
3310 return parser->tok < TOKEN_ERROR;
3311 }
3312
parse_eol(parser_t * parser)3313 static bool parse_eol(parser_t *parser)
3314 {
3315 if (!parse_skipwhite(parser))
3316 return false;
3317 return parser->tok == TOKEN_EOL;
3318 }
3319
parse_pragma_do(parser_t * parser)3320 static bool parse_pragma_do(parser_t *parser)
3321 {
3322 if (!parser_next(parser) ||
3323 parser->tok != TOKEN_IDENT ||
3324 strcmp(parser_tokval(parser), "pragma"))
3325 {
3326 parseerror(parser, "expected `pragma` keyword after `#`, got `%s`", parser_tokval(parser));
3327 return false;
3328 }
3329 if (!parse_skipwhite(parser) || parser->tok != TOKEN_IDENT) {
3330 parseerror(parser, "expected pragma, got `%s`", parser_tokval(parser));
3331 return false;
3332 }
3333
3334 if (!strcmp(parser_tokval(parser), "noref")) {
3335 if (!parse_skipwhite(parser) || parser->tok != TOKEN_INTCONST) {
3336 parseerror(parser, "`noref` pragma requires an argument: 0 or 1");
3337 return false;
3338 }
3339 parser->noref = !!parser_token(parser)->constval.i;
3340 if (!parse_eol(parser)) {
3341 parseerror(parser, "parse error after `noref` pragma");
3342 return false;
3343 }
3344 }
3345 else
3346 {
3347 (void)!parsewarning(parser, WARN_UNKNOWN_PRAGMAS, "ignoring #pragma %s", parser_tokval(parser));
3348
3349 /* skip to eol */
3350 while (!parse_eol(parser)) {
3351 parser_next(parser);
3352 }
3353
3354 return true;
3355 }
3356
3357 return true;
3358 }
3359
parse_pragma(parser_t * parser)3360 static bool parse_pragma(parser_t *parser)
3361 {
3362 bool rv;
3363 parser->lex->flags.preprocessing = true;
3364 parser->lex->flags.mergelines = true;
3365 rv = parse_pragma_do(parser);
3366 if (parser->tok != TOKEN_EOL) {
3367 parseerror(parser, "junk after pragma");
3368 rv = false;
3369 }
3370 parser->lex->flags.preprocessing = false;
3371 parser->lex->flags.mergelines = false;
3372 if (!parser_next(parser)) {
3373 parseerror(parser, "parse error after pragma");
3374 rv = false;
3375 }
3376 return rv;
3377 }
3378
parse_statement(parser_t * parser,ast_block * block,ast_expression ** out,bool allow_cases)3379 static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases)
3380 {
3381 bool noref, is_static;
3382 int cvq = CV_NONE;
3383 uint32_t qflags = 0;
3384 ast_value *typevar = nullptr;
3385 char *vstring = nullptr;
3386
3387 *out = nullptr;
3388
3389 if (parser->tok == TOKEN_IDENT)
3390 typevar = parser_find_typedef(parser, parser_tokval(parser), 0);
3391
3392 if (typevar || parser->tok == TOKEN_TYPENAME || parser->tok == '.' || parser->tok == TOKEN_DOTS)
3393 {
3394 /* local variable */
3395 if (!block) {
3396 parseerror(parser, "cannot declare a variable from here");
3397 return false;
3398 }
3399 if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) {
3400 if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable"))
3401 return false;
3402 }
3403 if (!parse_variable(parser, block, false, CV_NONE, typevar, false, false, 0, nullptr))
3404 return false;
3405 return true;
3406 }
3407 else if (parse_qualifiers(parser, !!block, &cvq, &noref, &is_static, &qflags, &vstring))
3408 {
3409 if (cvq == CV_WRONG)
3410 return false;
3411 return parse_variable(parser, block, false, cvq, nullptr, noref, is_static, qflags, vstring);
3412 }
3413 else if (parser->tok == TOKEN_KEYWORD)
3414 {
3415 if (!strcmp(parser_tokval(parser), "__builtin_debug_printtype"))
3416 {
3417 char ty[1024];
3418 ast_value *tdef;
3419
3420 if (!parser_next(parser)) {
3421 parseerror(parser, "parse error after __builtin_debug_printtype");
3422 return false;
3423 }
3424
3425 if (parser->tok == TOKEN_IDENT && (tdef = parser_find_typedef(parser, parser_tokval(parser), 0)))
3426 {
3427 ast_type_to_string(tdef, ty, sizeof(ty));
3428 con_out("__builtin_debug_printtype: `%s`=`%s`\n", tdef->m_name.c_str(), ty);
3429 if (!parser_next(parser)) {
3430 parseerror(parser, "parse error after __builtin_debug_printtype typename argument");
3431 return false;
3432 }
3433 }
3434 else
3435 {
3436 if (!parse_statement(parser, block, out, allow_cases))
3437 return false;
3438 if (!*out)
3439 con_out("__builtin_debug_printtype: got no output node\n");
3440 else
3441 {
3442 ast_type_to_string(*out, ty, sizeof(ty));
3443 con_out("__builtin_debug_printtype: `%s`\n", ty);
3444 }
3445 }
3446 return true;
3447 }
3448 else if (!strcmp(parser_tokval(parser), "return"))
3449 {
3450 return parse_return(parser, block, out);
3451 }
3452 else if (!strcmp(parser_tokval(parser), "if"))
3453 {
3454 return parse_if(parser, block, out);
3455 }
3456 else if (!strcmp(parser_tokval(parser), "while"))
3457 {
3458 return parse_while(parser, block, out);
3459 }
3460 else if (!strcmp(parser_tokval(parser), "do"))
3461 {
3462 return parse_dowhile(parser, block, out);
3463 }
3464 else if (!strcmp(parser_tokval(parser), "for"))
3465 {
3466 if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) {
3467 if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?"))
3468 return false;
3469 }
3470 return parse_for(parser, block, out);
3471 }
3472 else if (!strcmp(parser_tokval(parser), "break"))
3473 {
3474 return parse_break_continue(parser, block, out, false);
3475 }
3476 else if (!strcmp(parser_tokval(parser), "continue"))
3477 {
3478 return parse_break_continue(parser, block, out, true);
3479 }
3480 else if (!strcmp(parser_tokval(parser), "switch"))
3481 {
3482 return parse_switch(parser, block, out);
3483 }
3484 else if (!strcmp(parser_tokval(parser), "case") ||
3485 !strcmp(parser_tokval(parser), "default"))
3486 {
3487 if (!allow_cases) {
3488 parseerror(parser, "unexpected 'case' label");
3489 return false;
3490 }
3491 return true;
3492 }
3493 else if (!strcmp(parser_tokval(parser), "goto"))
3494 {
3495 return parse_goto(parser, out);
3496 }
3497 else if (!strcmp(parser_tokval(parser), "typedef"))
3498 {
3499 if (!parser_next(parser)) {
3500 parseerror(parser, "expected type definition after 'typedef'");
3501 return false;
3502 }
3503 return parse_typedef(parser);
3504 }
3505 parseerror(parser, "Unexpected keyword: `%s'", parser_tokval(parser));
3506 return false;
3507 }
3508 else if (parser->tok == '{')
3509 {
3510 ast_block *inner;
3511 inner = parse_block(parser);
3512 if (!inner)
3513 return false;
3514 *out = inner;
3515 return true;
3516 }
3517 else if (parser->tok == ':')
3518 {
3519 size_t i;
3520 ast_label *label;
3521 if (!parser_next(parser)) {
3522 parseerror(parser, "expected label name");
3523 return false;
3524 }
3525 if (parser->tok != TOKEN_IDENT) {
3526 parseerror(parser, "label must be an identifier");
3527 return false;
3528 }
3529 label = (ast_label*)parser_find_label(parser, parser_tokval(parser));
3530 if (label) {
3531 if (!label->m_undefined) {
3532 parseerror(parser, "label `%s` already defined", label->m_name);
3533 return false;
3534 }
3535 label->m_undefined = false;
3536 }
3537 else {
3538 label = new ast_label(parser_ctx(parser), parser_tokval(parser), false);
3539 parser->labels.push_back(label);
3540 }
3541 *out = label;
3542 if (!parser_next(parser)) {
3543 parseerror(parser, "parse error after label");
3544 return false;
3545 }
3546 for (i = 0; i < parser->gotos.size(); ++i) {
3547 if (parser->gotos[i]->m_name == label->m_name) {
3548 parser->gotos[i]->setLabel(label);
3549 parser->gotos.erase(parser->gotos.begin() + i);
3550 --i;
3551 }
3552 }
3553 return true;
3554 }
3555 else if (parser->tok == ';')
3556 {
3557 if (!parser_next(parser)) {
3558 parseerror(parser, "parse error after empty statement");
3559 return false;
3560 }
3561 return true;
3562 }
3563 else
3564 {
3565 lex_ctx_t ctx = parser_ctx(parser);
3566 ast_expression *exp = parse_expression(parser, false, false);
3567 if (!exp)
3568 return false;
3569 *out = exp;
3570 if (!exp->m_side_effects) {
3571 if (compile_warning(ctx, WARN_EFFECTLESS_STATEMENT, "statement has no effect"))
3572 return false;
3573 }
3574 return true;
3575 }
3576 }
3577
parse_enum(parser_t * parser)3578 static bool parse_enum(parser_t *parser)
3579 {
3580 bool flag = false;
3581 bool reverse = false;
3582 qcfloat_t num = 0;
3583 ast_value **values = nullptr;
3584 ast_value *var = nullptr;
3585 ast_value *asvalue;
3586
3587 ast_expression *old;
3588
3589 if (!parser_next(parser) || (parser->tok != '{' && parser->tok != ':')) {
3590 parseerror(parser, "expected `{` or `:` after `enum` keyword");
3591 return false;
3592 }
3593
3594 /* enumeration attributes (can add more later) */
3595 if (parser->tok == ':') {
3596 if (!parser_next(parser) || parser->tok != TOKEN_IDENT){
3597 parseerror(parser, "expected `flag` or `reverse` for enumeration attribute");
3598 return false;
3599 }
3600
3601 /* attributes? */
3602 if (!strcmp(parser_tokval(parser), "flag")) {
3603 num = 1;
3604 flag = true;
3605 }
3606 else if (!strcmp(parser_tokval(parser), "reverse")) {
3607 reverse = true;
3608 }
3609 else {
3610 parseerror(parser, "invalid attribute `%s` for enumeration", parser_tokval(parser));
3611 return false;
3612 }
3613
3614 if (!parser_next(parser) || parser->tok != '{') {
3615 parseerror(parser, "expected `{` after enum attribute ");
3616 return false;
3617 }
3618 }
3619
3620 while (true) {
3621 if (!parser_next(parser) || parser->tok != TOKEN_IDENT) {
3622 if (parser->tok == '}') {
3623 /* allow an empty enum */
3624 break;
3625 }
3626 parseerror(parser, "expected identifier or `}`");
3627 goto onerror;
3628 }
3629
3630 old = parser_find_field(parser, parser_tokval(parser));
3631 if (!old)
3632 old = parser_find_global(parser, parser_tokval(parser));
3633 if (old) {
3634 parseerror(parser, "value `%s` has already been declared here: %s:%i",
3635 parser_tokval(parser), old->m_context.file, old->m_context.line);
3636 goto onerror;
3637 }
3638
3639 var = new ast_value(parser_ctx(parser), parser_tokval(parser), TYPE_FLOAT);
3640 vec_push(values, var);
3641 var->m_cvq = CV_CONST;
3642 var->m_hasvalue = true;
3643
3644 /* for flagged enumerations increment in POTs of TWO */
3645 var->m_constval.vfloat = (flag) ? (num *= 2) : (num ++);
3646 parser_addglobal(parser, var->m_name, var);
3647
3648 if (!parser_next(parser)) {
3649 parseerror(parser, "expected `=`, `}` or comma after identifier");
3650 goto onerror;
3651 }
3652
3653 if (parser->tok == ',')
3654 continue;
3655 if (parser->tok == '}')
3656 break;
3657 if (parser->tok != '=') {
3658 parseerror(parser, "expected `=`, `}` or comma after identifier");
3659 goto onerror;
3660 }
3661
3662 if (!parser_next(parser)) {
3663 parseerror(parser, "expected expression after `=`");
3664 goto onerror;
3665 }
3666
3667 /* We got a value! */
3668 old = parse_expression_leave(parser, true, false, false);
3669 asvalue = (ast_value*)old;
3670 if (!ast_istype(old, ast_value) || asvalue->m_cvq != CV_CONST || !asvalue->m_hasvalue) {
3671 compile_error(var->m_context, "constant value or expression expected");
3672 goto onerror;
3673 }
3674 num = (var->m_constval.vfloat = asvalue->m_constval.vfloat) + 1;
3675
3676 if (parser->tok == '}')
3677 break;
3678 if (parser->tok != ',') {
3679 parseerror(parser, "expected `}` or comma after expression");
3680 goto onerror;
3681 }
3682 }
3683
3684 /* patch them all (for reversed attribute) */
3685 if (reverse) {
3686 size_t i;
3687 for (i = 0; i < vec_size(values); i++)
3688 values[i]->m_constval.vfloat = vec_size(values) - i - 1;
3689 }
3690
3691 if (parser->tok != '}') {
3692 parseerror(parser, "internal error: breaking without `}`");
3693 goto onerror;
3694 }
3695
3696 if (!parser_next(parser) || parser->tok != ';') {
3697 parseerror(parser, "expected semicolon after enumeration");
3698 goto onerror;
3699 }
3700
3701 if (!parser_next(parser)) {
3702 parseerror(parser, "parse error after enumeration");
3703 goto onerror;
3704 }
3705
3706 vec_free(values);
3707 return true;
3708
3709 onerror:
3710 vec_free(values);
3711 return false;
3712 }
3713
parse_block_into(parser_t * parser,ast_block * block)3714 static bool parse_block_into(parser_t *parser, ast_block *block)
3715 {
3716 bool retval = true;
3717
3718 parser_enterblock(parser);
3719
3720 if (!parser_next(parser)) { /* skip the '{' */
3721 parseerror(parser, "expected function body");
3722 goto cleanup;
3723 }
3724
3725 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
3726 {
3727 ast_expression *expr = nullptr;
3728 if (parser->tok == '}')
3729 break;
3730
3731 if (!parse_statement(parser, block, &expr, false)) {
3732 /* parseerror(parser, "parse error"); */
3733 block = nullptr;
3734 goto cleanup;
3735 }
3736 if (!expr)
3737 continue;
3738 if (!block->addExpr(expr)) {
3739 delete block;
3740 block = nullptr;
3741 goto cleanup;
3742 }
3743 }
3744
3745 if (parser->tok != '}') {
3746 block = nullptr;
3747 } else {
3748 (void)parser_next(parser);
3749 }
3750
3751 cleanup:
3752 if (!parser_leaveblock(parser))
3753 retval = false;
3754 return retval && !!block;
3755 }
3756
parse_block(parser_t * parser)3757 static ast_block* parse_block(parser_t *parser)
3758 {
3759 ast_block *block;
3760 block = new ast_block(parser_ctx(parser));
3761 if (!block)
3762 return nullptr;
3763 if (!parse_block_into(parser, block)) {
3764 delete block;
3765 return nullptr;
3766 }
3767 return block;
3768 }
3769
parse_statement_or_block(parser_t * parser,ast_expression ** out)3770 static bool parse_statement_or_block(parser_t *parser, ast_expression **out)
3771 {
3772 if (parser->tok == '{') {
3773 *out = parse_block(parser);
3774 return !!*out;
3775 }
3776 return parse_statement(parser, nullptr, out, false);
3777 }
3778
create_vector_members(ast_value * var,ast_member ** me)3779 static bool create_vector_members(ast_value *var, ast_member **me)
3780 {
3781 size_t i;
3782 size_t len = var->m_name.length();
3783
3784 for (i = 0; i < 3; ++i) {
3785 char *name = (char*)mem_a(len+3);
3786 memcpy(name, var->m_name.c_str(), len);
3787 name[len+0] = '_';
3788 name[len+1] = 'x'+i;
3789 name[len+2] = 0;
3790 me[i] = ast_member::make(var->m_context, var, i, name);
3791 mem_d(name);
3792 if (!me[i])
3793 break;
3794 }
3795 if (i == 3)
3796 return true;
3797
3798 /* unroll */
3799 do { delete me[--i]; } while(i);
3800 return false;
3801 }
3802
parse_function_body(parser_t * parser,ast_value * var)3803 static bool parse_function_body(parser_t *parser, ast_value *var)
3804 {
3805 ast_block *block = nullptr;
3806 ast_function *func;
3807 ast_function *old;
3808
3809 ast_expression *framenum = nullptr;
3810 ast_expression *nextthink = nullptr;
3811 /* None of the following have to be deleted */
3812 ast_expression *fld_think = nullptr, *fld_nextthink = nullptr, *fld_frame = nullptr;
3813 ast_expression *gbl_time = nullptr, *gbl_self = nullptr;
3814 bool has_frame_think;
3815
3816 bool retval = true;
3817
3818 has_frame_think = false;
3819 old = parser->function;
3820
3821 if (var->m_flags & AST_FLAG_ALIAS) {
3822 parseerror(parser, "function aliases cannot have bodies");
3823 return false;
3824 }
3825
3826 if (parser->gotos.size() || parser->labels.size()) {
3827 parseerror(parser, "gotos/labels leaking");
3828 return false;
3829 }
3830
3831 if (!OPTS_FLAG(VARIADIC_ARGS) && var->m_flags & AST_FLAG_VARIADIC) {
3832 if (parsewarning(parser, WARN_VARIADIC_FUNCTION,
3833 "variadic function with implementation will not be able to access additional parameters (try -fvariadic-args)"))
3834 {
3835 return false;
3836 }
3837 }
3838
3839 if (parser->tok == '[') {
3840 /* got a frame definition: [ framenum, nextthink ]
3841 * this translates to:
3842 * self.frame = framenum;
3843 * self.nextthink = time + 0.1;
3844 * self.think = nextthink;
3845 */
3846 nextthink = nullptr;
3847
3848 fld_think = parser_find_field(parser, "think");
3849 fld_nextthink = parser_find_field(parser, "nextthink");
3850 fld_frame = parser_find_field(parser, "frame");
3851 if (!fld_think || !fld_nextthink || !fld_frame) {
3852 parseerror(parser, "cannot use [frame,think] notation without the required fields");
3853 parseerror(parser, "please declare the following entityfields: `frame`, `think`, `nextthink`");
3854 return false;
3855 }
3856 gbl_time = parser_find_global(parser, "time");
3857 gbl_self = parser_find_global(parser, "self");
3858 if (!gbl_time || !gbl_self) {
3859 parseerror(parser, "cannot use [frame,think] notation without the required globals");
3860 parseerror(parser, "please declare the following globals: `time`, `self`");
3861 return false;
3862 }
3863
3864 if (!parser_next(parser))
3865 return false;
3866
3867 framenum = parse_expression_leave(parser, true, false, false);
3868 if (!framenum) {
3869 parseerror(parser, "expected a framenumber constant in[frame,think] notation");
3870 return false;
3871 }
3872 if (!ast_istype(framenum, ast_value) || !( (ast_value*)framenum )->m_hasvalue) {
3873 ast_unref(framenum);
3874 parseerror(parser, "framenumber in [frame,think] notation must be a constant");
3875 return false;
3876 }
3877
3878 if (parser->tok != ',') {
3879 ast_unref(framenum);
3880 parseerror(parser, "expected comma after frame number in [frame,think] notation");
3881 parseerror(parser, "Got a %i\n", parser->tok);
3882 return false;
3883 }
3884
3885 if (!parser_next(parser)) {
3886 ast_unref(framenum);
3887 return false;
3888 }
3889
3890 if (parser->tok == TOKEN_IDENT && !parser_find_var(parser, parser_tokval(parser)))
3891 {
3892 /* qc allows the use of not-yet-declared functions here
3893 * - this automatically creates a prototype */
3894 ast_value *thinkfunc;
3895 ast_expression *functype = fld_think->m_next;
3896
3897 thinkfunc = new ast_value(parser_ctx(parser), parser_tokval(parser), functype->m_vtype);
3898 if (!thinkfunc) { /* || !thinkfunc->adoptType(*functype)*/
3899 ast_unref(framenum);
3900 parseerror(parser, "failed to create implicit prototype for `%s`", parser_tokval(parser));
3901 return false;
3902 }
3903 thinkfunc->adoptType(*functype);
3904
3905 if (!parser_next(parser)) {
3906 ast_unref(framenum);
3907 delete thinkfunc;
3908 return false;
3909 }
3910
3911 parser_addglobal(parser, thinkfunc->m_name, thinkfunc);
3912
3913 nextthink = thinkfunc;
3914
3915 } else {
3916 nextthink = parse_expression_leave(parser, true, false, false);
3917 if (!nextthink) {
3918 ast_unref(framenum);
3919 parseerror(parser, "expected a think-function in [frame,think] notation");
3920 return false;
3921 }
3922 }
3923
3924 if (!ast_istype(nextthink, ast_value)) {
3925 parseerror(parser, "think-function in [frame,think] notation must be a constant");
3926 retval = false;
3927 }
3928
3929 if (retval && parser->tok != ']') {
3930 parseerror(parser, "expected closing `]` for [frame,think] notation");
3931 retval = false;
3932 }
3933
3934 if (retval && !parser_next(parser)) {
3935 retval = false;
3936 }
3937
3938 if (retval && parser->tok != '{') {
3939 parseerror(parser, "a function body has to be declared after a [frame,think] declaration");
3940 retval = false;
3941 }
3942
3943 if (!retval) {
3944 ast_unref(nextthink);
3945 ast_unref(framenum);
3946 return false;
3947 }
3948
3949 has_frame_think = true;
3950 }
3951
3952 block = new ast_block(parser_ctx(parser));
3953 if (!block) {
3954 parseerror(parser, "failed to allocate block");
3955 if (has_frame_think) {
3956 ast_unref(nextthink);
3957 ast_unref(framenum);
3958 }
3959 return false;
3960 }
3961
3962 if (has_frame_think) {
3963 if (!OPTS_FLAG(EMULATE_STATE)) {
3964 ast_state *state_op = new ast_state(parser_ctx(parser), framenum, nextthink);
3965 if (!block->addExpr(state_op)) {
3966 parseerror(parser, "failed to generate state op for [frame,think]");
3967 ast_unref(nextthink);
3968 ast_unref(framenum);
3969 delete block;
3970 return false;
3971 }
3972 } else {
3973 /* emulate OP_STATE in code: */
3974 lex_ctx_t ctx;
3975 ast_expression *self_frame;
3976 ast_expression *self_nextthink;
3977 ast_expression *self_think;
3978 ast_expression *time_plus_1;
3979 ast_store *store_frame;
3980 ast_store *store_nextthink;
3981 ast_store *store_think;
3982
3983 float frame_delta = 1.0f / (float)OPTS_OPTION_U32(OPTION_STATE_FPS);
3984
3985 ctx = parser_ctx(parser);
3986 self_frame = new ast_entfield(ctx, gbl_self, fld_frame);
3987 self_nextthink = new ast_entfield(ctx, gbl_self, fld_nextthink);
3988 self_think = new ast_entfield(ctx, gbl_self, fld_think);
3989
3990 time_plus_1 = new ast_binary(ctx, INSTR_ADD_F,
3991 gbl_time, parser->m_fold.constgen_float(frame_delta, false));
3992
3993 if (!self_frame || !self_nextthink || !self_think || !time_plus_1) {
3994 if (self_frame) delete self_frame;
3995 if (self_nextthink) delete self_nextthink;
3996 if (self_think) delete self_think;
3997 if (time_plus_1) delete time_plus_1;
3998 retval = false;
3999 }
4000
4001 if (retval)
4002 {
4003 store_frame = new ast_store(ctx, INSTR_STOREP_F, self_frame, framenum);
4004 store_nextthink = new ast_store(ctx, INSTR_STOREP_F, self_nextthink, time_plus_1);
4005 store_think = new ast_store(ctx, INSTR_STOREP_FNC, self_think, nextthink);
4006
4007 if (!store_frame) {
4008 delete self_frame;
4009 retval = false;
4010 }
4011 if (!store_nextthink) {
4012 delete self_nextthink;
4013 retval = false;
4014 }
4015 if (!store_think) {
4016 delete self_think;
4017 retval = false;
4018 }
4019 if (!retval) {
4020 if (store_frame) delete store_frame;
4021 if (store_nextthink) delete store_nextthink;
4022 if (store_think) delete store_think;
4023 retval = false;
4024 }
4025 if (!block->addExpr(store_frame) ||
4026 !block->addExpr(store_nextthink) ||
4027 !block->addExpr(store_think))
4028 {
4029 retval = false;
4030 }
4031 }
4032
4033 if (!retval) {
4034 parseerror(parser, "failed to generate code for [frame,think]");
4035 ast_unref(nextthink);
4036 ast_unref(framenum);
4037 delete block;
4038 return false;
4039 }
4040 }
4041 }
4042
4043 if (var->m_hasvalue) {
4044 if (!(var->m_flags & AST_FLAG_ACCUMULATE)) {
4045 parseerror(parser, "function `%s` declared with multiple bodies", var->m_name);
4046 delete block;
4047 goto enderr;
4048 }
4049 func = var->m_constval.vfunc;
4050
4051 if (!func) {
4052 parseerror(parser, "internal error: nullptr function: `%s`", var->m_name);
4053 delete block;
4054 goto enderr;
4055 }
4056 } else {
4057 func = ast_function::make(var->m_context, var->m_name, var);
4058
4059 if (!func) {
4060 parseerror(parser, "failed to allocate function for `%s`", var->m_name);
4061 delete block;
4062 goto enderr;
4063 }
4064 parser->functions.push_back(func);
4065 }
4066
4067 parser_enterblock(parser);
4068
4069 for (auto &it : var->m_type_params) {
4070 size_t e;
4071 ast_member *me[3];
4072
4073 if (it->m_vtype != TYPE_VECTOR &&
4074 (it->m_vtype != TYPE_FIELD ||
4075 it->m_next->m_vtype != TYPE_VECTOR))
4076 {
4077 continue;
4078 }
4079
4080 if (!create_vector_members(it.get(), me)) {
4081 delete block;
4082 goto enderrfn;
4083 }
4084
4085 for (e = 0; e < 3; ++e) {
4086 parser_addlocal(parser, me[e]->m_name, me[e]);
4087 block->collect(me[e]);
4088 }
4089 }
4090
4091 if (var->m_argcounter && !func->m_argc) {
4092 ast_value *argc = new ast_value(var->m_context, var->m_argcounter, TYPE_FLOAT);
4093 parser_addlocal(parser, argc->m_name, argc);
4094 func->m_argc.reset(argc);
4095 }
4096
4097 if (OPTS_FLAG(VARIADIC_ARGS) && var->m_flags & AST_FLAG_VARIADIC && !func->m_varargs) {
4098 char name[1024];
4099 ast_value *varargs = new ast_value(var->m_context, "reserved:va_args", TYPE_ARRAY);
4100 varargs->m_flags |= AST_FLAG_IS_VARARG;
4101 varargs->m_next = new ast_value(var->m_context, "", TYPE_VECTOR);
4102 varargs->m_count = 0;
4103 util_snprintf(name, sizeof(name), "%s##va##SET", var->m_name.c_str());
4104 if (!parser_create_array_setter_proto(parser, varargs, name)) {
4105 delete varargs;
4106 delete block;
4107 goto enderrfn;
4108 }
4109 util_snprintf(name, sizeof(name), "%s##va##GET", var->m_name.c_str());
4110 if (!parser_create_array_getter_proto(parser, varargs, varargs->m_next, name)) {
4111 delete varargs;
4112 delete block;
4113 goto enderrfn;
4114 }
4115 func->m_varargs.reset(varargs);
4116 func->m_fixedparams = (ast_value*)parser->m_fold.constgen_float(var->m_type_params.size(), false);
4117 }
4118
4119 parser->function = func;
4120 if (!parse_block_into(parser, block)) {
4121 delete block;
4122 goto enderrfn;
4123 }
4124
4125 func->m_blocks.emplace_back(block);
4126
4127 parser->function = old;
4128 if (!parser_leaveblock(parser))
4129 retval = false;
4130 if (vec_size(parser->variables) != PARSER_HT_LOCALS) {
4131 parseerror(parser, "internal error: local scopes left");
4132 retval = false;
4133 }
4134
4135 if (parser->tok == ';')
4136 return parser_next(parser);
4137 else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC)
4138 parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)");
4139 return retval;
4140
4141 enderrfn:
4142 (void)!parser_leaveblock(parser);
4143 parser->functions.pop_back();
4144 delete func;
4145 var->m_constval.vfunc = nullptr;
4146
4147 enderr:
4148 parser->function = old;
4149 return false;
4150 }
4151
array_accessor_split(parser_t * parser,ast_value * array,ast_value * index,size_t middle,ast_expression * left,ast_expression * right)4152 static ast_expression *array_accessor_split(
4153 parser_t *parser,
4154 ast_value *array,
4155 ast_value *index,
4156 size_t middle,
4157 ast_expression *left,
4158 ast_expression *right
4159 )
4160 {
4161 ast_ifthen *ifthen;
4162 ast_binary *cmp;
4163
4164 lex_ctx_t ctx = array->m_context;
4165
4166 if (!left || !right) {
4167 if (left) delete left;
4168 if (right) delete right;
4169 return nullptr;
4170 }
4171
4172 cmp = new ast_binary(ctx, INSTR_LT,
4173 index,
4174 parser->m_fold.constgen_float(middle, false));
4175 if (!cmp) {
4176 delete left;
4177 delete right;
4178 parseerror(parser, "internal error: failed to create comparison for array setter");
4179 return nullptr;
4180 }
4181
4182 ifthen = new ast_ifthen(ctx, cmp, left, right);
4183 if (!ifthen) {
4184 delete cmp; /* will delete left and right */
4185 parseerror(parser, "internal error: failed to create conditional jump for array setter");
4186 return nullptr;
4187 }
4188
4189 return ifthen;
4190 }
4191
array_setter_node(parser_t * parser,ast_value * array,ast_value * index,ast_value * value,size_t from,size_t afterend)4192 static ast_expression *array_setter_node(parser_t *parser, ast_value *array, ast_value *index, ast_value *value, size_t from, size_t afterend)
4193 {
4194 lex_ctx_t ctx = array->m_context;
4195
4196 if (from+1 == afterend) {
4197 /* set this value */
4198 ast_block *block;
4199 ast_return *ret;
4200 ast_array_index *subscript;
4201 ast_store *st;
4202 int assignop = type_store_instr[value->m_vtype];
4203
4204 if (value->m_vtype == TYPE_FIELD && value->m_next->m_vtype == TYPE_VECTOR)
4205 assignop = INSTR_STORE_V;
4206
4207 subscript = ast_array_index::make(ctx, array, parser->m_fold.constgen_float(from, false));
4208 if (!subscript)
4209 return nullptr;
4210
4211 st = new ast_store(ctx, assignop, subscript, value);
4212 if (!st) {
4213 delete subscript;
4214 return nullptr;
4215 }
4216
4217 block = new ast_block(ctx);
4218 if (!block) {
4219 delete st;
4220 return nullptr;
4221 }
4222
4223 if (!block->addExpr(st)) {
4224 delete block;
4225 return nullptr;
4226 }
4227
4228 ret = new ast_return(ctx, nullptr);
4229 if (!ret) {
4230 delete block;
4231 return nullptr;
4232 }
4233
4234 if (!block->addExpr(ret)) {
4235 delete block;
4236 return nullptr;
4237 }
4238
4239 return block;
4240 } else {
4241 ast_expression *left, *right;
4242 size_t diff = afterend - from;
4243 size_t middle = from + diff/2;
4244 left = array_setter_node(parser, array, index, value, from, middle);
4245 right = array_setter_node(parser, array, index, value, middle, afterend);
4246 return array_accessor_split(parser, array, index, middle, left, right);
4247 }
4248 }
4249
array_field_setter_node(parser_t * parser,ast_value * array,ast_value * entity,ast_value * index,ast_value * value,size_t from,size_t afterend)4250 static ast_expression *array_field_setter_node(
4251 parser_t *parser,
4252 ast_value *array,
4253 ast_value *entity,
4254 ast_value *index,
4255 ast_value *value,
4256 size_t from,
4257 size_t afterend)
4258 {
4259 lex_ctx_t ctx = array->m_context;
4260
4261 if (from+1 == afterend) {
4262 /* set this value */
4263 ast_block *block;
4264 ast_return *ret;
4265 ast_entfield *entfield;
4266 ast_array_index *subscript;
4267 ast_store *st;
4268 int assignop = type_storep_instr[value->m_vtype];
4269
4270 if (value->m_vtype == TYPE_FIELD && value->m_next->m_vtype == TYPE_VECTOR)
4271 assignop = INSTR_STOREP_V;
4272
4273 subscript = ast_array_index::make(ctx, array, parser->m_fold.constgen_float(from, false));
4274 if (!subscript)
4275 return nullptr;
4276
4277 subscript->m_next = new ast_expression(ast_copy_type, subscript->m_context, *subscript);
4278 subscript->m_vtype = TYPE_FIELD;
4279
4280 entfield = new ast_entfield(ctx, entity, subscript, subscript);
4281 if (!entfield) {
4282 delete subscript;
4283 return nullptr;
4284 }
4285
4286 st = new ast_store(ctx, assignop, entfield, value);
4287 if (!st) {
4288 delete entfield;
4289 return nullptr;
4290 }
4291
4292 block = new ast_block(ctx);
4293 if (!block) {
4294 delete st;
4295 return nullptr;
4296 }
4297
4298 if (!block->addExpr(st)) {
4299 delete block;
4300 return nullptr;
4301 }
4302
4303 ret = new ast_return(ctx, nullptr);
4304 if (!ret) {
4305 delete block;
4306 return nullptr;
4307 }
4308
4309 if (!block->addExpr(ret)) {
4310 delete block;
4311 return nullptr;
4312 }
4313
4314 return block;
4315 } else {
4316 ast_expression *left, *right;
4317 size_t diff = afterend - from;
4318 size_t middle = from + diff/2;
4319 left = array_field_setter_node(parser, array, entity, index, value, from, middle);
4320 right = array_field_setter_node(parser, array, entity, index, value, middle, afterend);
4321 return array_accessor_split(parser, array, index, middle, left, right);
4322 }
4323 }
4324
array_getter_node(parser_t * parser,ast_value * array,ast_value * index,size_t from,size_t afterend)4325 static ast_expression *array_getter_node(parser_t *parser, ast_value *array, ast_value *index, size_t from, size_t afterend)
4326 {
4327 lex_ctx_t ctx = array->m_context;
4328
4329 if (from+1 == afterend) {
4330 ast_return *ret;
4331 ast_array_index *subscript;
4332
4333 subscript = ast_array_index::make(ctx, array, parser->m_fold.constgen_float(from, false));
4334 if (!subscript)
4335 return nullptr;
4336
4337 ret = new ast_return(ctx, subscript);
4338 if (!ret) {
4339 delete subscript;
4340 return nullptr;
4341 }
4342
4343 return ret;
4344 } else {
4345 ast_expression *left, *right;
4346 size_t diff = afterend - from;
4347 size_t middle = from + diff/2;
4348 left = array_getter_node(parser, array, index, from, middle);
4349 right = array_getter_node(parser, array, index, middle, afterend);
4350 return array_accessor_split(parser, array, index, middle, left, right);
4351 }
4352 }
4353
parser_create_array_accessor(parser_t * parser,ast_value * array,const char * funcname,ast_value ** out)4354 static bool parser_create_array_accessor(parser_t *parser, ast_value *array, const char *funcname, ast_value **out)
4355 {
4356 ast_function *func = nullptr;
4357 ast_value *fval = nullptr;
4358 ast_block *body = nullptr;
4359
4360 fval = new ast_value(array->m_context, funcname, TYPE_FUNCTION);
4361 if (!fval) {
4362 parseerror(parser, "failed to create accessor function value");
4363 return false;
4364 }
4365 fval->m_flags &= ~(AST_FLAG_COVERAGE_MASK);
4366
4367 func = ast_function::make(array->m_context, funcname, fval);
4368 if (!func) {
4369 delete fval;
4370 parseerror(parser, "failed to create accessor function node");
4371 return false;
4372 }
4373
4374 body = new ast_block(array->m_context);
4375 if (!body) {
4376 parseerror(parser, "failed to create block for array accessor");
4377 delete fval;
4378 delete func;
4379 return false;
4380 }
4381
4382 func->m_blocks.emplace_back(body);
4383 *out = fval;
4384
4385 parser->accessors.push_back(fval);
4386
4387 return true;
4388 }
4389
parser_create_array_setter_proto(parser_t * parser,ast_value * array,const char * funcname)4390 static ast_value* parser_create_array_setter_proto(parser_t *parser, ast_value *array, const char *funcname)
4391 {
4392 ast_value *index = nullptr;
4393 ast_value *value = nullptr;
4394 ast_function *func;
4395 ast_value *fval;
4396
4397 if (!ast_istype(array->m_next, ast_value)) {
4398 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
4399 return nullptr;
4400 }
4401
4402 if (!parser_create_array_accessor(parser, array, funcname, &fval))
4403 return nullptr;
4404 func = fval->m_constval.vfunc;
4405 fval->m_next = new ast_value(array->m_context, "<void>", TYPE_VOID);
4406
4407 index = new ast_value(array->m_context, "index", TYPE_FLOAT);
4408 value = new ast_value(ast_copy_type, *(ast_value*)array->m_next);
4409
4410 if (!index || !value) {
4411 parseerror(parser, "failed to create locals for array accessor");
4412 goto cleanup;
4413 }
4414 value->m_name = "value"; // not important
4415 fval->m_type_params.emplace_back(index);
4416 fval->m_type_params.emplace_back(value);
4417
4418 array->m_setter = fval;
4419 return fval;
4420 cleanup:
4421 if (index) delete index;
4422 if (value) delete value;
4423 delete func;
4424 delete fval;
4425 return nullptr;
4426 }
4427
parser_create_array_setter_impl(parser_t * parser,ast_value * array)4428 static bool parser_create_array_setter_impl(parser_t *parser, ast_value *array)
4429 {
4430 ast_expression *root = nullptr;
4431 root = array_setter_node(parser, array,
4432 array->m_setter->m_type_params[0].get(),
4433 array->m_setter->m_type_params[1].get(),
4434 0, array->m_count);
4435 if (!root) {
4436 parseerror(parser, "failed to build accessor search tree");
4437 return false;
4438 }
4439 if (!array->m_setter->m_constval.vfunc->m_blocks[0].get()->addExpr(root)) {
4440 delete root;
4441 return false;
4442 }
4443 return true;
4444 }
4445
parser_create_array_setter(parser_t * parser,ast_value * array,const char * funcname)4446 static bool parser_create_array_setter(parser_t *parser, ast_value *array, const char *funcname)
4447 {
4448 if (!parser_create_array_setter_proto(parser, array, funcname))
4449 return false;
4450 return parser_create_array_setter_impl(parser, array);
4451 }
4452
parser_create_array_field_setter(parser_t * parser,ast_value * array,const char * funcname)4453 static bool parser_create_array_field_setter(parser_t *parser, ast_value *array, const char *funcname)
4454 {
4455 ast_expression *root = nullptr;
4456 ast_value *entity = nullptr;
4457 ast_value *index = nullptr;
4458 ast_value *value = nullptr;
4459 ast_function *func;
4460 ast_value *fval;
4461
4462 if (!ast_istype(array->m_next, ast_value)) {
4463 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
4464 return false;
4465 }
4466
4467 if (!parser_create_array_accessor(parser, array, funcname, &fval))
4468 return false;
4469 func = fval->m_constval.vfunc;
4470 fval->m_next = new ast_value(array->m_context, "<void>", TYPE_VOID);
4471
4472 entity = new ast_value(array->m_context, "entity", TYPE_ENTITY);
4473 index = new ast_value(array->m_context, "index", TYPE_FLOAT);
4474 value = new ast_value(ast_copy_type, *(ast_value*)array->m_next);
4475 if (!entity || !index || !value) {
4476 parseerror(parser, "failed to create locals for array accessor");
4477 goto cleanup;
4478 }
4479 value->m_name = "value"; // not important
4480 fval->m_type_params.emplace_back(entity);
4481 fval->m_type_params.emplace_back(index);
4482 fval->m_type_params.emplace_back(value);
4483
4484 root = array_field_setter_node(parser, array, entity, index, value, 0, array->m_count);
4485 if (!root) {
4486 parseerror(parser, "failed to build accessor search tree");
4487 goto cleanup;
4488 }
4489
4490 array->m_setter = fval;
4491 return func->m_blocks[0].get()->addExpr(root);
4492 cleanup:
4493 if (entity) delete entity;
4494 if (index) delete index;
4495 if (value) delete value;
4496 if (root) delete root;
4497 delete func;
4498 delete fval;
4499 return false;
4500 }
4501
parser_create_array_getter_proto(parser_t * parser,ast_value * array,const ast_expression * elemtype,const char * funcname)4502 static ast_value* parser_create_array_getter_proto(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname)
4503 {
4504 ast_value *index = nullptr;
4505 ast_value *fval;
4506 ast_function *func;
4507
4508 /* NOTE: checking array->m_next rather than elemtype since
4509 * for fields elemtype is a temporary fieldtype.
4510 */
4511 if (!ast_istype(array->m_next, ast_value)) {
4512 parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type");
4513 return nullptr;
4514 }
4515
4516 if (!parser_create_array_accessor(parser, array, funcname, &fval))
4517 return nullptr;
4518 func = fval->m_constval.vfunc;
4519 fval->m_next = new ast_expression(ast_copy_type, array->m_context, *elemtype);
4520
4521 index = new ast_value(array->m_context, "index", TYPE_FLOAT);
4522
4523 if (!index) {
4524 parseerror(parser, "failed to create locals for array accessor");
4525 goto cleanup;
4526 }
4527 fval->m_type_params.emplace_back(index);
4528
4529 array->m_getter = fval;
4530 return fval;
4531 cleanup:
4532 if (index) delete index;
4533 delete func;
4534 delete fval;
4535 return nullptr;
4536 }
4537
parser_create_array_getter_impl(parser_t * parser,ast_value * array)4538 static bool parser_create_array_getter_impl(parser_t *parser, ast_value *array)
4539 {
4540 ast_expression *root = nullptr;
4541
4542 root = array_getter_node(parser, array, array->m_getter->m_type_params[0].get(), 0, array->m_count);
4543 if (!root) {
4544 parseerror(parser, "failed to build accessor search tree");
4545 return false;
4546 }
4547 if (!array->m_getter->m_constval.vfunc->m_blocks[0].get()->addExpr(root)) {
4548 delete root;
4549 return false;
4550 }
4551 return true;
4552 }
4553
parser_create_array_getter(parser_t * parser,ast_value * array,const ast_expression * elemtype,const char * funcname)4554 static bool parser_create_array_getter(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname)
4555 {
4556 if (!parser_create_array_getter_proto(parser, array, elemtype, funcname))
4557 return false;
4558 return parser_create_array_getter_impl(parser, array);
4559 }
4560
parse_parameter_list(parser_t * parser,ast_value * var)4561 static ast_value *parse_parameter_list(parser_t *parser, ast_value *var)
4562 {
4563 lex_ctx_t ctx = parser_ctx(parser);
4564 std::vector<std::unique_ptr<ast_value>> params;
4565 ast_value *fval;
4566 bool first = true;
4567 bool variadic = false;
4568 ast_value *varparam = nullptr;
4569 char *argcounter = nullptr;
4570
4571 /* for the sake of less code we parse-in in this function */
4572 if (!parser_next(parser)) {
4573 delete var;
4574 parseerror(parser, "expected parameter list");
4575 return nullptr;
4576 }
4577
4578 /* parse variables until we hit a closing paren */
4579 while (parser->tok != ')') {
4580 bool is_varargs = false;
4581
4582 if (!first) {
4583 /* there must be commas between them */
4584 if (parser->tok != ',') {
4585 parseerror(parser, "expected comma or end of parameter list");
4586 goto on_error;
4587 }
4588 if (!parser_next(parser)) {
4589 parseerror(parser, "expected parameter");
4590 goto on_error;
4591 }
4592 }
4593 first = false;
4594
4595 ast_value *param = parse_typename(parser, nullptr, nullptr, &is_varargs);
4596 if (!param && !is_varargs)
4597 goto on_error;
4598 if (is_varargs) {
4599 /* '...' indicates a varargs function */
4600 variadic = true;
4601 if (parser->tok != ')' && parser->tok != TOKEN_IDENT) {
4602 parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
4603 goto on_error;
4604 }
4605 if (parser->tok == TOKEN_IDENT) {
4606 argcounter = util_strdup(parser_tokval(parser));
4607 if (!parser_next(parser) || parser->tok != ')') {
4608 parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
4609 goto on_error;
4610 }
4611 }
4612 } else {
4613 params.emplace_back(param);
4614 if (param->m_vtype >= TYPE_VARIANT) {
4615 char tname[1024]; /* typename is reserved in C++ */
4616 ast_type_to_string(param, tname, sizeof(tname));
4617 parseerror(parser, "type not supported as part of a parameter list: %s", tname);
4618 goto on_error;
4619 }
4620 /* type-restricted varargs */
4621 if (parser->tok == TOKEN_DOTS) {
4622 variadic = true;
4623 varparam = params.back().release();
4624 params.pop_back();
4625 if (!parser_next(parser) || (parser->tok != ')' && parser->tok != TOKEN_IDENT)) {
4626 parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
4627 goto on_error;
4628 }
4629 if (parser->tok == TOKEN_IDENT) {
4630 argcounter = util_strdup(parser_tokval(parser));
4631 param->m_name = argcounter;
4632 if (!parser_next(parser) || parser->tok != ')') {
4633 parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
4634 goto on_error;
4635 }
4636 }
4637 }
4638 if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_FTEQCC && param->m_name[0] == '<') {
4639 parseerror(parser, "parameter name omitted");
4640 goto on_error;
4641 }
4642 }
4643 }
4644
4645 if (params.size() == 1 && params[0]->m_vtype == TYPE_VOID)
4646 params.clear();
4647
4648 /* sanity check */
4649 if (params.size() > 8 && OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC)
4650 (void)!parsewarning(parser, WARN_EXTENSIONS, "more than 8 parameters are not supported by this standard");
4651
4652 /* parse-out */
4653 if (!parser_next(parser)) {
4654 parseerror(parser, "parse error after typename");
4655 goto on_error;
4656 }
4657
4658 /* now turn 'var' into a function type */
4659 fval = new ast_value(ctx, "<type()>", TYPE_FUNCTION);
4660 fval->m_next = var;
4661 if (variadic)
4662 fval->m_flags |= AST_FLAG_VARIADIC;
4663 var = fval;
4664
4665 var->m_type_params = move(params);
4666 var->m_varparam = varparam;
4667 var->m_argcounter = argcounter;
4668
4669 return var;
4670
4671 on_error:
4672 if (argcounter)
4673 mem_d(argcounter);
4674 if (varparam)
4675 delete varparam;
4676 delete var;
4677 return nullptr;
4678 }
4679
parse_arraysize(parser_t * parser,ast_value * var)4680 static ast_value *parse_arraysize(parser_t *parser, ast_value *var)
4681 {
4682 ast_expression *cexp;
4683 ast_value *cval, *tmp;
4684 lex_ctx_t ctx;
4685
4686 ctx = parser_ctx(parser);
4687
4688 if (!parser_next(parser)) {
4689 delete var;
4690 parseerror(parser, "expected array-size");
4691 return nullptr;
4692 }
4693
4694 if (parser->tok != ']') {
4695 cexp = parse_expression_leave(parser, true, false, false);
4696
4697 if (!cexp || !ast_istype(cexp, ast_value)) {
4698 if (cexp)
4699 ast_unref(cexp);
4700 delete var;
4701 parseerror(parser, "expected array-size as constant positive integer");
4702 return nullptr;
4703 }
4704 cval = (ast_value*)cexp;
4705 }
4706 else {
4707 cexp = nullptr;
4708 cval = nullptr;
4709 }
4710
4711 tmp = new ast_value(ctx, "<type[]>", TYPE_ARRAY);
4712 tmp->m_next = var;
4713 var = tmp;
4714
4715 if (cval) {
4716 if (cval->m_vtype == TYPE_INTEGER)
4717 tmp->m_count = cval->m_constval.vint;
4718 else if (cval->m_vtype == TYPE_FLOAT)
4719 tmp->m_count = cval->m_constval.vfloat;
4720 else {
4721 ast_unref(cexp);
4722 delete var;
4723 parseerror(parser, "array-size must be a positive integer constant");
4724 return nullptr;
4725 }
4726
4727 ast_unref(cexp);
4728 } else {
4729 var->m_count = -1;
4730 var->m_flags |= AST_FLAG_ARRAY_INIT;
4731 }
4732
4733 if (parser->tok != ']') {
4734 delete var;
4735 parseerror(parser, "expected ']' after array-size");
4736 return nullptr;
4737 }
4738 if (!parser_next(parser)) {
4739 delete var;
4740 parseerror(parser, "error after parsing array size");
4741 return nullptr;
4742 }
4743 return var;
4744 }
4745
4746 /* Parse a complete typename.
4747 * for single-variables (ie. function parameters or typedefs) storebase should be nullptr
4748 * but when parsing variables separated by comma
4749 * 'storebase' should point to where the base-type should be kept.
4750 * The base type makes up every bit of type information which comes *before* the
4751 * variable name.
4752 *
4753 * NOTE: The value must either be named, have a nullptr name, or a name starting
4754 * with '<'. In the first case, this will be the actual variable or type
4755 * name, in the other cases it is assumed that the name will appear
4756 * later, and an error is generated otherwise.
4757 *
4758 * The following will be parsed in its entirety:
4759 * void() foo()
4760 * The 'basetype' in this case is 'void()'
4761 * and if there's a comma after it, say:
4762 * void() foo(), bar
4763 * then the type-information 'void()' can be stored in 'storebase'
4764 */
parse_typename(parser_t * parser,ast_value ** storebase,ast_value * cached_typedef,bool * is_vararg)4765 static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef, bool *is_vararg)
4766 {
4767 ast_value *var, *tmp;
4768 lex_ctx_t ctx;
4769
4770 const char *name = nullptr;
4771 bool isfield = false;
4772 bool wasarray = false;
4773 size_t morefields = 0;
4774
4775 bool vararg = (parser->tok == TOKEN_DOTS);
4776
4777 ctx = parser_ctx(parser);
4778
4779 /* types may start with a dot */
4780 if (parser->tok == '.' || parser->tok == TOKEN_DOTS) {
4781 isfield = true;
4782 if (parser->tok == TOKEN_DOTS)
4783 morefields += 2;
4784 /* if we parsed a dot we need a typename now */
4785 if (!parser_next(parser)) {
4786 parseerror(parser, "expected typename for field definition");
4787 return nullptr;
4788 }
4789
4790 /* Further dots are handled seperately because they won't be part of the
4791 * basetype
4792 */
4793 while (true) {
4794 if (parser->tok == '.')
4795 ++morefields;
4796 else if (parser->tok == TOKEN_DOTS)
4797 morefields += 3;
4798 else
4799 break;
4800 vararg = false;
4801 if (!parser_next(parser)) {
4802 parseerror(parser, "expected typename for field definition");
4803 return nullptr;
4804 }
4805 }
4806 }
4807 if (parser->tok == TOKEN_IDENT)
4808 cached_typedef = parser_find_typedef(parser, parser_tokval(parser), 0);
4809 if (!cached_typedef && parser->tok != TOKEN_TYPENAME) {
4810 if (vararg && is_vararg) {
4811 *is_vararg = true;
4812 return nullptr;
4813 }
4814 parseerror(parser, "expected typename");
4815 return nullptr;
4816 }
4817
4818 /* generate the basic type value */
4819 if (cached_typedef) {
4820 var = new ast_value(ast_copy_type, *cached_typedef);
4821 var->m_name = "<type(from_def)>";
4822 } else
4823 var = new ast_value(ctx, "<type>", parser_token(parser)->constval.t);
4824
4825 for (; morefields; --morefields) {
4826 tmp = new ast_value(ctx, "<.type>", TYPE_FIELD);
4827 tmp->m_next = var;
4828 var = tmp;
4829 }
4830
4831 /* do not yet turn into a field - remember:
4832 * .void() foo; is a field too
4833 * .void()() foo; is a function
4834 */
4835
4836 /* parse on */
4837 if (!parser_next(parser)) {
4838 delete var;
4839 parseerror(parser, "parse error after typename");
4840 return nullptr;
4841 }
4842
4843 /* an opening paren now starts the parameter-list of a function
4844 * this is where original-QC has parameter lists.
4845 * We allow a single parameter list here.
4846 * Much like fteqcc we don't allow `float()() x`
4847 */
4848 if (parser->tok == '(') {
4849 var = parse_parameter_list(parser, var);
4850 if (!var)
4851 return nullptr;
4852 }
4853
4854 /* store the base if requested */
4855 if (storebase) {
4856 *storebase = new ast_value(ast_copy_type, *var);
4857 if (isfield) {
4858 tmp = new ast_value(ctx, "<type:f>", TYPE_FIELD);
4859 tmp->m_next = *storebase;
4860 *storebase = tmp;
4861 }
4862 }
4863
4864 /* there may be a name now */
4865 if (parser->tok == TOKEN_IDENT || parser->tok == TOKEN_KEYWORD) {
4866 if (!strcmp(parser_tokval(parser), "break"))
4867 (void)!parsewarning(parser, WARN_BREAKDEF, "break definition ignored (suggest removing it)");
4868 else if (parser->tok == TOKEN_KEYWORD)
4869 goto leave;
4870
4871 name = util_strdup(parser_tokval(parser));
4872
4873 /* parse on */
4874 if (!parser_next(parser)) {
4875 delete var;
4876 mem_d(name);
4877 parseerror(parser, "error after variable or field declaration");
4878 return nullptr;
4879 }
4880 }
4881
4882 leave:
4883 /* now this may be an array */
4884 if (parser->tok == '[') {
4885 wasarray = true;
4886 var = parse_arraysize(parser, var);
4887 if (!var) {
4888 if (name) mem_d(name);
4889 return nullptr;
4890 }
4891 }
4892
4893 /* This is the point where we can turn it into a field */
4894 if (isfield) {
4895 /* turn it into a field if desired */
4896 tmp = new ast_value(ctx, "<type:f>", TYPE_FIELD);
4897 tmp->m_next = var;
4898 var = tmp;
4899 }
4900
4901 /* now there may be function parens again */
4902 if (parser->tok == '(' && OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC)
4903 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
4904 if (parser->tok == '(' && wasarray)
4905 parseerror(parser, "arrays as part of a return type is not supported");
4906 while (parser->tok == '(') {
4907 var = parse_parameter_list(parser, var);
4908 if (!var) {
4909 if (name) mem_d(name);
4910 return nullptr;
4911 }
4912 }
4913
4914 /* finally name it */
4915 if (name) {
4916 var->m_name = name;
4917 // free the name, ast_value_set_name duplicates
4918 mem_d(name);
4919 }
4920
4921 return var;
4922 }
4923
parse_typedef(parser_t * parser)4924 static bool parse_typedef(parser_t *parser)
4925 {
4926 ast_value *typevar, *oldtype;
4927 ast_expression *old;
4928
4929 typevar = parse_typename(parser, nullptr, nullptr, nullptr);
4930
4931 if (!typevar)
4932 return false;
4933
4934 // while parsing types, the ast_value's get named '<something>'
4935 if (!typevar->m_name.length() || typevar->m_name[0] == '<') {
4936 parseerror(parser, "missing name in typedef");
4937 delete typevar;
4938 return false;
4939 }
4940
4941 if ( (old = parser_find_var(parser, typevar->m_name)) ) {
4942 parseerror(parser, "cannot define a type with the same name as a variable: %s\n"
4943 " -> `%s` has been declared here: %s:%i",
4944 typevar->m_name, old->m_context.file, old->m_context.line);
4945 delete typevar;
4946 return false;
4947 }
4948
4949 if ( (oldtype = parser_find_typedef(parser, typevar->m_name, vec_last(parser->_blocktypedefs))) ) {
4950 parseerror(parser, "type `%s` has already been declared here: %s:%i",
4951 typevar->m_name, oldtype->m_context.file, oldtype->m_context.line);
4952 delete typevar;
4953 return false;
4954 }
4955
4956 vec_push(parser->_typedefs, typevar);
4957 util_htset(vec_last(parser->typedefs), typevar->m_name.c_str(), typevar);
4958
4959 if (parser->tok != ';') {
4960 parseerror(parser, "expected semicolon after typedef");
4961 return false;
4962 }
4963 if (!parser_next(parser)) {
4964 parseerror(parser, "parse error after typedef");
4965 return false;
4966 }
4967
4968 return true;
4969 }
4970
cvq_to_str(int cvq)4971 static const char *cvq_to_str(int cvq) {
4972 switch (cvq) {
4973 case CV_NONE: return "none";
4974 case CV_VAR: return "`var`";
4975 case CV_CONST: return "`const`";
4976 default: return "<INVALID>";
4977 }
4978 }
4979
parser_check_qualifiers(parser_t * parser,const ast_value * var,const ast_value * proto)4980 static bool parser_check_qualifiers(parser_t *parser, const ast_value *var, const ast_value *proto)
4981 {
4982 bool av, ao;
4983 if (proto->m_cvq != var->m_cvq) {
4984 if (!(proto->m_cvq == CV_CONST && var->m_cvq == CV_NONE &&
4985 !OPTS_FLAG(INITIALIZED_NONCONSTANTS) &&
4986 parser->tok == '='))
4987 {
4988 return !parsewarning(parser, WARN_DIFFERENT_QUALIFIERS,
4989 "`%s` declared with different qualifiers: %s\n"
4990 " -> previous declaration here: %s:%i uses %s",
4991 var->m_name, cvq_to_str(var->m_cvq),
4992 proto->m_context.file, proto->m_context.line,
4993 cvq_to_str(proto->m_cvq));
4994 }
4995 }
4996 av = (var ->m_flags & AST_FLAG_NORETURN);
4997 ao = (proto->m_flags & AST_FLAG_NORETURN);
4998 if (!av != !ao) {
4999 return !parsewarning(parser, WARN_DIFFERENT_ATTRIBUTES,
5000 "`%s` declared with different attributes%s\n"
5001 " -> previous declaration here: %s:%i",
5002 var->m_name, (av ? ": noreturn" : ""),
5003 proto->m_context.file, proto->m_context.line,
5004 (ao ? ": noreturn" : ""));
5005 }
5006 return true;
5007 }
5008
create_array_accessors(parser_t * parser,ast_value * var)5009 static bool create_array_accessors(parser_t *parser, ast_value *var)
5010 {
5011 char name[1024];
5012 util_snprintf(name, sizeof(name), "%s##SET", var->m_name.c_str());
5013 if (!parser_create_array_setter(parser, var, name))
5014 return false;
5015 util_snprintf(name, sizeof(name), "%s##GET", var->m_name.c_str());
5016 if (!parser_create_array_getter(parser, var, var->m_next, name))
5017 return false;
5018 return true;
5019 }
5020
parse_array(parser_t * parser,ast_value * array)5021 static bool parse_array(parser_t *parser, ast_value *array)
5022 {
5023 size_t i;
5024 if (array->m_initlist.size()) {
5025 parseerror(parser, "array already initialized elsewhere");
5026 return false;
5027 }
5028 if (!parser_next(parser)) {
5029 parseerror(parser, "parse error in array initializer");
5030 return false;
5031 }
5032 i = 0;
5033 while (parser->tok != '}') {
5034 ast_value *v = (ast_value*)parse_expression_leave(parser, true, false, false);
5035 if (!v)
5036 return false;
5037 if (!ast_istype(v, ast_value) || !v->m_hasvalue || v->m_cvq != CV_CONST) {
5038 ast_unref(v);
5039 parseerror(parser, "initializing element must be a compile time constant");
5040 return false;
5041 }
5042 array->m_initlist.push_back(v->m_constval);
5043 if (v->m_vtype == TYPE_STRING) {
5044 array->m_initlist[i].vstring = util_strdupe(array->m_initlist[i].vstring);
5045 ++i;
5046 }
5047 ast_unref(v);
5048 if (parser->tok == '}')
5049 break;
5050 if (parser->tok != ',' || !parser_next(parser)) {
5051 parseerror(parser, "expected comma or '}' in element list");
5052 return false;
5053 }
5054 }
5055 if (!parser_next(parser) || parser->tok != ';') {
5056 parseerror(parser, "expected semicolon after initializer, got %s");
5057 return false;
5058 }
5059 /*
5060 if (!parser_next(parser)) {
5061 parseerror(parser, "parse error after initializer");
5062 return false;
5063 }
5064 */
5065
5066 if (array->m_flags & AST_FLAG_ARRAY_INIT) {
5067 if (array->m_count != (size_t)-1) {
5068 parseerror(parser, "array `%s' has already been initialized with %u elements",
5069 array->m_name, (unsigned)array->m_count);
5070 }
5071 array->m_count = array->m_initlist.size();
5072 if (!create_array_accessors(parser, array))
5073 return false;
5074 }
5075 return true;
5076 }
5077
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)5078 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)
5079 {
5080 ast_value *var;
5081 ast_value *proto;
5082 ast_expression *old;
5083 bool was_end;
5084 size_t i;
5085
5086 ast_value *basetype = nullptr;
5087 bool retval = true;
5088 bool isparam = false;
5089 bool isvector = false;
5090 bool cleanvar = true;
5091 bool wasarray = false;
5092
5093 ast_member *me[3] = { nullptr, nullptr, nullptr };
5094 ast_member *last_me[3] = { nullptr, nullptr, nullptr };
5095
5096 if (!localblock && is_static)
5097 parseerror(parser, "`static` qualifier is not supported in global scope");
5098
5099 /* get the first complete variable */
5100 var = parse_typename(parser, &basetype, cached_typedef, nullptr);
5101 if (!var) {
5102 if (basetype)
5103 delete basetype;
5104 return false;
5105 }
5106
5107 /* while parsing types, the ast_value's get named '<something>' */
5108 if (!var->m_name.length() || var->m_name[0] == '<') {
5109 parseerror(parser, "declaration does not declare anything");
5110 if (basetype)
5111 delete basetype;
5112 return false;
5113 }
5114
5115 while (true) {
5116 proto = nullptr;
5117 wasarray = false;
5118
5119 /* Part 0: finish the type */
5120 if (parser->tok == '(') {
5121 if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC)
5122 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
5123 var = parse_parameter_list(parser, var);
5124 if (!var) {
5125 retval = false;
5126 goto cleanup;
5127 }
5128 }
5129 /* we only allow 1-dimensional arrays */
5130 if (parser->tok == '[') {
5131 wasarray = true;
5132 var = parse_arraysize(parser, var);
5133 if (!var) {
5134 retval = false;
5135 goto cleanup;
5136 }
5137 }
5138 if (parser->tok == '(' && wasarray) {
5139 parseerror(parser, "arrays as part of a return type is not supported");
5140 /* we'll still parse the type completely for now */
5141 }
5142 /* for functions returning functions */
5143 while (parser->tok == '(') {
5144 if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC)
5145 parseerror(parser, "C-style function syntax is not allowed in -std=qcc");
5146 var = parse_parameter_list(parser, var);
5147 if (!var) {
5148 retval = false;
5149 goto cleanup;
5150 }
5151 }
5152
5153 var->m_cvq = qualifier;
5154 if (qflags & AST_FLAG_COVERAGE) /* specified in QC, drop our default */
5155 var->m_flags &= ~(AST_FLAG_COVERAGE_MASK);
5156 var->m_flags |= qflags;
5157
5158 /*
5159 * store the vstring back to var for alias and
5160 * deprecation messages.
5161 */
5162 if (var->m_flags & AST_FLAG_DEPRECATED ||
5163 var->m_flags & AST_FLAG_ALIAS)
5164 var->m_desc = vstring;
5165
5166 if (parser_find_global(parser, var->m_name) && var->m_flags & AST_FLAG_ALIAS) {
5167 parseerror(parser, "function aliases cannot be forward declared");
5168 retval = false;
5169 goto cleanup;
5170 }
5171
5172
5173 /* Part 1:
5174 * check for validity: (end_sys_..., multiple-definitions, prototypes, ...)
5175 * Also: if there was a prototype, `var` will be deleted and set to `proto` which
5176 * is then filled with the previous definition and the parameter-names replaced.
5177 */
5178 if (var->m_name == "nil") {
5179 if (OPTS_FLAG(UNTYPED_NIL)) {
5180 if (!localblock || !OPTS_FLAG(PERMISSIVE))
5181 parseerror(parser, "name `nil` not allowed (try -fpermissive)");
5182 } else
5183 (void)!parsewarning(parser, WARN_RESERVED_NAMES, "variable name `nil` is reserved");
5184 }
5185 if (!localblock) {
5186 /* Deal with end_sys_ vars */
5187 was_end = false;
5188 if (var->m_name == "end_sys_globals") {
5189 var->m_uses++;
5190 parser->crc_globals = parser->globals.size();
5191 was_end = true;
5192 }
5193 else if (var->m_name == "end_sys_fields") {
5194 var->m_uses++;
5195 parser->crc_fields = parser->fields.size();
5196 was_end = true;
5197 }
5198 if (was_end && var->m_vtype == TYPE_FIELD) {
5199 if (parsewarning(parser, WARN_END_SYS_FIELDS,
5200 "global '%s' hint should not be a field",
5201 parser_tokval(parser)))
5202 {
5203 retval = false;
5204 goto cleanup;
5205 }
5206 }
5207
5208 if (!nofields && var->m_vtype == TYPE_FIELD)
5209 {
5210 /* deal with field declarations */
5211 old = parser_find_field(parser, var->m_name);
5212 if (old) {
5213 if (parsewarning(parser, WARN_FIELD_REDECLARED, "field `%s` already declared here: %s:%i",
5214 var->m_name, old->m_context.file, (int)old->m_context.line))
5215 {
5216 retval = false;
5217 goto cleanup;
5218 }
5219 delete var;
5220 var = nullptr;
5221 goto skipvar;
5222 /*
5223 parseerror(parser, "field `%s` already declared here: %s:%i",
5224 var->m_name, old->m_context.file, old->m_context.line);
5225 retval = false;
5226 goto cleanup;
5227 */
5228 }
5229 if ((OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC || OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_FTEQCC) &&
5230 (old = parser_find_global(parser, var->m_name)))
5231 {
5232 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
5233 parseerror(parser, "field `%s` already declared here: %s:%i",
5234 var->m_name, old->m_context.file, old->m_context.line);
5235 retval = false;
5236 goto cleanup;
5237 }
5238 }
5239 else
5240 {
5241 /* deal with other globals */
5242 old = parser_find_global(parser, var->m_name);
5243 if (old && var->m_vtype == TYPE_FUNCTION && old->m_vtype == TYPE_FUNCTION)
5244 {
5245 /* This is a function which had a prototype */
5246 if (!ast_istype(old, ast_value)) {
5247 parseerror(parser, "internal error: prototype is not an ast_value");
5248 retval = false;
5249 goto cleanup;
5250 }
5251 proto = (ast_value*)old;
5252 proto->m_desc = var->m_desc;
5253 if (!proto->compareType(*var)) {
5254 parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i",
5255 proto->m_name,
5256 proto->m_context.file, proto->m_context.line);
5257 retval = false;
5258 goto cleanup;
5259 }
5260 /* we need the new parameter-names */
5261 for (i = 0; i < proto->m_type_params.size(); ++i)
5262 proto->m_type_params[i]->m_name = var->m_type_params[i]->m_name;
5263 if (!parser_check_qualifiers(parser, var, proto)) {
5264 retval = false;
5265 proto = nullptr;
5266 goto cleanup;
5267 }
5268 proto->m_flags |= var->m_flags;
5269 delete var;
5270 var = proto;
5271 }
5272 else
5273 {
5274 /* other globals */
5275 if (old) {
5276 if (parsewarning(parser, WARN_DOUBLE_DECLARATION,
5277 "global `%s` already declared here: %s:%i",
5278 var->m_name, old->m_context.file, old->m_context.line))
5279 {
5280 retval = false;
5281 goto cleanup;
5282 }
5283 if (old->m_flags & AST_FLAG_FINAL_DECL) {
5284 parseerror(parser, "cannot redeclare variable `%s`, declared final here: %s:%i",
5285 var->m_name, old->m_context.file, old->m_context.line);
5286 retval = false;
5287 goto cleanup;
5288 }
5289 proto = (ast_value*)old;
5290 if (!ast_istype(old, ast_value)) {
5291 parseerror(parser, "internal error: not an ast_value");
5292 retval = false;
5293 proto = nullptr;
5294 goto cleanup;
5295 }
5296 if (!parser_check_qualifiers(parser, var, proto)) {
5297 retval = false;
5298 proto = nullptr;
5299 goto cleanup;
5300 }
5301 proto->m_flags |= var->m_flags;
5302 /* copy the context for finals,
5303 * so the error can show where it was actually made 'final'
5304 */
5305 if (proto->m_flags & AST_FLAG_FINAL_DECL)
5306 old->m_context = var->m_context;
5307 delete var;
5308 var = proto;
5309 }
5310 if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC &&
5311 (old = parser_find_field(parser, var->m_name)))
5312 {
5313 parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc");
5314 parseerror(parser, "global `%s` already declared here: %s:%i",
5315 var->m_name, old->m_context.file, old->m_context.line);
5316 retval = false;
5317 goto cleanup;
5318 }
5319 }
5320 }
5321 }
5322 else /* it's not a global */
5323 {
5324 old = parser_find_local(parser, var->m_name, vec_size(parser->variables)-1, &isparam);
5325 if (old && !isparam) {
5326 parseerror(parser, "local `%s` already declared here: %s:%i",
5327 var->m_name, old->m_context.file, (int)old->m_context.line);
5328 retval = false;
5329 goto cleanup;
5330 }
5331 /* doing this here as the above is just for a single scope */
5332 old = parser_find_local(parser, var->m_name, 0, &isparam);
5333 if (old && isparam) {
5334 if (parsewarning(parser, WARN_LOCAL_SHADOWS,
5335 "local `%s` is shadowing a parameter", var->m_name))
5336 {
5337 parseerror(parser, "local `%s` already declared here: %s:%i",
5338 var->m_name, old->m_context.file, (int)old->m_context.line);
5339 retval = false;
5340 goto cleanup;
5341 }
5342 if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_GMQCC) {
5343 delete var;
5344 if (ast_istype(old, ast_value))
5345 var = proto = (ast_value*)old;
5346 else {
5347 var = nullptr;
5348 goto skipvar;
5349 }
5350 }
5351 }
5352 }
5353
5354 /* in a noref section we simply bump the usecount */
5355 if (noref || parser->noref)
5356 var->m_uses++;
5357
5358 /* Part 2:
5359 * Create the global/local, and deal with vector types.
5360 */
5361 if (!proto) {
5362 if (var->m_vtype == TYPE_VECTOR)
5363 isvector = true;
5364 else if (var->m_vtype == TYPE_FIELD &&
5365 var->m_next->m_vtype == TYPE_VECTOR)
5366 isvector = true;
5367
5368 if (isvector) {
5369 if (!create_vector_members(var, me)) {
5370 retval = false;
5371 goto cleanup;
5372 }
5373 }
5374
5375 if (!localblock) {
5376 /* deal with global variables, fields, functions */
5377 if (!nofields && var->m_vtype == TYPE_FIELD && parser->tok != '=') {
5378 var->m_isfield = true;
5379 parser->fields.push_back(var);
5380 util_htset(parser->htfields, var->m_name.c_str(), var);
5381 if (isvector) {
5382 for (i = 0; i < 3; ++i) {
5383 parser->fields.push_back(me[i]);
5384 util_htset(parser->htfields, me[i]->m_name.c_str(), me[i]);
5385 }
5386 }
5387 }
5388 else {
5389 if (!(var->m_flags & AST_FLAG_ALIAS)) {
5390 parser_addglobal(parser, var->m_name, var);
5391 if (isvector) {
5392 for (i = 0; i < 3; ++i) {
5393 parser_addglobal(parser, me[i]->m_name.c_str(), me[i]);
5394 }
5395 }
5396 } else {
5397 ast_expression *find = parser_find_global(parser, var->m_desc);
5398
5399 if (!find) {
5400 compile_error(parser_ctx(parser), "undeclared variable `%s` for alias `%s`", var->m_desc, var->m_name);
5401 return false;
5402 }
5403
5404 if (!var->compareType(*find)) {
5405 char ty1[1024];
5406 char ty2[1024];
5407
5408 ast_type_to_string(find, ty1, sizeof(ty1));
5409 ast_type_to_string(var, ty2, sizeof(ty2));
5410
5411 compile_error(parser_ctx(parser), "incompatible types `%s` and `%s` for alias `%s`",
5412 ty1, ty2, var->m_name
5413 );
5414 return false;
5415 }
5416
5417 util_htset(parser->aliases, var->m_name.c_str(), find);
5418
5419 /* generate aliases for vector components */
5420 if (isvector) {
5421 char *buffer[3];
5422
5423 util_asprintf(&buffer[0], "%s_x", var->m_desc.c_str());
5424 util_asprintf(&buffer[1], "%s_y", var->m_desc.c_str());
5425 util_asprintf(&buffer[2], "%s_z", var->m_desc.c_str());
5426
5427 util_htset(parser->aliases, me[0]->m_name.c_str(), parser_find_global(parser, buffer[0]));
5428 util_htset(parser->aliases, me[1]->m_name.c_str(), parser_find_global(parser, buffer[1]));
5429 util_htset(parser->aliases, me[2]->m_name.c_str(), parser_find_global(parser, buffer[2]));
5430
5431 mem_d(buffer[0]);
5432 mem_d(buffer[1]);
5433 mem_d(buffer[2]);
5434 }
5435 }
5436 }
5437 } else {
5438 if (is_static) {
5439 // a static adds itself to be generated like any other global
5440 // but is added to the local namespace instead
5441 std::string defname;
5442 size_t prefix_len;
5443 size_t sn, sn_size;
5444
5445 defname = parser->function->m_name;
5446 defname.append(2, ':');
5447
5448 // remember the length up to here
5449 prefix_len = defname.length();
5450
5451 // Add it to the local scope
5452 util_htset(vec_last(parser->variables), var->m_name.c_str(), (void*)var);
5453
5454 // now rename the global
5455 defname.append(var->m_name);
5456 // if a variable of that name already existed, add the
5457 // counter value.
5458 // The counter is incremented either way.
5459 sn_size = parser->function->m_static_names.size();
5460 for (sn = 0; sn != sn_size; ++sn) {
5461 if (parser->function->m_static_names[sn] == var->m_name.c_str())
5462 break;
5463 }
5464 if (sn != sn_size) {
5465 char *num = nullptr;
5466 int len = util_asprintf(&num, "#%u", parser->function->m_static_count);
5467 defname.append(num, 0, len);
5468 mem_d(num);
5469 }
5470 else
5471 parser->function->m_static_names.emplace_back(var->m_name);
5472 parser->function->m_static_count++;
5473 var->m_name = defname;
5474
5475 // push it to the to-be-generated globals
5476 parser->globals.push_back(var);
5477
5478 // same game for the vector members
5479 if (isvector) {
5480 defname.erase(prefix_len);
5481 for (i = 0; i < 3; ++i) {
5482 util_htset(vec_last(parser->variables), me[i]->m_name.c_str(), (void*)(me[i]));
5483 me[i]->m_name = move(defname + me[i]->m_name);
5484 parser->globals.push_back(me[i]);
5485 }
5486 }
5487 } else {
5488 localblock->m_locals.push_back(var);
5489 parser_addlocal(parser, var->m_name, var);
5490 if (isvector) {
5491 for (i = 0; i < 3; ++i) {
5492 parser_addlocal(parser, me[i]->m_name, me[i]);
5493 localblock->collect(me[i]);
5494 }
5495 }
5496 }
5497 }
5498 }
5499 memcpy(last_me, me, sizeof(me));
5500 me[0] = me[1] = me[2] = nullptr;
5501 cleanvar = false;
5502 /* Part 2.2
5503 * deal with arrays
5504 */
5505 if (var->m_vtype == TYPE_ARRAY) {
5506 if (var->m_count != (size_t)-1) {
5507 if (!create_array_accessors(parser, var))
5508 goto cleanup;
5509 }
5510 }
5511 else if (!localblock && !nofields &&
5512 var->m_vtype == TYPE_FIELD &&
5513 var->m_next->m_vtype == TYPE_ARRAY)
5514 {
5515 char name[1024];
5516 ast_expression *telem;
5517 ast_value *tfield;
5518 ast_value *array = (ast_value*)var->m_next;
5519
5520 if (!ast_istype(var->m_next, ast_value)) {
5521 parseerror(parser, "internal error: field element type must be an ast_value");
5522 goto cleanup;
5523 }
5524
5525 util_snprintf(name, sizeof(name), "%s##SETF", var->m_name.c_str());
5526 if (!parser_create_array_field_setter(parser, array, name))
5527 goto cleanup;
5528
5529 telem = new ast_expression(ast_copy_type, var->m_context, *array->m_next);
5530 tfield = new ast_value(var->m_context, "<.type>", TYPE_FIELD);
5531 tfield->m_next = telem;
5532 util_snprintf(name, sizeof(name), "%s##GETFP", var->m_name.c_str());
5533 if (!parser_create_array_getter(parser, array, tfield, name)) {
5534 delete tfield;
5535 goto cleanup;
5536 }
5537 delete tfield;
5538 }
5539
5540 skipvar:
5541 if (parser->tok == ';') {
5542 delete basetype;
5543 if (!parser_next(parser)) {
5544 parseerror(parser, "error after variable declaration");
5545 return false;
5546 }
5547 return true;
5548 }
5549
5550 if (parser->tok == ',')
5551 goto another;
5552
5553 /*
5554 if (!var || (!localblock && !nofields && basetype->m_vtype == TYPE_FIELD)) {
5555 */
5556 if (!var) {
5557 parseerror(parser, "missing comma or semicolon while parsing variables");
5558 break;
5559 }
5560
5561 if (localblock && OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) {
5562 if (parsewarning(parser, WARN_LOCAL_CONSTANTS,
5563 "initializing expression turns variable `%s` into a constant in this standard",
5564 var->m_name) )
5565 {
5566 break;
5567 }
5568 }
5569
5570 if (parser->tok != '{' || var->m_vtype != TYPE_FUNCTION) {
5571 if (parser->tok != '=') {
5572 parseerror(parser, "missing semicolon or initializer, got: `%s`", parser_tokval(parser));
5573 break;
5574 }
5575
5576 if (!parser_next(parser)) {
5577 parseerror(parser, "error parsing initializer");
5578 break;
5579 }
5580 }
5581 else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) {
5582 parseerror(parser, "expected '=' before function body in this standard");
5583 }
5584
5585 if (parser->tok == '#') {
5586 ast_function *func = nullptr;
5587 ast_value *number = nullptr;
5588 float fractional;
5589 float integral;
5590 int builtin_num;
5591
5592 if (localblock) {
5593 parseerror(parser, "cannot declare builtins within functions");
5594 break;
5595 }
5596 if (var->m_vtype != TYPE_FUNCTION) {
5597 parseerror(parser, "unexpected builtin number, '%s' is not a function", var->m_name);
5598 break;
5599 }
5600 if (!parser_next(parser)) {
5601 parseerror(parser, "expected builtin number");
5602 break;
5603 }
5604
5605 if (OPTS_FLAG(EXPRESSIONS_FOR_BUILTINS)) {
5606 number = (ast_value*)parse_expression_leave(parser, true, false, false);
5607 if (!number) {
5608 parseerror(parser, "builtin number expected");
5609 break;
5610 }
5611 if (!ast_istype(number, ast_value) || !number->m_hasvalue || number->m_cvq != CV_CONST)
5612 {
5613 ast_unref(number);
5614 parseerror(parser, "builtin number must be a compile time constant");
5615 break;
5616 }
5617 if (number->m_vtype == TYPE_INTEGER)
5618 builtin_num = number->m_constval.vint;
5619 else if (number->m_vtype == TYPE_FLOAT)
5620 builtin_num = number->m_constval.vfloat;
5621 else {
5622 ast_unref(number);
5623 parseerror(parser, "builtin number must be an integer constant");
5624 break;
5625 }
5626 ast_unref(number);
5627
5628 fractional = modff(builtin_num, &integral);
5629 if (builtin_num < 0 || fractional != 0) {
5630 parseerror(parser, "builtin number must be an integer greater than zero");
5631 break;
5632 }
5633
5634 /* we only want the integral part anyways */
5635 builtin_num = integral;
5636 } else if (parser->tok == TOKEN_INTCONST) {
5637 builtin_num = parser_token(parser)->constval.i;
5638 } else {
5639 parseerror(parser, "builtin number must be a compile time constant");
5640 break;
5641 }
5642
5643 if (var->m_hasvalue) {
5644 (void)!parsewarning(parser, WARN_DOUBLE_DECLARATION,
5645 "builtin `%s` has already been defined\n"
5646 " -> previous declaration here: %s:%i",
5647 var->m_name, var->m_context.file, (int)var->m_context.line);
5648 }
5649 else
5650 {
5651 func = ast_function::make(var->m_context, var->m_name, var);
5652 if (!func) {
5653 parseerror(parser, "failed to allocate function for `%s`", var->m_name);
5654 break;
5655 }
5656 parser->functions.push_back(func);
5657
5658 func->m_builtin = -builtin_num-1;
5659 }
5660
5661 if (OPTS_FLAG(EXPRESSIONS_FOR_BUILTINS)
5662 ? (parser->tok != ',' && parser->tok != ';')
5663 : (!parser_next(parser)))
5664 {
5665 parseerror(parser, "expected comma or semicolon");
5666 delete func;
5667 var->m_constval.vfunc = nullptr;
5668 break;
5669 }
5670 }
5671 else if (var->m_vtype == TYPE_ARRAY && parser->tok == '{')
5672 {
5673 if (localblock) {
5674 /* Note that fteqcc and most others don't even *have*
5675 * local arrays, so this is not a high priority.
5676 */
5677 parseerror(parser, "TODO: initializers for local arrays");
5678 break;
5679 }
5680
5681 var->m_hasvalue = true;
5682 if (!parse_array(parser, var))
5683 break;
5684 }
5685 else if (var->m_vtype == TYPE_FUNCTION && (parser->tok == '{' || parser->tok == '['))
5686 {
5687 if (localblock) {
5688 parseerror(parser, "cannot declare functions within functions");
5689 break;
5690 }
5691
5692 if (proto)
5693 proto->m_context = parser_ctx(parser);
5694
5695 if (!parse_function_body(parser, var))
5696 break;
5697 delete basetype;
5698 for (auto &it : parser->gotos)
5699 parseerror(parser, "undefined label: `%s`", it->m_name);
5700 parser->gotos.clear();
5701 parser->labels.clear();
5702 return true;
5703 } else {
5704 ast_expression *cexp;
5705 ast_value *cval;
5706 bool folded_const = false;
5707
5708 cexp = parse_expression_leave(parser, true, false, false);
5709 if (!cexp)
5710 break;
5711 cval = ast_istype(cexp, ast_value) ? (ast_value*)cexp : nullptr;
5712
5713 /* deal with foldable constants: */
5714 if (localblock &&
5715 var->m_cvq == CV_CONST && cval && cval->m_hasvalue && cval->m_cvq == CV_CONST && !cval->m_isfield)
5716 {
5717 /* remove it from the current locals */
5718 if (isvector) {
5719 for (i = 0; i < 3; ++i) {
5720 vec_pop(parser->_locals);
5721 localblock->m_collect.pop_back();
5722 }
5723 }
5724 /* do sanity checking, this function really needs refactoring */
5725 if (vec_last(parser->_locals) != var)
5726 parseerror(parser, "internal error: unexpected change in local variable handling");
5727 else
5728 vec_pop(parser->_locals);
5729 if (localblock->m_locals.back() != var)
5730 parseerror(parser, "internal error: unexpected change in local variable handling (2)");
5731 else
5732 localblock->m_locals.pop_back();
5733 /* push it to the to-be-generated globals */
5734 parser->globals.push_back(var);
5735 if (isvector)
5736 for (i = 0; i < 3; ++i)
5737 parser->globals.push_back(last_me[i]);
5738 folded_const = true;
5739 }
5740
5741 if (folded_const || !localblock || is_static) {
5742 if (cval != parser->nil &&
5743 (!cval || ((!cval->m_hasvalue || cval->m_cvq != CV_CONST) && !cval->m_isfield))
5744 )
5745 {
5746 parseerror(parser, "initializer is non constant");
5747 }
5748 else
5749 {
5750 if (!is_static &&
5751 !OPTS_FLAG(INITIALIZED_NONCONSTANTS) &&
5752 qualifier != CV_VAR)
5753 {
5754 var->m_cvq = CV_CONST;
5755 }
5756 if (cval == parser->nil)
5757 var->m_flags |= AST_FLAG_INITIALIZED;
5758 else
5759 {
5760 var->m_hasvalue = true;
5761 if (cval->m_vtype == TYPE_STRING)
5762 var->m_constval.vstring = parser_strdup(cval->m_constval.vstring);
5763 else if (cval->m_vtype == TYPE_FIELD)
5764 var->m_constval.vfield = cval;
5765 else
5766 memcpy(&var->m_constval, &cval->m_constval, sizeof(var->m_constval));
5767 ast_unref(cval);
5768 }
5769 }
5770 } else {
5771 int cvq;
5772 shunt sy;
5773 cvq = var->m_cvq;
5774 var->m_cvq = CV_NONE;
5775 sy.out.push_back(syexp(var->m_context, var));
5776 sy.out.push_back(syexp(cexp->m_context, cexp));
5777 sy.ops.push_back(syop(var->m_context, parser->assign_op));
5778 if (!parser_sy_apply_operator(parser, &sy))
5779 ast_unref(cexp);
5780 else {
5781 if (sy.out.size() != 1 && sy.ops.size() != 0)
5782 parseerror(parser, "internal error: leaked operands");
5783 if (!localblock->addExpr(sy.out[0].out))
5784 break;
5785 }
5786 var->m_cvq = cvq;
5787 }
5788 /* a constant initialized to an inexact value should be marked inexact:
5789 * const float x = <inexact>; should propagate the inexact flag
5790 */
5791 if (var->m_cvq == CV_CONST && var->m_vtype == TYPE_FLOAT) {
5792 if (cval && cval->m_hasvalue && cval->m_cvq == CV_CONST)
5793 var->m_inexact = cval->m_inexact;
5794 }
5795 }
5796
5797 another:
5798 if (parser->tok == ',') {
5799 if (!parser_next(parser)) {
5800 parseerror(parser, "expected another variable");
5801 break;
5802 }
5803
5804 if (parser->tok != TOKEN_IDENT) {
5805 parseerror(parser, "expected another variable");
5806 break;
5807 }
5808 var = new ast_value(ast_copy_type, *basetype);
5809 cleanvar = true;
5810 var->m_name = parser_tokval(parser);
5811 if (!parser_next(parser)) {
5812 parseerror(parser, "error parsing variable declaration");
5813 break;
5814 }
5815 continue;
5816 }
5817
5818 if (parser->tok != ';') {
5819 parseerror(parser, "missing semicolon after variables");
5820 break;
5821 }
5822
5823 if (!parser_next(parser)) {
5824 parseerror(parser, "parse error after variable declaration");
5825 break;
5826 }
5827
5828 delete basetype;
5829 return true;
5830 }
5831
5832 if (cleanvar && var)
5833 delete var;
5834 delete basetype;
5835 return false;
5836
5837 cleanup:
5838 delete basetype;
5839 if (cleanvar && var)
5840 delete var;
5841 delete me[0];
5842 delete me[1];
5843 delete me[2];
5844 return retval;
5845 }
5846
parser_global_statement(parser_t * parser)5847 static bool parser_global_statement(parser_t *parser)
5848 {
5849 int cvq = CV_WRONG;
5850 bool noref = false;
5851 bool is_static = false;
5852 uint32_t qflags = 0;
5853 ast_value *istype = nullptr;
5854 char *vstring = nullptr;
5855
5856 if (parser->tok == TOKEN_IDENT)
5857 istype = parser_find_typedef(parser, parser_tokval(parser), 0);
5858
5859 if (istype || parser->tok == TOKEN_TYPENAME || parser->tok == '.' || parser->tok == TOKEN_DOTS)
5860 {
5861 return parse_variable(parser, nullptr, false, CV_NONE, istype, false, false, 0, nullptr);
5862 }
5863 else if (parse_qualifiers(parser, false, &cvq, &noref, &is_static, &qflags, &vstring))
5864 {
5865 if (cvq == CV_WRONG)
5866 return false;
5867 return parse_variable(parser, nullptr, false, cvq, nullptr, noref, is_static, qflags, vstring);
5868 }
5869 else if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "enum"))
5870 {
5871 return parse_enum(parser);
5872 }
5873 else if (parser->tok == TOKEN_KEYWORD)
5874 {
5875 if (!strcmp(parser_tokval(parser), "typedef")) {
5876 if (!parser_next(parser)) {
5877 parseerror(parser, "expected type definition after 'typedef'");
5878 return false;
5879 }
5880 return parse_typedef(parser);
5881 }
5882 parseerror(parser, "unrecognized keyword `%s`", parser_tokval(parser));
5883 return false;
5884 }
5885 else if (parser->tok == '#')
5886 {
5887 return parse_pragma(parser);
5888 }
5889 else if (parser->tok == '$')
5890 {
5891 if (!parser_next(parser)) {
5892 parseerror(parser, "parse error");
5893 return false;
5894 }
5895 }
5896 else
5897 {
5898 parseerror(parser, "unexpected token: `%s`", parser->lex->tok.value);
5899 return false;
5900 }
5901 return true;
5902 }
5903
progdefs_crc_sum(uint16_t old,const char * str)5904 static uint16_t progdefs_crc_sum(uint16_t old, const char *str)
5905 {
5906 return util_crc16(old, str, strlen(str));
5907 }
5908
progdefs_crc_file(const char * str)5909 static void progdefs_crc_file(const char *str)
5910 {
5911 /* write to progdefs.h here */
5912 (void)str;
5913 }
5914
progdefs_crc_both(uint16_t old,const char * str)5915 static uint16_t progdefs_crc_both(uint16_t old, const char *str)
5916 {
5917 old = progdefs_crc_sum(old, str);
5918 progdefs_crc_file(str);
5919 return old;
5920 }
5921
generate_checksum(parser_t * parser,ir_builder * ir)5922 static void generate_checksum(parser_t *parser, ir_builder *ir)
5923 {
5924 uint16_t crc = 0xFFFF;
5925 size_t i;
5926 ast_value *value;
5927
5928 crc = progdefs_crc_both(crc, "\n/* file generated by qcc, do not modify */\n\ntypedef struct\n{");
5929 crc = progdefs_crc_sum(crc, "\tint\tpad[28];\n");
5930 /*
5931 progdefs_crc_file("\tint\tpad;\n");
5932 progdefs_crc_file("\tint\tofs_return[3];\n");
5933 progdefs_crc_file("\tint\tofs_parm0[3];\n");
5934 progdefs_crc_file("\tint\tofs_parm1[3];\n");
5935 progdefs_crc_file("\tint\tofs_parm2[3];\n");
5936 progdefs_crc_file("\tint\tofs_parm3[3];\n");
5937 progdefs_crc_file("\tint\tofs_parm4[3];\n");
5938 progdefs_crc_file("\tint\tofs_parm5[3];\n");
5939 progdefs_crc_file("\tint\tofs_parm6[3];\n");
5940 progdefs_crc_file("\tint\tofs_parm7[3];\n");
5941 */
5942 for (i = 0; i < parser->crc_globals; ++i) {
5943 if (!ast_istype(parser->globals[i], ast_value))
5944 continue;
5945 value = (ast_value*)(parser->globals[i]);
5946 switch (value->m_vtype) {
5947 case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break;
5948 case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break;
5949 case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break;
5950 case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break;
5951 default:
5952 crc = progdefs_crc_both(crc, "\tint\t");
5953 break;
5954 }
5955 crc = progdefs_crc_both(crc, value->m_name.c_str());
5956 crc = progdefs_crc_both(crc, ";\n");
5957 }
5958 crc = progdefs_crc_both(crc, "} globalvars_t;\n\ntypedef struct\n{\n");
5959 for (i = 0; i < parser->crc_fields; ++i) {
5960 if (!ast_istype(parser->fields[i], ast_value))
5961 continue;
5962 value = (ast_value*)(parser->fields[i]);
5963 switch (value->m_next->m_vtype) {
5964 case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break;
5965 case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break;
5966 case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break;
5967 case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break;
5968 default:
5969 crc = progdefs_crc_both(crc, "\tint\t");
5970 break;
5971 }
5972 crc = progdefs_crc_both(crc, value->m_name.c_str());
5973 crc = progdefs_crc_both(crc, ";\n");
5974 }
5975 crc = progdefs_crc_both(crc, "} entvars_t;\n\n");
5976 ir->m_code->crc = crc;
5977 }
5978
parser_create()5979 parser_t *parser_create()
5980 {
5981 parser_t *parser;
5982 lex_ctx_t empty_ctx;
5983 size_t i;
5984
5985 parser = (parser_t*)mem_a(sizeof(parser_t));
5986 if (!parser)
5987 return nullptr;
5988
5989 memset(parser, 0, sizeof(*parser));
5990
5991 // TODO: remove
5992 new (parser) parser_t();
5993
5994 for (i = 0; i < operator_count; ++i) {
5995 if (operators[i].id == opid1('=')) {
5996 parser->assign_op = operators+i;
5997 break;
5998 }
5999 }
6000 if (!parser->assign_op) {
6001 con_err("internal error: initializing parser: failed to find assign operator\n");
6002 mem_d(parser);
6003 return nullptr;
6004 }
6005
6006 vec_push(parser->variables, parser->htfields = util_htnew(PARSER_HT_SIZE));
6007 vec_push(parser->variables, parser->htglobals = util_htnew(PARSER_HT_SIZE));
6008 vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE));
6009 vec_push(parser->_blocktypedefs, 0);
6010
6011 parser->aliases = util_htnew(PARSER_HT_SIZE);
6012
6013 empty_ctx.file = "<internal>";
6014 empty_ctx.line = 0;
6015 empty_ctx.column = 0;
6016 parser->nil = new ast_value(empty_ctx, "nil", TYPE_NIL);
6017 parser->nil->m_cvq = CV_CONST;
6018 if (OPTS_FLAG(UNTYPED_NIL))
6019 util_htset(parser->htglobals, "nil", (void*)parser->nil);
6020
6021 parser->max_param_count = 1;
6022
6023 parser->const_vec[0] = new ast_value(empty_ctx, "<vector.x>", TYPE_NOEXPR);
6024 parser->const_vec[1] = new ast_value(empty_ctx, "<vector.y>", TYPE_NOEXPR);
6025 parser->const_vec[2] = new ast_value(empty_ctx, "<vector.z>", TYPE_NOEXPR);
6026
6027 if (OPTS_OPTION_BOOL(OPTION_ADD_INFO)) {
6028 parser->reserved_version = new ast_value(empty_ctx, "reserved:version", TYPE_STRING);
6029 parser->reserved_version->m_cvq = CV_CONST;
6030 parser->reserved_version->m_hasvalue = true;
6031 parser->reserved_version->m_flags |= AST_FLAG_INCLUDE_DEF;
6032 parser->reserved_version->m_constval.vstring = util_strdup(GMQCC_FULL_VERSION_STRING);
6033 } else {
6034 parser->reserved_version = nullptr;
6035 }
6036
6037 parser->m_fold = fold(parser);
6038 parser->m_intrin = intrin(parser);
6039 return parser;
6040 }
6041
parser_compile(parser_t * parser)6042 static bool parser_compile(parser_t *parser)
6043 {
6044 /* initial lexer/parser state */
6045 parser->lex->flags.noops = true;
6046
6047 if (parser_next(parser))
6048 {
6049 while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR)
6050 {
6051 if (!parser_global_statement(parser)) {
6052 if (parser->tok == TOKEN_EOF)
6053 parseerror(parser, "unexpected end of file");
6054 else if (compile_errors)
6055 parseerror(parser, "there have been errors, bailing out");
6056 lex_close(parser->lex);
6057 parser->lex = nullptr;
6058 return false;
6059 }
6060 }
6061 } else {
6062 parseerror(parser, "parse error");
6063 lex_close(parser->lex);
6064 parser->lex = nullptr;
6065 return false;
6066 }
6067
6068 lex_close(parser->lex);
6069 parser->lex = nullptr;
6070
6071 return !compile_errors;
6072 }
6073
parser_compile_file(parser_t * parser,const char * filename)6074 bool parser_compile_file(parser_t *parser, const char *filename)
6075 {
6076 parser->lex = lex_open(filename);
6077 if (!parser->lex) {
6078 con_err("failed to open file \"%s\"\n", filename);
6079 return false;
6080 }
6081 return parser_compile(parser);
6082 }
6083
parser_compile_string(parser_t * parser,const char * name,const char * str,size_t len)6084 bool parser_compile_string(parser_t *parser, const char *name, const char *str, size_t len)
6085 {
6086 parser->lex = lex_open_string(str, len, name);
6087 if (!parser->lex) {
6088 con_err("failed to create lexer for string \"%s\"\n", name);
6089 return false;
6090 }
6091 return parser_compile(parser);
6092 }
6093
parser_remove_ast(parser_t * parser)6094 static void parser_remove_ast(parser_t *parser)
6095 {
6096 size_t i;
6097 if (parser->ast_cleaned)
6098 return;
6099 parser->ast_cleaned = true;
6100 for (auto &it : parser->accessors) {
6101 delete it->m_constval.vfunc;
6102 it->m_constval.vfunc = nullptr;
6103 delete it;
6104 }
6105 for (auto &it : parser->functions) delete it;
6106 for (auto &it : parser->globals) delete it;
6107 for (auto &it : parser->fields) delete it;
6108
6109 for (i = 0; i < vec_size(parser->variables); ++i)
6110 util_htdel(parser->variables[i]);
6111 vec_free(parser->variables);
6112 vec_free(parser->_blocklocals);
6113 vec_free(parser->_locals);
6114
6115 for (i = 0; i < vec_size(parser->_typedefs); ++i)
6116 delete parser->_typedefs[i];
6117 vec_free(parser->_typedefs);
6118 for (i = 0; i < vec_size(parser->typedefs); ++i)
6119 util_htdel(parser->typedefs[i]);
6120 vec_free(parser->typedefs);
6121 vec_free(parser->_blocktypedefs);
6122
6123 vec_free(parser->_block_ctx);
6124
6125 delete parser->nil;
6126
6127 delete parser->const_vec[0];
6128 delete parser->const_vec[1];
6129 delete parser->const_vec[2];
6130
6131 if (parser->reserved_version)
6132 delete parser->reserved_version;
6133
6134 util_htdel(parser->aliases);
6135 }
6136
parser_cleanup(parser_t * parser)6137 void parser_cleanup(parser_t *parser)
6138 {
6139 parser_remove_ast(parser);
6140 parser->~parser_t();
6141 mem_d(parser);
6142 }
6143
parser_set_coverage_func(parser_t * parser,ir_builder * ir)6144 static bool parser_set_coverage_func(parser_t *parser, ir_builder *ir) {
6145 ast_expression *expr;
6146 ast_value *cov;
6147 ast_function *func;
6148
6149 if (!OPTS_OPTION_BOOL(OPTION_COVERAGE))
6150 return true;
6151
6152 func = nullptr;
6153 for (auto &it : parser->functions) {
6154 if (it->m_name == "coverage") {
6155 func = it;
6156 break;
6157 }
6158 }
6159 if (!func) {
6160 if (OPTS_OPTION_BOOL(OPTION_COVERAGE)) {
6161 con_out("coverage support requested but no coverage() builtin declared\n");
6162 delete ir;
6163 return false;
6164 }
6165 return true;
6166 }
6167
6168 cov = func->m_function_type;
6169 expr = cov;
6170
6171 if (expr->m_vtype != TYPE_FUNCTION || expr->m_type_params.size()) {
6172 char ty[1024];
6173 ast_type_to_string(expr, ty, sizeof(ty));
6174 con_out("invalid type for coverage(): %s\n", ty);
6175 delete ir;
6176 return false;
6177 }
6178
6179 ir->m_coverage_func = func->m_ir_func->m_value;
6180 return true;
6181 }
6182
parser_finish(parser_t * parser,const char * output)6183 bool parser_finish(parser_t *parser, const char *output)
6184 {
6185 ir_builder *ir;
6186 bool retval = true;
6187
6188 if (compile_errors) {
6189 con_out("*** there were compile errors\n");
6190 return false;
6191 }
6192
6193 ir = new ir_builder("gmqcc_out");
6194 if (!ir) {
6195 con_out("failed to allocate builder\n");
6196 return false;
6197 }
6198
6199 for (auto &it : parser->fields) {
6200 bool hasvalue;
6201 if (!ast_istype(it, ast_value))
6202 continue;
6203 ast_value *field = (ast_value*)it;
6204 hasvalue = field->m_hasvalue;
6205 field->m_hasvalue = false;
6206 if (!reinterpret_cast<ast_value*>(field)->generateGlobal(ir, true)) {
6207 con_out("failed to generate field %s\n", field->m_name.c_str());
6208 delete ir;
6209 return false;
6210 }
6211 if (hasvalue) {
6212 ir_value *ifld;
6213 ast_expression *subtype;
6214 field->m_hasvalue = true;
6215 subtype = field->m_next;
6216 ifld = ir->createField(field->m_name, subtype->m_vtype);
6217 if (subtype->m_vtype == TYPE_FIELD)
6218 ifld->m_fieldtype = subtype->m_next->m_vtype;
6219 else if (subtype->m_vtype == TYPE_FUNCTION)
6220 ifld->m_outtype = subtype->m_next->m_vtype;
6221 (void)!field->m_ir_v->setField(ifld);
6222 }
6223 }
6224 for (auto &it : parser->globals) {
6225 ast_value *asvalue;
6226 if (!ast_istype(it, ast_value))
6227 continue;
6228 asvalue = (ast_value*)it;
6229 if (!asvalue->m_uses && !asvalue->m_hasvalue && asvalue->m_vtype != TYPE_FUNCTION) {
6230 retval = retval && !compile_warning(asvalue->m_context, WARN_UNUSED_VARIABLE,
6231 "unused global: `%s`", asvalue->m_name);
6232 }
6233 if (!asvalue->generateGlobal(ir, false)) {
6234 con_out("failed to generate global %s\n", asvalue->m_name.c_str());
6235 delete ir;
6236 return false;
6237 }
6238 }
6239 /* Build function vararg accessor ast tree now before generating
6240 * immediates, because the accessors may add new immediates
6241 */
6242 for (auto &f : parser->functions) {
6243 if (f->m_varargs) {
6244 if (parser->max_param_count > f->m_function_type->m_type_params.size()) {
6245 f->m_varargs->m_count = parser->max_param_count - f->m_function_type->m_type_params.size();
6246 if (!parser_create_array_setter_impl(parser, f->m_varargs.get())) {
6247 con_out("failed to generate vararg setter for %s\n", f->m_name.c_str());
6248 delete ir;
6249 return false;
6250 }
6251 if (!parser_create_array_getter_impl(parser, f->m_varargs.get())) {
6252 con_out("failed to generate vararg getter for %s\n", f->m_name.c_str());
6253 delete ir;
6254 return false;
6255 }
6256 } else {
6257 f->m_varargs = nullptr;
6258 }
6259 }
6260 }
6261 /* Now we can generate immediates */
6262 if (!parser->m_fold.generate(ir))
6263 return false;
6264
6265 /* before generating any functions we need to set the coverage_func */
6266 if (!parser_set_coverage_func(parser, ir))
6267 return false;
6268 for (auto &it : parser->globals) {
6269 if (!ast_istype(it, ast_value))
6270 continue;
6271 ast_value *asvalue = (ast_value*)it;
6272 if (!(asvalue->m_flags & AST_FLAG_INITIALIZED))
6273 {
6274 if (asvalue->m_cvq == CV_CONST && !asvalue->m_hasvalue)
6275 (void)!compile_warning(asvalue->m_context, WARN_UNINITIALIZED_CONSTANT,
6276 "uninitialized constant: `%s`",
6277 asvalue->m_name);
6278 else if ((asvalue->m_cvq == CV_NONE || asvalue->m_cvq == CV_CONST) && !asvalue->m_hasvalue)
6279 (void)!compile_warning(asvalue->m_context, WARN_UNINITIALIZED_GLOBAL,
6280 "uninitialized global: `%s`",
6281 asvalue->m_name);
6282 }
6283 if (!asvalue->generateAccessors(ir)) {
6284 delete ir;
6285 return false;
6286 }
6287 }
6288 for (auto &it : parser->fields) {
6289 ast_value *asvalue = (ast_value*)it->m_next;
6290 if (!ast_istype(asvalue, ast_value))
6291 continue;
6292 if (asvalue->m_vtype != TYPE_ARRAY)
6293 continue;
6294 if (!asvalue->generateAccessors(ir)) {
6295 delete ir;
6296 return false;
6297 }
6298 }
6299 if (parser->reserved_version &&
6300 !parser->reserved_version->generateGlobal(ir, false))
6301 {
6302 con_out("failed to generate reserved::version");
6303 delete ir;
6304 return false;
6305 }
6306 for (auto &f : parser->functions) {
6307 if (!f->generateFunction(ir)) {
6308 con_out("failed to generate function %s\n", f->m_name.c_str());
6309 delete ir;
6310 return false;
6311 }
6312 }
6313
6314 generate_checksum(parser, ir);
6315
6316 if (OPTS_OPTION_BOOL(OPTION_DUMP))
6317 ir->dump(con_out);
6318 for (auto &it : parser->functions) {
6319 if (!ir_function_finalize(it->m_ir_func)) {
6320 con_out("failed to finalize function %s\n", it->m_name.c_str());
6321 delete ir;
6322 return false;
6323 }
6324 }
6325 parser_remove_ast(parser);
6326
6327 if (compile_Werrors) {
6328 con_out("*** there were warnings treated as errors\n");
6329 compile_show_werrors();
6330 retval = false;
6331 }
6332
6333 if (retval) {
6334 if (OPTS_OPTION_BOOL(OPTION_DUMPFIN))
6335 ir->dump(con_out);
6336
6337 if (!ir->generate(output)) {
6338 con_out("*** failed to generate output file\n");
6339 delete ir;
6340 return false;
6341 }
6342 }
6343 delete ir;
6344 return retval;
6345 }
6346