#include #include #include "parser.h" #define PARSER_HT_LOCALS 2 #define PARSER_HT_SIZE 512 #define TYPEDEF_HT_SIZE 512 static void parser_enterblock(parser_t *parser); static bool parser_leaveblock(parser_t *parser); static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e); static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e); static bool parse_typedef(parser_t *parser); 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); static ast_block* parse_block(parser_t *parser); static bool parse_block_into(parser_t *parser, ast_block *block); static bool parse_statement_or_block(parser_t *parser, ast_expression **out); static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases); static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma, bool truthvalue, bool with_labels); static ast_expression* parse_expression(parser_t *parser, bool stopatcomma, bool with_labels); static ast_value* parser_create_array_setter_proto(parser_t *parser, ast_value *array, const char *funcname); static ast_value* parser_create_array_getter_proto(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname); static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef, bool *is_vararg); static void parseerror(parser_t *parser, const char *fmt, ...) { va_list ap; va_start(ap, fmt); vcompile_error(parser->lex->tok.ctx, fmt, ap); va_end(ap); } /* returns true if it counts as an error */ static bool GMQCC_WARN parsewarning(parser_t *parser, int warntype, const char *fmt, ...) { bool r; va_list ap; va_start(ap, fmt); r = vcompile_warning(parser->lex->tok.ctx, warntype, fmt, ap); va_end(ap); return r; } /********************************************************************** * parsing */ static bool parser_next(parser_t *parser) { /* lex_do kills the previous token */ parser->tok = lex_do(parser->lex); if (parser->tok == TOKEN_EOF) return true; if (parser->tok >= TOKEN_ERROR) { parseerror(parser, "lex error"); return false; } return true; } #define parser_tokval(p) ((p)->lex->tok.value) #define parser_token(p) (&((p)->lex->tok)) char *parser_strdup(const char *str) { if (str && !*str) { /* actually dup empty strings */ char *out = (char*)mem_a(1); *out = 0; return out; } return util_strdup(str); } static ast_expression* parser_find_field(parser_t *parser, const char *name) { return ( ast_expression*)util_htget(parser->htfields, name); } static ast_expression* parser_find_label(parser_t *parser, const char *name) { for (auto &it : parser->labels) if (!strcmp(it->name, name)) return (ast_expression*)it; return nullptr; } ast_expression* parser_find_global(parser_t *parser, const char *name) { ast_expression *var = (ast_expression*)util_htget(parser->aliases, parser_tokval(parser)); if (var) return var; return (ast_expression*)util_htget(parser->htglobals, name); } static ast_expression* parser_find_param(parser_t *parser, const char *name) { ast_value *fun; if (!parser->function) return NULL; fun = parser->function->vtype; for (auto &it : fun->expression.params) { if (!strcmp(it->name, name)) return (ast_expression*)it; } return NULL; } static ast_expression* parser_find_local(parser_t *parser, const char *name, size_t upto, bool *isparam) { size_t i, hash; ast_expression *e; hash = util_hthash(parser->htglobals, name); *isparam = false; for (i = vec_size(parser->variables); i > upto;) { --i; if ( (e = (ast_expression*)util_htgeth(parser->variables[i], name, hash)) ) return e; } *isparam = true; return parser_find_param(parser, name); } static ast_expression* parser_find_var(parser_t *parser, const char *name) { bool dummy; ast_expression *v; v = parser_find_local(parser, name, 0, &dummy); if (!v) v = parser_find_global(parser, name); return v; } static ast_value* parser_find_typedef(parser_t *parser, const char *name, size_t upto) { size_t i, hash; ast_value *e; hash = util_hthash(parser->typedefs[0], name); for (i = vec_size(parser->typedefs); i > upto;) { --i; if ( (e = (ast_value*)util_htgeth(parser->typedefs[i], name, hash)) ) return e; } return NULL; } struct sy_elem { size_t etype; /* 0 = expression, others are operators */ bool isparen; size_t off; ast_expression *out; ast_block *block; /* for commas and function calls */ lex_ctx_t ctx; }; enum { PAREN_EXPR, PAREN_FUNC, PAREN_INDEX, PAREN_TERNARY1, PAREN_TERNARY2 }; struct shunt { sy_elem *out; sy_elem *ops; size_t *argc; unsigned int *paren; }; static sy_elem syexp(lex_ctx_t ctx, ast_expression *v) { sy_elem e; e.etype = 0; e.off = 0; e.out = v; e.block = NULL; e.ctx = ctx; e.isparen = false; return e; } static sy_elem syblock(lex_ctx_t ctx, ast_block *v) { sy_elem e; e.etype = 0; e.off = 0; e.out = (ast_expression*)v; e.block = v; e.ctx = ctx; e.isparen = false; return e; } static sy_elem syop(lex_ctx_t ctx, const oper_info *op) { sy_elem e; e.etype = 1 + (op - operators); e.off = 0; e.out = NULL; e.block = NULL; e.ctx = ctx; e.isparen = false; return e; } static sy_elem syparen(lex_ctx_t ctx, size_t off) { sy_elem e; e.etype = 0; e.off = off; e.out = NULL; e.block = NULL; e.ctx = ctx; e.isparen = true; return e; } /* With regular precedence rules, ent.foo[n] is the same as (ent.foo)[n], * so we need to rotate it to become ent.(foo[n]). */ static bool rotate_entfield_array_index_nodes(ast_expression **out) { ast_array_index *index, *oldindex; ast_entfield *entfield; ast_value *field; ast_expression *sub; ast_expression *entity; lex_ctx_t ctx = ast_ctx(*out); if (!ast_istype(*out, ast_array_index)) return false; index = (ast_array_index*)*out; if (!ast_istype(index->array, ast_entfield)) return false; entfield = (ast_entfield*)index->array; if (!ast_istype(entfield->field, ast_value)) return false; field = (ast_value*)entfield->field; sub = index->index; entity = entfield->entity; oldindex = index; index = ast_array_index_new(ctx, (ast_expression*)field, sub); entfield = ast_entfield_new(ctx, entity, (ast_expression*)index); *out = (ast_expression*)entfield; oldindex->array = NULL; oldindex->index = NULL; ast_delete(oldindex); return true; } static bool check_write_to(lex_ctx_t ctx, ast_expression *expr) { if (ast_istype(expr, ast_value)) { ast_value *val = (ast_value*)expr; if (val->cvq == CV_CONST) { if (val->name[0] == '#') { compile_error(ctx, "invalid assignment to a literal constant"); return false; } /* * To work around quakeworld we must elide the error and make it * a warning instead. */ if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_QCC) compile_error(ctx, "assignment to constant `%s`", val->name); else (void)!compile_warning(ctx, WARN_CONST_OVERWRITE, "assignment to constant `%s`", val->name); return false; } } return true; } static bool parser_sy_apply_operator(parser_t *parser, shunt *sy) { const oper_info *op; lex_ctx_t ctx; ast_expression *out = NULL; ast_expression *exprs[3]; ast_block *blocks[3]; ast_binstore *asbinstore; size_t i, assignop, addop, subop; qcint_t generated_op = 0; char ty1[1024]; char ty2[1024]; if (!vec_size(sy->ops)) { parseerror(parser, "internal error: missing operator"); return false; } if (vec_last(sy->ops).isparen) { parseerror(parser, "unmatched parenthesis"); return false; } op = &operators[vec_last(sy->ops).etype - 1]; ctx = vec_last(sy->ops).ctx; if (vec_size(sy->out) < op->operands) { if (op->flags & OP_PREFIX) compile_error(ctx, "expected expression after unary operator `%s`", op->op, (int)op->id); else /* this should have errored previously already */ compile_error(ctx, "expected expression after operator `%s`", op->op, (int)op->id); return false; } vec_shrinkby(sy->ops, 1); /* op(:?) has no input and no output */ if (!op->operands) return true; vec_shrinkby(sy->out, op->operands); for (i = 0; i < op->operands; ++i) { exprs[i] = sy->out[vec_size(sy->out)+i].out; blocks[i] = sy->out[vec_size(sy->out)+i].block; if (exprs[i]->vtype == TYPE_NOEXPR && !(i != 0 && op->id == opid2('?',':')) && !(i == 1 && op->id == opid1('.'))) { if (ast_istype(exprs[i], ast_label)) compile_error(ast_ctx(exprs[i]), "expected expression, got an unknown identifier"); else compile_error(ast_ctx(exprs[i]), "not an expression"); (void)!compile_warning(ast_ctx(exprs[i]), WARN_DEBUG, "expression %u\n", (unsigned int)i); } } if (blocks[0] && blocks[0]->exprs.empty() && op->id != opid1(',')) { compile_error(ctx, "internal error: operator cannot be applied on empty blocks"); return false; } #define NotSameType(T) \ (exprs[0]->vtype != exprs[1]->vtype || \ exprs[0]->vtype != T) switch (op->id) { default: compile_error(ctx, "internal error: unhandled operator: %s (%i)", op->op, (int)op->id); return false; case opid1('.'): if (exprs[0]->vtype == TYPE_VECTOR && exprs[1]->vtype == TYPE_NOEXPR) { if (exprs[1] == (ast_expression*)parser->const_vec[0]) out = (ast_expression*)ast_member_new(ctx, exprs[0], 0, NULL); else if (exprs[1] == (ast_expression*)parser->const_vec[1]) out = (ast_expression*)ast_member_new(ctx, exprs[0], 1, NULL); else if (exprs[1] == (ast_expression*)parser->const_vec[2]) out = (ast_expression*)ast_member_new(ctx, exprs[0], 2, NULL); else { compile_error(ctx, "access to invalid vector component"); return false; } } else if (exprs[0]->vtype == TYPE_ENTITY) { if (exprs[1]->vtype != TYPE_FIELD) { compile_error(ast_ctx(exprs[1]), "type error: right hand of member-operand should be an entity-field"); return false; } out = (ast_expression*)ast_entfield_new(ctx, exprs[0], exprs[1]); } else if (exprs[0]->vtype == TYPE_VECTOR) { compile_error(ast_ctx(exprs[1]), "vectors cannot be accessed this way"); return false; } else { compile_error(ast_ctx(exprs[1]), "type error: member-of operator on something that is not an entity or vector"); return false; } break; case opid1('['): if (exprs[0]->vtype != TYPE_ARRAY && !(exprs[0]->vtype == TYPE_FIELD && exprs[0]->next->vtype == TYPE_ARRAY)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); compile_error(ast_ctx(exprs[0]), "cannot index value of type %s", ty1); return false; } if (exprs[1]->vtype != TYPE_FLOAT) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); compile_error(ast_ctx(exprs[1]), "index must be of type float, not %s", ty1); return false; } out = (ast_expression*)ast_array_index_new(ctx, exprs[0], exprs[1]); rotate_entfield_array_index_nodes(&out); break; case opid1(','): if (vec_size(sy->paren) && vec_last(sy->paren) == PAREN_FUNC) { vec_push(sy->out, syexp(ctx, exprs[0])); vec_push(sy->out, syexp(ctx, exprs[1])); vec_last(sy->argc)++; return true; } if (blocks[0]) { if (!ast_block_add_expr(blocks[0], exprs[1])) return false; } else { blocks[0] = ast_block_new(ctx); if (!ast_block_add_expr(blocks[0], exprs[0]) || !ast_block_add_expr(blocks[0], exprs[1])) { return false; } } ast_block_set_type(blocks[0], exprs[1]); vec_push(sy->out, syblock(ctx, blocks[0])); return true; case opid2('+','P'): out = exprs[0]; break; case opid2('-','P'): if ((out = fold_op(parser->fold, op, exprs))) break; if (exprs[0]->vtype != TYPE_FLOAT && exprs[0]->vtype != TYPE_VECTOR) { compile_error(ctx, "invalid types used in unary expression: cannot negate type %s", type_name[exprs[0]->vtype]); return false; } if (exprs[0]->vtype == TYPE_FLOAT) out = (ast_expression*)ast_unary_new(ctx, VINSTR_NEG_F, exprs[0]); else out = (ast_expression*)ast_unary_new(ctx, VINSTR_NEG_V, exprs[0]); break; case opid2('!','P'): if (!(out = fold_op(parser->fold, op, exprs))) { switch (exprs[0]->vtype) { case TYPE_FLOAT: out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]); break; case TYPE_VECTOR: out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]); break; case TYPE_STRING: if (OPTS_FLAG(TRUE_EMPTY_STRINGS)) out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]); else out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[0]); break; /* we don't constant-fold NOT for these types */ case TYPE_ENTITY: out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_ENT, exprs[0]); break; case TYPE_FUNCTION: out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_FNC, exprs[0]); break; default: compile_error(ctx, "invalid types used in expression: cannot logically negate type %s", type_name[exprs[0]->vtype]); return false; } } break; case opid1('+'): if (exprs[0]->vtype != exprs[1]->vtype || (exprs[0]->vtype != TYPE_VECTOR && exprs[0]->vtype != TYPE_FLOAT) ) { compile_error(ctx, "invalid types used in expression: cannot add type %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { switch (exprs[0]->vtype) { case TYPE_FLOAT: out = fold_binary(ctx, INSTR_ADD_F, exprs[0], exprs[1]); break; case TYPE_VECTOR: out = fold_binary(ctx, INSTR_ADD_V, exprs[0], exprs[1]); break; default: compile_error(ctx, "invalid types used in expression: cannot add type %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } } break; case opid1('-'): if (exprs[0]->vtype != exprs[1]->vtype || (exprs[0]->vtype != TYPE_VECTOR && exprs[0]->vtype != TYPE_FLOAT)) { compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s", type_name[exprs[1]->vtype], type_name[exprs[0]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { switch (exprs[0]->vtype) { case TYPE_FLOAT: out = fold_binary(ctx, INSTR_SUB_F, exprs[0], exprs[1]); break; case TYPE_VECTOR: out = fold_binary(ctx, INSTR_SUB_V, exprs[0], exprs[1]); break; default: compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s", type_name[exprs[1]->vtype], type_name[exprs[0]->vtype]); return false; } } break; case opid1('*'): if (exprs[0]->vtype != exprs[1]->vtype && !(exprs[0]->vtype == TYPE_VECTOR && exprs[1]->vtype == TYPE_FLOAT) && !(exprs[1]->vtype == TYPE_VECTOR && exprs[0]->vtype == TYPE_FLOAT) ) { compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s", type_name[exprs[1]->vtype], type_name[exprs[0]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { switch (exprs[0]->vtype) { case TYPE_FLOAT: if (exprs[1]->vtype == TYPE_VECTOR) out = fold_binary(ctx, INSTR_MUL_FV, exprs[0], exprs[1]); else out = fold_binary(ctx, INSTR_MUL_F, exprs[0], exprs[1]); break; case TYPE_VECTOR: if (exprs[1]->vtype == TYPE_FLOAT) out = fold_binary(ctx, INSTR_MUL_VF, exprs[0], exprs[1]); else out = fold_binary(ctx, INSTR_MUL_V, exprs[0], exprs[1]); break; default: compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s", type_name[exprs[1]->vtype], type_name[exprs[0]->vtype]); return false; } } break; case opid1('/'): if (exprs[1]->vtype != TYPE_FLOAT) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in expression: cannot divide types %s and %s", ty1, ty2); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { if (exprs[0]->vtype == TYPE_FLOAT) out = fold_binary(ctx, INSTR_DIV_F, exprs[0], exprs[1]); else { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in expression: cannot divide types %s and %s", ty1, ty2); return false; } } break; case opid1('%'): if (NotSameType(TYPE_FLOAT)) { compile_error(ctx, "invalid types used in expression: cannot perform modulo operation between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } else if (!(out = fold_op(parser->fold, op, exprs))) { /* generate a call to __builtin_mod */ ast_expression *mod = intrin_func(parser->intrin, "mod"); ast_call *call = NULL; if (!mod) return false; /* can return null for missing floor */ call = ast_call_new(parser_ctx(parser), mod); call->params.push_back(exprs[0]); call->params.push_back(exprs[1]); out = (ast_expression*)call; } break; case opid2('%','='): compile_error(ctx, "%= is unimplemented"); return false; case opid1('|'): case opid1('&'): case opid1('^'): if ( !(exprs[0]->vtype == TYPE_FLOAT && exprs[1]->vtype == TYPE_FLOAT) && !(exprs[0]->vtype == TYPE_VECTOR && exprs[1]->vtype == TYPE_FLOAT) && !(exprs[0]->vtype == TYPE_VECTOR && exprs[1]->vtype == TYPE_VECTOR)) { compile_error(ctx, "invalid types used in expression: cannot perform bit operations between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { /* * IF the first expression is float, the following will be too * since scalar ^ vector is not allowed. */ if (exprs[0]->vtype == TYPE_FLOAT) { out = fold_binary(ctx, (op->id == opid1('^') ? VINSTR_BITXOR : op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND), exprs[0], exprs[1]); } else { /* * The first is a vector: vector is allowed to bitop with vector and * with scalar, branch here for the second operand. */ if (exprs[1]->vtype == TYPE_VECTOR) { /* * Bitop all the values of the vector components against the * vectors components in question. */ out = fold_binary(ctx, (op->id == opid1('^') ? VINSTR_BITXOR_V : op->id == opid1('|') ? VINSTR_BITOR_V : VINSTR_BITAND_V), exprs[0], exprs[1]); } else { out = fold_binary(ctx, (op->id == opid1('^') ? VINSTR_BITXOR_VF : op->id == opid1('|') ? VINSTR_BITOR_VF : VINSTR_BITAND_VF), exprs[0], exprs[1]); } } } break; case opid2('<','<'): case opid2('>','>'): if (NotSameType(TYPE_FLOAT)) { compile_error(ctx, "invalid types used in expression: cannot perform shift between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { ast_expression *shift = intrin_func(parser->intrin, (op->id == opid2('<','<')) ? "__builtin_lshift" : "__builtin_rshift"); ast_call *call = ast_call_new(parser_ctx(parser), shift); call->params.push_back(exprs[0]); call->params.push_back(exprs[1]); out = (ast_expression*)call; } break; case opid3('<','<','='): case opid3('>','>','='): if (NotSameType(TYPE_FLOAT)) { compile_error(ctx, "invalid types used in expression: cannot perform shift operation between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if(!(out = fold_op(parser->fold, op, exprs))) { ast_expression *shift = intrin_func(parser->intrin, (op->id == opid3('<','<','=')) ? "__builtin_lshift" : "__builtin_rshift"); ast_call *call = ast_call_new(parser_ctx(parser), shift); call->params.push_back(exprs[0]); call->params.push_back(exprs[1]); out = (ast_expression*)ast_store_new( parser_ctx(parser), INSTR_STORE_F, exprs[0], (ast_expression*)call ); } break; case opid2('|','|'): generated_op += 1; /* INSTR_OR */ case opid2('&','&'): generated_op += INSTR_AND; if (!(out = fold_op(parser->fold, op, exprs))) { if (OPTS_FLAG(PERL_LOGIC) && !ast_compare_type(exprs[0], exprs[1])) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types for logical operation with -fperl-logic: %s and %s", ty1, ty2); return false; } for (i = 0; i < 2; ++i) { if (OPTS_FLAG(CORRECT_LOGIC) && exprs[i]->vtype == TYPE_VECTOR) { out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[i]); if (!out) break; out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, out); if (!out) break; exprs[i] = out; out = NULL; if (OPTS_FLAG(PERL_LOGIC)) { /* here we want to keep the right expressions' type */ break; } } else if (OPTS_FLAG(FALSE_EMPTY_STRINGS) && exprs[i]->vtype == TYPE_STRING) { out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_S, exprs[i]); if (!out) break; out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, out); if (!out) break; exprs[i] = out; out = NULL; if (OPTS_FLAG(PERL_LOGIC)) { /* here we want to keep the right expressions' type */ break; } } } out = fold_binary(ctx, generated_op, exprs[0], exprs[1]); } break; case opid2('?',':'): if (vec_last(sy->paren) != PAREN_TERNARY2) { compile_error(ctx, "mismatched parenthesis/ternary"); return false; } vec_pop(sy->paren); if (!ast_compare_type(exprs[1], exprs[2])) { ast_type_to_string(exprs[1], ty1, sizeof(ty1)); ast_type_to_string(exprs[2], ty2, sizeof(ty2)); compile_error(ctx, "operands of ternary expression must have the same type, got %s and %s", ty1, ty2); return false; } if (!(out = fold_op(parser->fold, op, exprs))) out = (ast_expression*)ast_ternary_new(ctx, exprs[0], exprs[1], exprs[2]); break; case opid2('*', '*'): if (NotSameType(TYPE_FLOAT)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in exponentiation: %s and %s", ty1, ty2); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { ast_call *gencall = ast_call_new(parser_ctx(parser), intrin_func(parser->intrin, "pow")); gencall->params.push_back(exprs[0]); gencall->params.push_back(exprs[1]); out = (ast_expression*)gencall; } break; case opid2('>', '<'): if (NotSameType(TYPE_VECTOR)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in cross product: %s and %s", ty1, ty2); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { out = fold_binary( parser_ctx(parser), VINSTR_CROSS, exprs[0], exprs[1] ); } break; case opid3('<','=','>'): /* -1, 0, or 1 */ if (NotSameType(TYPE_FLOAT)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in comparision: %s and %s", ty1, ty2); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { /* This whole block is NOT fold_binary safe */ ast_binary *eq = ast_binary_new(ctx, INSTR_EQ_F, exprs[0], exprs[1]); eq->refs = AST_REF_NONE; /* if (lt) { */ out = (ast_expression*)ast_ternary_new(ctx, (ast_expression*)ast_binary_new(ctx, INSTR_LT, exprs[0], exprs[1]), /* out = -1 */ (ast_expression*)parser->fold->imm_float[2], /* } else { */ /* if (eq) { */ (ast_expression*)ast_ternary_new(ctx, (ast_expression*)eq, /* out = 0 */ (ast_expression*)parser->fold->imm_float[0], /* } else { */ /* out = 1 */ (ast_expression*)parser->fold->imm_float[1] /* } */ ) /* } */ ); } break; case opid1('>'): generated_op += 1; /* INSTR_GT */ case opid1('<'): generated_op += 1; /* INSTR_LT */ case opid2('>', '='): generated_op += 1; /* INSTR_GE */ case opid2('<', '='): generated_op += INSTR_LE; if (NotSameType(TYPE_FLOAT)) { compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) out = fold_binary(ctx, generated_op, exprs[0], exprs[1]); break; case opid2('!', '='): if (exprs[0]->vtype != exprs[1]->vtype) { compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) out = fold_binary(ctx, type_ne_instr[exprs[0]->vtype], exprs[0], exprs[1]); break; case opid2('=', '='): if (exprs[0]->vtype != exprs[1]->vtype) { compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; } if (!(out = fold_op(parser->fold, op, exprs))) out = fold_binary(ctx, type_eq_instr[exprs[0]->vtype], exprs[0], exprs[1]); break; case opid1('='): if (ast_istype(exprs[0], ast_entfield)) { ast_expression *field = ((ast_entfield*)exprs[0])->field; if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) && exprs[0]->vtype == TYPE_FIELD && exprs[0]->next->vtype == TYPE_VECTOR) { assignop = type_storep_instr[TYPE_VECTOR]; } else assignop = type_storep_instr[exprs[0]->vtype]; if (assignop == VINSTR_END || !ast_compare_type(field->next, exprs[1])) { ast_type_to_string(field->next, ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) && field->next->vtype == TYPE_FUNCTION && exprs[1]->vtype == TYPE_FUNCTION) { (void)!compile_warning(ctx, WARN_ASSIGN_FUNCTION_TYPES, "invalid types in assignment: cannot assign %s to %s", ty2, ty1); } else compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1); } } else { if (OPTS_FLAG(ADJUST_VECTOR_FIELDS) && exprs[0]->vtype == TYPE_FIELD && exprs[0]->next->vtype == TYPE_VECTOR) { assignop = type_store_instr[TYPE_VECTOR]; } else { assignop = type_store_instr[exprs[0]->vtype]; } if (assignop == VINSTR_END) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1); } else if (!ast_compare_type(exprs[0], exprs[1])) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) && exprs[0]->vtype == TYPE_FUNCTION && exprs[1]->vtype == TYPE_FUNCTION) { (void)!compile_warning(ctx, WARN_ASSIGN_FUNCTION_TYPES, "invalid types in assignment: cannot assign %s to %s", ty2, ty1); } else compile_error(ctx, "invalid types in assignment: cannot assign %s to %s", ty2, ty1); } } (void)check_write_to(ctx, exprs[0]); /* When we're a vector of part of an entity field we use STOREP */ if (ast_istype(exprs[0], ast_member) && ast_istype(((ast_member*)exprs[0])->owner, ast_entfield)) assignop = INSTR_STOREP_F; out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]); break; case opid3('+','+','P'): case opid3('-','-','P'): /* prefix ++ */ if (exprs[0]->vtype != TYPE_FLOAT) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); compile_error(ast_ctx(exprs[0]), "invalid type for prefix increment: %s", ty1); return false; } if (op->id == opid3('+','+','P')) addop = INSTR_ADD_F; else addop = INSTR_SUB_F; (void)check_write_to(ast_ctx(exprs[0]), exprs[0]); if (ast_istype(exprs[0], ast_entfield)) { out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop, exprs[0], (ast_expression*)parser->fold->imm_float[1]); } else { out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop, exprs[0], (ast_expression*)parser->fold->imm_float[1]); } break; case opid3('S','+','+'): case opid3('S','-','-'): /* prefix ++ */ if (exprs[0]->vtype != TYPE_FLOAT) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); compile_error(ast_ctx(exprs[0]), "invalid type for suffix increment: %s", ty1); return false; } if (op->id == opid3('S','+','+')) { addop = INSTR_ADD_F; subop = INSTR_SUB_F; } else { addop = INSTR_SUB_F; subop = INSTR_ADD_F; } (void)check_write_to(ast_ctx(exprs[0]), exprs[0]); if (ast_istype(exprs[0], ast_entfield)) { out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop, exprs[0], (ast_expression*)parser->fold->imm_float[1]); } else { out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop, exprs[0], (ast_expression*)parser->fold->imm_float[1]); } if (!out) return false; out = fold_binary(ctx, subop, out, (ast_expression*)parser->fold->imm_float[1]); break; case opid2('+','='): case opid2('-','='): if (exprs[0]->vtype != exprs[1]->vtype || (exprs[0]->vtype != TYPE_VECTOR && exprs[0]->vtype != TYPE_FLOAT) ) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s", ty1, ty2); return false; } (void)check_write_to(ctx, exprs[0]); if (ast_istype(exprs[0], ast_entfield)) assignop = type_storep_instr[exprs[0]->vtype]; else assignop = type_store_instr[exprs[0]->vtype]; switch (exprs[0]->vtype) { case TYPE_FLOAT: out = (ast_expression*)ast_binstore_new(ctx, assignop, (op->id == opid2('+','=') ? INSTR_ADD_F : INSTR_SUB_F), exprs[0], exprs[1]); break; case TYPE_VECTOR: out = (ast_expression*)ast_binstore_new(ctx, assignop, (op->id == opid2('+','=') ? INSTR_ADD_V : INSTR_SUB_V), exprs[0], exprs[1]); break; default: compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; }; break; case opid2('*','='): case opid2('/','='): if (exprs[1]->vtype != TYPE_FLOAT || !(exprs[0]->vtype == TYPE_FLOAT || exprs[0]->vtype == TYPE_VECTOR)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in expression: %s and %s", ty1, ty2); return false; } (void)check_write_to(ctx, exprs[0]); if (ast_istype(exprs[0], ast_entfield)) assignop = type_storep_instr[exprs[0]->vtype]; else assignop = type_store_instr[exprs[0]->vtype]; switch (exprs[0]->vtype) { case TYPE_FLOAT: out = (ast_expression*)ast_binstore_new(ctx, assignop, (op->id == opid2('*','=') ? INSTR_MUL_F : INSTR_DIV_F), exprs[0], exprs[1]); break; case TYPE_VECTOR: if (op->id == opid2('*','=')) { out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_MUL_VF, exprs[0], exprs[1]); } else { out = fold_binary(ctx, INSTR_DIV_F, (ast_expression*)parser->fold->imm_float[1], exprs[1]); if (!out) { compile_error(ctx, "internal error: failed to generate division"); return false; } out = (ast_expression*)ast_binstore_new(ctx, assignop, INSTR_MUL_VF, exprs[0], out); } break; default: compile_error(ctx, "invalid types used in expression: cannot add or subtract type %s and %s", type_name[exprs[0]->vtype], type_name[exprs[1]->vtype]); return false; }; break; case opid2('&','='): case opid2('|','='): case opid2('^','='): if (NotSameType(TYPE_FLOAT) && NotSameType(TYPE_VECTOR)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in expression: %s and %s", ty1, ty2); return false; } (void)check_write_to(ctx, exprs[0]); if (ast_istype(exprs[0], ast_entfield)) assignop = type_storep_instr[exprs[0]->vtype]; else assignop = type_store_instr[exprs[0]->vtype]; if (exprs[0]->vtype == TYPE_FLOAT) out = (ast_expression*)ast_binstore_new(ctx, assignop, (op->id == opid2('^','=') ? VINSTR_BITXOR : op->id == opid2('&','=') ? INSTR_BITAND : INSTR_BITOR), exprs[0], exprs[1]); else out = (ast_expression*)ast_binstore_new(ctx, assignop, (op->id == opid2('^','=') ? VINSTR_BITXOR_V : op->id == opid2('&','=') ? VINSTR_BITAND_V : VINSTR_BITOR_V), exprs[0], exprs[1]); break; case opid3('&','~','='): /* This is like: a &= ~(b); * But QC has no bitwise-not, so we implement it as * a -= a & (b); */ if (NotSameType(TYPE_FLOAT) && NotSameType(TYPE_VECTOR)) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); ast_type_to_string(exprs[1], ty2, sizeof(ty2)); compile_error(ctx, "invalid types used in expression: %s and %s", ty1, ty2); return false; } if (ast_istype(exprs[0], ast_entfield)) assignop = type_storep_instr[exprs[0]->vtype]; else assignop = type_store_instr[exprs[0]->vtype]; if (exprs[0]->vtype == TYPE_FLOAT) out = fold_binary(ctx, INSTR_BITAND, exprs[0], exprs[1]); else out = fold_binary(ctx, VINSTR_BITAND_V, exprs[0], exprs[1]); if (!out) return false; (void)check_write_to(ctx, exprs[0]); if (exprs[0]->vtype == TYPE_FLOAT) asbinstore = ast_binstore_new(ctx, assignop, INSTR_SUB_F, exprs[0], out); else asbinstore = ast_binstore_new(ctx, assignop, INSTR_SUB_V, exprs[0], out); asbinstore->keep_dest = true; out = (ast_expression*)asbinstore; break; case opid3('l', 'e', 'n'): if (exprs[0]->vtype != TYPE_STRING && exprs[0]->vtype != TYPE_ARRAY) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); compile_error(ast_ctx(exprs[0]), "invalid type for length operator: %s", ty1); return false; } /* strings must be const, arrays are statically sized */ if (exprs[0]->vtype == TYPE_STRING && !(((ast_value*)exprs[0])->hasvalue && ((ast_value*)exprs[0])->cvq == CV_CONST)) { compile_error(ast_ctx(exprs[0]), "operand of length operator not a valid constant expression"); return false; } out = fold_op(parser->fold, op, exprs); break; case opid2('~', 'P'): if (exprs[0]->vtype != TYPE_FLOAT && exprs[0]->vtype != TYPE_VECTOR) { ast_type_to_string(exprs[0], ty1, sizeof(ty1)); compile_error(ast_ctx(exprs[0]), "invalid type for bit not: %s", ty1); return false; } if (!(out = fold_op(parser->fold, op, exprs))) { if (exprs[0]->vtype == TYPE_FLOAT) { out = fold_binary(ctx, INSTR_SUB_F, (ast_expression*)parser->fold->imm_float[2], exprs[0]); } else { out = fold_binary(ctx, INSTR_SUB_V, (ast_expression*)parser->fold->imm_vector[1], exprs[0]); } } break; } #undef NotSameType if (!out) { compile_error(ctx, "failed to apply operator %s", op->op); return false; } vec_push(sy->out, syexp(ctx, out)); return true; } static bool parser_close_call(parser_t *parser, shunt *sy) { /* was a function call */ ast_expression *fun; ast_value *funval = NULL; ast_call *call; size_t fid; size_t paramcount, i; bool fold = true; fid = vec_last(sy->ops).off; vec_shrinkby(sy->ops, 1); /* out[fid] is the function * everything above is parameters... */ if (!vec_size(sy->argc)) { parseerror(parser, "internal error: no argument counter available"); return false; } paramcount = vec_last(sy->argc); vec_pop(sy->argc); if (vec_size(sy->out) < fid) { parseerror(parser, "internal error: broken function call%lu < %lu+%lu\n", (unsigned long)vec_size(sy->out), (unsigned long)fid, (unsigned long)paramcount); return false; } /* * TODO handle this at the intrinsic level with an ast_intrinsic * node and codegen. */ if ((fun = sy->out[fid].out) == intrin_debug_typestring(parser->intrin)) { char ty[1024]; if (fid+2 != vec_size(sy->out) || vec_last(sy->out).block) { parseerror(parser, "intrinsic __builtin_debug_typestring requires exactly 1 parameter"); return false; } ast_type_to_string(vec_last(sy->out).out, ty, sizeof(ty)); ast_unref(vec_last(sy->out).out); sy->out[fid] = syexp(ast_ctx(vec_last(sy->out).out), (ast_expression*)fold_constgen_string(parser->fold, ty, false)); vec_shrinkby(sy->out, 1); return true; } /* * Now we need to determine if the function that is being called is * an intrinsic so we can evaluate if the arguments to it are constant * and than fruitfully fold them. */ #define fold_can_1(X) \ (ast_istype(((ast_expression*)(X)), ast_value) && (X)->hasvalue && ((X)->cvq == CV_CONST) && \ ((ast_expression*)(X))->vtype != TYPE_FUNCTION) if (fid + 1 < vec_size(sy->out)) ++paramcount; for (i = 0; i < paramcount; ++i) { if (!fold_can_1((ast_value*)sy->out[fid + 1 + i].out)) { fold = false; break; } } /* * All is well which ends well, if we make it into here we can ignore the * intrinsic call and just evaluate it i.e constant fold it. */ if (fold && ast_istype(fun, ast_value) && ((ast_value*)fun)->intrinsic) { ast_expression **exprs = NULL; ast_expression *foldval = NULL; for (i = 0; i < paramcount; i++) vec_push(exprs, sy->out[fid+1 + i].out); if (!(foldval = intrin_fold(parser->intrin, (ast_value*)fun, exprs))) { vec_free(exprs); goto fold_leave; } /* * Blub: what sorts of unreffing and resizing of * sy->out should I be doing here? */ sy->out[fid] = syexp(foldval->node.context, foldval); vec_shrinkby(sy->out, paramcount); vec_free(exprs); return true; } fold_leave: call = ast_call_new(sy->ops[vec_size(sy->ops)].ctx, fun); if (!call) return false; if (fid+1 + paramcount != vec_size(sy->out)) { parseerror(parser, "internal error: parameter count mismatch: (%lu+1+%lu), %lu", (unsigned long)fid, (unsigned long)paramcount, (unsigned long)vec_size(sy->out)); return false; } for (i = 0; i < paramcount; ++i) call->params.push_back(sy->out[fid+1 + i].out); vec_shrinkby(sy->out, paramcount); (void)!ast_call_check_types(call, parser->function->vtype->expression.varparam); if (parser->max_param_count < paramcount) parser->max_param_count = paramcount; if (ast_istype(fun, ast_value)) { funval = (ast_value*)fun; if ((fun->flags & AST_FLAG_VARIADIC) && !(/*funval->cvq == CV_CONST && */ funval->hasvalue && funval->constval.vfunc->builtin)) { call->va_count = (ast_expression*)fold_constgen_float(parser->fold, (qcfloat_t)paramcount, false); } } /* overwrite fid, the function, with a call */ sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call); if (fun->vtype != TYPE_FUNCTION) { parseerror(parser, "not a function (%s)", type_name[fun->vtype]); return false; } if (!fun->next) { parseerror(parser, "could not determine function return type"); return false; } else { ast_value *fval = (ast_istype(fun, ast_value) ? ((ast_value*)fun) : NULL); if (fun->flags & AST_FLAG_DEPRECATED) { if (!fval) { return !parsewarning(parser, WARN_DEPRECATED, "call to function (which is marked deprecated)\n", "-> it has been declared here: %s:%i", ast_ctx(fun).file, ast_ctx(fun).line); } if (!fval->desc) { return !parsewarning(parser, WARN_DEPRECATED, "call to `%s` (which is marked deprecated)\n" "-> `%s` declared here: %s:%i", fval->name, fval->name, ast_ctx(fun).file, ast_ctx(fun).line); } return !parsewarning(parser, WARN_DEPRECATED, "call to `%s` (deprecated: %s)\n" "-> `%s` declared here: %s:%i", fval->name, fval->desc, fval->name, ast_ctx(fun).file, ast_ctx(fun).line); } if (fun->params.size() != paramcount && !((fun->flags & AST_FLAG_VARIADIC) && fun->params.size() < paramcount)) { const char *fewmany = (fun->params.size() > paramcount) ? "few" : "many"; if (fval) return !parsewarning(parser, WARN_INVALID_PARAMETER_COUNT, "too %s parameters for call to %s: expected %i, got %i\n" " -> `%s` has been declared here: %s:%i", fewmany, fval->name, (int)fun->params.size(), (int)paramcount, fval->name, ast_ctx(fun).file, (int)ast_ctx(fun).line); else return !parsewarning(parser, WARN_INVALID_PARAMETER_COUNT, "too %s parameters for function call: expected %i, got %i\n" " -> it has been declared here: %s:%i", fewmany, (int)fun->params.size(), (int)paramcount, ast_ctx(fun).file, (int)ast_ctx(fun).line); } } return true; } static bool parser_close_paren(parser_t *parser, shunt *sy) { if (!vec_size(sy->ops)) { parseerror(parser, "unmatched closing paren"); return false; } while (vec_size(sy->ops)) { if (vec_last(sy->ops).isparen) { if (vec_last(sy->paren) == PAREN_FUNC) { vec_pop(sy->paren); if (!parser_close_call(parser, sy)) return false; break; } if (vec_last(sy->paren) == PAREN_EXPR) { vec_pop(sy->paren); if (!vec_size(sy->out)) { compile_error(vec_last(sy->ops).ctx, "empty paren expression"); vec_shrinkby(sy->ops, 1); return false; } vec_shrinkby(sy->ops, 1); break; } if (vec_last(sy->paren) == PAREN_INDEX) { vec_pop(sy->paren); /* pop off the parenthesis */ vec_shrinkby(sy->ops, 1); /* then apply the index operator */ if (!parser_sy_apply_operator(parser, sy)) return false; break; } if (vec_last(sy->paren) == PAREN_TERNARY1) { vec_last(sy->paren) = PAREN_TERNARY2; /* pop off the parenthesis */ vec_shrinkby(sy->ops, 1); break; } compile_error(vec_last(sy->ops).ctx, "invalid parenthesis"); return false; } if (!parser_sy_apply_operator(parser, sy)) return false; } return true; } static void parser_reclassify_token(parser_t *parser) { size_t i; if (parser->tok >= TOKEN_START) return; for (i = 0; i < operator_count; ++i) { if (!strcmp(parser_tokval(parser), operators[i].op)) { parser->tok = TOKEN_OPERATOR; return; } } } static ast_expression* parse_vararg_do(parser_t *parser) { ast_expression *idx, *out; ast_value *typevar; ast_value *funtype = parser->function->vtype; lex_ctx_t ctx = parser_ctx(parser); if (!parser->function->varargs) { parseerror(parser, "function has no variable argument list"); return NULL; } if (!parser_next(parser) || parser->tok != '(') { parseerror(parser, "expected parameter index and type in parenthesis"); return NULL; } if (!parser_next(parser)) { parseerror(parser, "error parsing parameter index"); return NULL; } idx = parse_expression_leave(parser, true, false, false); if (!idx) return NULL; if (parser->tok != ',') { if (parser->tok != ')') { ast_unref(idx); parseerror(parser, "expected comma after parameter index"); return NULL; } /* vararg piping: ...(start) */ out = (ast_expression*)ast_argpipe_new(ctx, idx); return out; } if (!parser_next(parser) || (parser->tok != TOKEN_IDENT && parser->tok != TOKEN_TYPENAME)) { ast_unref(idx); parseerror(parser, "expected typename for vararg"); return NULL; } typevar = parse_typename(parser, NULL, NULL, NULL); if (!typevar) { ast_unref(idx); return NULL; } if (parser->tok != ')') { ast_unref(idx); ast_delete(typevar); parseerror(parser, "expected closing paren"); return NULL; } if (funtype->expression.varparam && !ast_compare_type((ast_expression*)typevar, (ast_expression*)funtype->expression.varparam)) { char ty1[1024]; char ty2[1024]; ast_type_to_string((ast_expression*)typevar, ty1, sizeof(ty1)); ast_type_to_string((ast_expression*)funtype->expression.varparam, ty2, sizeof(ty2)); compile_error(ast_ctx(typevar), "function was declared to take varargs of type `%s`, requested type is: %s", ty2, ty1); } out = (ast_expression*)ast_array_index_new(ctx, (ast_expression*)(parser->function->varargs), idx); ast_type_adopt(out, typevar); ast_delete(typevar); return out; } static ast_expression* parse_vararg(parser_t *parser) { bool old_noops = parser->lex->flags.noops; ast_expression *out; parser->lex->flags.noops = true; out = parse_vararg_do(parser); parser->lex->flags.noops = old_noops; return out; } /* not to be exposed */ bool ftepp_predef_exists(const char *name); static bool parse_sya_operand(parser_t *parser, shunt *sy, bool with_labels) { if (OPTS_FLAG(TRANSLATABLE_STRINGS) && parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "_")) { /* a translatable string */ ast_value *val; parser->lex->flags.noops = true; if (!parser_next(parser) || parser->tok != '(') { parseerror(parser, "use _(\"string\") to create a translatable string constant"); return false; } parser->lex->flags.noops = false; if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) { parseerror(parser, "expected a constant string in translatable-string extension"); return false; } val = (ast_value*)fold_constgen_string(parser->fold, parser_tokval(parser), true); if (!val) return false; vec_push(sy->out, syexp(parser_ctx(parser), (ast_expression*)val)); if (!parser_next(parser) || parser->tok != ')') { parseerror(parser, "expected closing paren after translatable string"); return false; } return true; } else if (parser->tok == TOKEN_DOTS) { ast_expression *va; if (!OPTS_FLAG(VARIADIC_ARGS)) { parseerror(parser, "cannot access varargs (try -fvariadic-args)"); return false; } va = parse_vararg(parser); if (!va) return false; vec_push(sy->out, syexp(parser_ctx(parser), va)); return true; } else if (parser->tok == TOKEN_FLOATCONST) { ast_expression *val = fold_constgen_float(parser->fold, (parser_token(parser)->constval.f), false); if (!val) return false; vec_push(sy->out, syexp(parser_ctx(parser), val)); return true; } else if (parser->tok == TOKEN_INTCONST || parser->tok == TOKEN_CHARCONST) { ast_expression *val = fold_constgen_float(parser->fold, (qcfloat_t)(parser_token(parser)->constval.i), false); if (!val) return false; vec_push(sy->out, syexp(parser_ctx(parser), val)); return true; } else if (parser->tok == TOKEN_STRINGCONST) { ast_expression *val = fold_constgen_string(parser->fold, parser_tokval(parser), false); if (!val) return false; vec_push(sy->out, syexp(parser_ctx(parser), val)); return true; } else if (parser->tok == TOKEN_VECTORCONST) { ast_expression *val = fold_constgen_vector(parser->fold, parser_token(parser)->constval.v); if (!val) return false; vec_push(sy->out, syexp(parser_ctx(parser), val)); return true; } else if (parser->tok == TOKEN_IDENT) { const char *ctoken = parser_tokval(parser); ast_expression *prev = vec_size(sy->out) ? vec_last(sy->out).out : NULL; ast_expression *var; /* a_vector.{x,y,z} */ if (!vec_size(sy->ops) || !vec_last(sy->ops).etype || operators[vec_last(sy->ops).etype-1].id != opid1('.')) { /* When adding more intrinsics, fix the above condition */ prev = NULL; } if (prev && prev->vtype == TYPE_VECTOR && ctoken[0] >= 'x' && ctoken[0] <= 'z' && !ctoken[1]) { var = (ast_expression*)parser->const_vec[ctoken[0]-'x']; } else { var = parser_find_var(parser, parser_tokval(parser)); if (!var) var = parser_find_field(parser, parser_tokval(parser)); } if (!var && with_labels) { var = (ast_expression*)parser_find_label(parser, parser_tokval(parser)); if (!with_labels) { ast_label *lbl = ast_label_new(parser_ctx(parser), parser_tokval(parser), true); var = (ast_expression*)lbl; parser->labels.push_back(lbl); } } if (!var && !strcmp(parser_tokval(parser), "__FUNC__")) var = (ast_expression*)fold_constgen_string(parser->fold, parser->function->name, false); if (!var) { /* * now we try for the real intrinsic hashtable. If the string * begins with __builtin, we simply skip past it, otherwise we * use the identifier as is. */ if (!strncmp(parser_tokval(parser), "__builtin_", 10)) { var = intrin_func(parser->intrin, parser_tokval(parser)); } /* * Try it again, intrin_func deals with the alias method as well * the first one masks for __builtin though, we emit warning here. */ if (!var) { if ((var = intrin_func(parser->intrin, parser_tokval(parser)))) { (void)!compile_warning( parser_ctx(parser), WARN_BUILTINS, "using implicitly defined builtin `__builtin_%s' for `%s'", parser_tokval(parser), parser_tokval(parser) ); } } if (!var) { /* * sometimes people use preprocessing predefs without enabling them * i've done this thousands of times already myself. Lets check for * it in the predef table. And diagnose it better :) */ if (!OPTS_FLAG(FTEPP_PREDEFS) && ftepp_predef_exists(parser_tokval(parser))) { parseerror(parser, "unexpected identifier: %s (use -fftepp-predef to enable pre-defined macros)", parser_tokval(parser)); return false; } parseerror(parser, "unexpected identifier: %s", parser_tokval(parser)); return false; } } else { if (ast_istype(var, ast_value)) { ((ast_value*)var)->uses++; } else if (ast_istype(var, ast_member)) { ast_member *mem = (ast_member*)var; if (ast_istype(mem->owner, ast_value)) ((ast_value*)(mem->owner))->uses++; } } vec_push(sy->out, syexp(parser_ctx(parser), var)); return true; } parseerror(parser, "unexpected token `%s`", parser_tokval(parser)); return false; } static ast_expression* parse_expression_leave(parser_t *parser, bool stopatcomma, bool truthvalue, bool with_labels) { ast_expression *expr = NULL; shunt sy; size_t i; bool wantop = false; /* only warn once about an assignment in a truth value because the current code * would trigger twice on: if(a = b && ...), once for the if-truth-value, once for the && part */ bool warn_parenthesis = true; /* count the parens because an if starts with one, so the * end of a condition is an unmatched closing paren */ int ternaries = 0; memset(&sy, 0, sizeof(sy)); parser->lex->flags.noops = false; parser_reclassify_token(parser); while (true) { if (parser->tok == TOKEN_TYPENAME) { parseerror(parser, "unexpected typename `%s`", parser_tokval(parser)); goto onerr; } if (parser->tok == TOKEN_OPERATOR) { /* classify the operator */ const oper_info *op; const oper_info *olast = NULL; size_t o; for (o = 0; o < operator_count; ++o) { if (((!(operators[o].flags & OP_PREFIX) == !!wantop)) && /* !(operators[o].flags & OP_SUFFIX) && / * remove this */ !strcmp(parser_tokval(parser), operators[o].op)) { break; } } if (o == operator_count) { compile_error(parser_ctx(parser), "unexpected operator: %s", parser_tokval(parser)); goto onerr; } /* found an operator */ op = &operators[o]; /* when declaring variables, a comma starts a new variable */ if (op->id == opid1(',') && !vec_size(sy.paren) && stopatcomma) { /* fixup the token */ parser->tok = ','; break; } /* a colon without a pervious question mark cannot be a ternary */ if (!ternaries && op->id == opid2(':','?')) { parser->tok = ':'; break; } if (op->id == opid1(',')) { if (vec_size(sy.paren) && vec_last(sy.paren) == PAREN_TERNARY2) { (void)!parsewarning(parser, WARN_TERNARY_PRECEDENCE, "suggesting parenthesis around ternary expression"); } } if (vec_size(sy.ops) && !vec_last(sy.ops).isparen) olast = &operators[vec_last(sy.ops).etype-1]; /* first only apply higher precedences, assoc_left+equal comes after we warn about precedence rules */ while (olast && op->prec < olast->prec) { if (!parser_sy_apply_operator(parser, &sy)) goto onerr; if (vec_size(sy.ops) && !vec_last(sy.ops).isparen) olast = &operators[vec_last(sy.ops).etype-1]; else olast = NULL; } #define IsAssignOp(x) (\ (x) == opid1('=') || \ (x) == opid2('+','=') || \ (x) == opid2('-','=') || \ (x) == opid2('*','=') || \ (x) == opid2('/','=') || \ (x) == opid2('%','=') || \ (x) == opid2('&','=') || \ (x) == opid2('|','=') || \ (x) == opid3('&','~','=') \ ) if (warn_parenthesis) { if ( (olast && IsAssignOp(olast->id) && (op->id == opid2('&','&') || op->id == opid2('|','|'))) || (olast && IsAssignOp(op->id) && (olast->id == opid2('&','&') || olast->id == opid2('|','|'))) || (truthvalue && !vec_size(sy.paren) && IsAssignOp(op->id)) ) { (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around assignment used as truth value"); warn_parenthesis = false; } if (olast && olast->id != op->id) { if ((op->id == opid1('&') || op->id == opid1('|') || op->id == opid1('^')) && (olast->id == opid1('&') || olast->id == opid1('|') || olast->id == opid1('^'))) { (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around bitwise operations"); warn_parenthesis = false; } else if ((op->id == opid2('&','&') || op->id == opid2('|','|')) && (olast->id == opid2('&','&') || olast->id == opid2('|','|'))) { (void)!parsewarning(parser, WARN_PARENTHESIS, "suggesting parenthesis around logical operations"); warn_parenthesis = false; } } } while (olast && ( (op->prec < olast->prec) || (op->assoc == ASSOC_LEFT && op->prec <= olast->prec) ) ) { if (!parser_sy_apply_operator(parser, &sy)) goto onerr; if (vec_size(sy.ops) && !vec_last(sy.ops).isparen) olast = &operators[vec_last(sy.ops).etype-1]; else olast = NULL; } if (op->id == opid1('(')) { if (wantop) { size_t sycount = vec_size(sy.out); /* we expected an operator, this is the function-call operator */ vec_push(sy.paren, PAREN_FUNC); vec_push(sy.ops, syparen(parser_ctx(parser), sycount-1)); vec_push(sy.argc, 0); } else { vec_push(sy.paren, PAREN_EXPR); vec_push(sy.ops, syparen(parser_ctx(parser), 0)); } wantop = false; } else if (op->id == opid1('[')) { if (!wantop) { parseerror(parser, "unexpected array subscript"); goto onerr; } vec_push(sy.paren, PAREN_INDEX); /* push both the operator and the paren, this makes life easier */ vec_push(sy.ops, syop(parser_ctx(parser), op)); vec_push(sy.ops, syparen(parser_ctx(parser), 0)); wantop = false; } else if (op->id == opid2('?',':')) { vec_push(sy.ops, syop(parser_ctx(parser), op)); vec_push(sy.ops, syparen(parser_ctx(parser), 0)); wantop = false; ++ternaries; vec_push(sy.paren, PAREN_TERNARY1); } else if (op->id == opid2(':','?')) { if (!vec_size(sy.paren)) { parseerror(parser, "unexpected colon outside ternary expression (missing parenthesis?)"); goto onerr; } if (vec_last(sy.paren) != PAREN_TERNARY1) { parseerror(parser, "unexpected colon outside ternary expression (missing parenthesis?)"); goto onerr; } if (!parser_close_paren(parser, &sy)) goto onerr; vec_push(sy.ops, syop(parser_ctx(parser), op)); wantop = false; --ternaries; } else { vec_push(sy.ops, syop(parser_ctx(parser), op)); wantop = !!(op->flags & OP_SUFFIX); } } else if (parser->tok == ')') { while (vec_size(sy.paren) && vec_last(sy.paren) == PAREN_TERNARY2) { if (!parser_sy_apply_operator(parser, &sy)) goto onerr; } if (!vec_size(sy.paren)) break; if (wantop) { if (vec_last(sy.paren) == PAREN_TERNARY1) { parseerror(parser, "mismatched parentheses (closing paren in ternary expression?)"); goto onerr; } if (!parser_close_paren(parser, &sy)) goto onerr; } else { /* must be a function call without parameters */ if (vec_last(sy.paren) != PAREN_FUNC) { parseerror(parser, "closing paren in invalid position"); goto onerr; } if (!parser_close_paren(parser, &sy)) goto onerr; } wantop = true; } else if (parser->tok == '(') { parseerror(parser, "internal error: '(' should be classified as operator"); goto onerr; } else if (parser->tok == '[') { parseerror(parser, "internal error: '[' should be classified as operator"); goto onerr; } else if (parser->tok == ']') { while (vec_size(sy.paren) && vec_last(sy.paren) == PAREN_TERNARY2) { if (!parser_sy_apply_operator(parser, &sy)) goto onerr; } if (!vec_size(sy.paren)) break; if (vec_last(sy.paren) != PAREN_INDEX) { parseerror(parser, "mismatched parentheses, unexpected ']'"); goto onerr; } if (!parser_close_paren(parser, &sy)) goto onerr; wantop = true; } else if (!wantop) { if (!parse_sya_operand(parser, &sy, with_labels)) goto onerr; wantop = true; } else { /* in this case we might want to allow constant string concatenation */ bool concatenated = false; if (parser->tok == TOKEN_STRINGCONST && vec_size(sy.out)) { ast_expression *lexpr = vec_last(sy.out).out; if (ast_istype(lexpr, ast_value)) { ast_value *last = (ast_value*)lexpr; if (last->isimm == true && last->cvq == CV_CONST && last->hasvalue && last->expression.vtype == TYPE_STRING) { char *newstr = NULL; util_asprintf(&newstr, "%s%s", last->constval.vstring, parser_tokval(parser)); vec_last(sy.out).out = (ast_expression*)fold_constgen_string(parser->fold, newstr, false); mem_d(newstr); concatenated = true; } } } if (!concatenated) { parseerror(parser, "expected operator or end of statement"); goto onerr; } } if (!parser_next(parser)) { goto onerr; } if (parser->tok == ';' || ((!vec_size(sy.paren) || (vec_size(sy.paren) == 1 && vec_last(sy.paren) == PAREN_TERNARY2)) && (parser->tok == ']' || parser->tok == ')' || parser->tok == '}'))) { break; } } while (vec_size(sy.ops)) { if (!parser_sy_apply_operator(parser, &sy)) goto onerr; } parser->lex->flags.noops = true; if (vec_size(sy.out) != 1) { parseerror(parser, "expression expected"); expr = NULL; } else expr = sy.out[0].out; vec_free(sy.out); vec_free(sy.ops); if (vec_size(sy.paren)) { parseerror(parser, "internal error: vec_size(sy.paren) = %lu", (unsigned long)vec_size(sy.paren)); return NULL; } vec_free(sy.paren); vec_free(sy.argc); return expr; onerr: parser->lex->flags.noops = true; for (i = 0; i < vec_size(sy.out); ++i) { if (sy.out[i].out) ast_unref(sy.out[i].out); } vec_free(sy.out); vec_free(sy.ops); vec_free(sy.paren); vec_free(sy.argc); return NULL; } static ast_expression* parse_expression(parser_t *parser, bool stopatcomma, bool with_labels) { ast_expression *e = parse_expression_leave(parser, stopatcomma, false, with_labels); if (!e) return NULL; if (parser->tok != ';') { parseerror(parser, "semicolon expected after expression"); ast_unref(e); return NULL; } if (!parser_next(parser)) { ast_unref(e); return NULL; } return e; } static void parser_enterblock(parser_t *parser) { vec_push(parser->variables, util_htnew(PARSER_HT_SIZE)); vec_push(parser->_blocklocals, vec_size(parser->_locals)); vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE)); vec_push(parser->_blocktypedefs, vec_size(parser->_typedefs)); vec_push(parser->_block_ctx, parser_ctx(parser)); } static bool parser_leaveblock(parser_t *parser) { bool rv = true; size_t locals, typedefs; if (vec_size(parser->variables) <= PARSER_HT_LOCALS) { parseerror(parser, "internal error: parser_leaveblock with no block"); return false; } util_htdel(vec_last(parser->variables)); vec_pop(parser->variables); if (!vec_size(parser->_blocklocals)) { parseerror(parser, "internal error: parser_leaveblock with no block (2)"); return false; } locals = vec_last(parser->_blocklocals); vec_pop(parser->_blocklocals); while (vec_size(parser->_locals) != locals) { ast_expression *e = vec_last(parser->_locals); ast_value *v = (ast_value*)e; vec_pop(parser->_locals); if (ast_istype(e, ast_value) && !v->uses) { if (compile_warning(ast_ctx(v), WARN_UNUSED_VARIABLE, "unused variable: `%s`", v->name)) rv = false; } } typedefs = vec_last(parser->_blocktypedefs); while (vec_size(parser->_typedefs) != typedefs) { ast_delete(vec_last(parser->_typedefs)); vec_pop(parser->_typedefs); } util_htdel(vec_last(parser->typedefs)); vec_pop(parser->typedefs); vec_pop(parser->_block_ctx); return rv; } static void parser_addlocal(parser_t *parser, const char *name, ast_expression *e) { vec_push(parser->_locals, e); util_htset(vec_last(parser->variables), name, (void*)e); } static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e) { parser->globals.push_back(e); util_htset(parser->htglobals, name, e); } static ast_expression* process_condition(parser_t *parser, ast_expression *cond, bool *_ifnot) { bool ifnot = false; ast_unary *unary; ast_expression *prev; if (cond->vtype == TYPE_VOID || cond->vtype >= TYPE_VARIANT) { char ty[1024]; ast_type_to_string(cond, ty, sizeof(ty)); compile_error(ast_ctx(cond), "invalid type for if() condition: %s", ty); } if (OPTS_FLAG(FALSE_EMPTY_STRINGS) && cond->vtype == TYPE_STRING) { prev = cond; cond = (ast_expression*)ast_unary_new(ast_ctx(cond), INSTR_NOT_S, cond); if (!cond) { ast_unref(prev); parseerror(parser, "internal error: failed to process condition"); return NULL; } ifnot = !ifnot; } else if (OPTS_FLAG(CORRECT_LOGIC) && cond->vtype == TYPE_VECTOR) { /* vector types need to be cast to true booleans */ ast_binary *bin = (ast_binary*)cond; if (!OPTS_FLAG(PERL_LOGIC) || !ast_istype(cond, ast_binary) || !(bin->op == INSTR_AND || bin->op == INSTR_OR)) { /* in perl-logic, AND and OR take care of the -fcorrect-logic */ prev = cond; cond = (ast_expression*)ast_unary_new(ast_ctx(cond), INSTR_NOT_V, cond); if (!cond) { ast_unref(prev); parseerror(parser, "internal error: failed to process condition"); return NULL; } ifnot = !ifnot; } } unary = (ast_unary*)cond; /* ast_istype dereferences cond, should test here for safety */ while (cond && ast_istype(cond, ast_unary) && unary->op == INSTR_NOT_F) { cond = unary->operand; unary->operand = NULL; ast_delete(unary); ifnot = !ifnot; unary = (ast_unary*)cond; } if (!cond) parseerror(parser, "internal error: failed to process condition"); if (ifnot) *_ifnot = !*_ifnot; return cond; } static bool parse_if(parser_t *parser, ast_block *block, ast_expression **out) { ast_ifthen *ifthen; ast_expression *cond, *ontrue = NULL, *onfalse = NULL; bool ifnot = false; lex_ctx_t ctx = parser_ctx(parser); (void)block; /* not touching */ /* skip the 'if', parse an optional 'not' and check for an opening paren */ if (!parser_next(parser)) { parseerror(parser, "expected condition or 'not'"); return false; } if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "not")) { ifnot = true; if (!parser_next(parser)) { parseerror(parser, "expected condition in parenthesis"); return false; } } if (parser->tok != '(') { parseerror(parser, "expected 'if' condition in parenthesis"); return false; } /* parse into the expression */ if (!parser_next(parser)) { parseerror(parser, "expected 'if' condition after opening paren"); return false; } /* parse the condition */ cond = parse_expression_leave(parser, false, true, false); if (!cond) return false; /* closing paren */ if (parser->tok != ')') { parseerror(parser, "expected closing paren after 'if' condition"); ast_unref(cond); return false; } /* parse into the 'then' branch */ if (!parser_next(parser)) { parseerror(parser, "expected statement for on-true branch of 'if'"); ast_unref(cond); return false; } if (!parse_statement_or_block(parser, &ontrue)) { ast_unref(cond); return false; } if (!ontrue) ontrue = (ast_expression*)ast_block_new(parser_ctx(parser)); /* check for an else */ if (!strcmp(parser_tokval(parser), "else")) { /* parse into the 'else' branch */ if (!parser_next(parser)) { parseerror(parser, "expected on-false branch after 'else'"); ast_delete(ontrue); ast_unref(cond); return false; } if (!parse_statement_or_block(parser, &onfalse)) { ast_delete(ontrue); ast_unref(cond); return false; } } cond = process_condition(parser, cond, &ifnot); if (!cond) { if (ontrue) ast_delete(ontrue); if (onfalse) ast_delete(onfalse); return false; } if (ifnot) ifthen = ast_ifthen_new(ctx, cond, onfalse, ontrue); else ifthen = ast_ifthen_new(ctx, cond, ontrue, onfalse); *out = (ast_expression*)ifthen; return true; } static bool parse_while_go(parser_t *parser, ast_block *block, ast_expression **out); static bool parse_while(parser_t *parser, ast_block *block, ast_expression **out) { bool rv; char *label = NULL; /* skip the 'while' and get the body */ if (!parser_next(parser)) { if (OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "expected loop label or 'while' condition in parenthesis"); else parseerror(parser, "expected 'while' condition in parenthesis"); return false; } if (parser->tok == ':') { if (!OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "labeled loops not activated, try using -floop-labels"); if (!parser_next(parser) || parser->tok != TOKEN_IDENT) { parseerror(parser, "expected loop label"); return false; } label = util_strdup(parser_tokval(parser)); if (!parser_next(parser)) { mem_d(label); parseerror(parser, "expected 'while' condition in parenthesis"); return false; } } if (parser->tok != '(') { parseerror(parser, "expected 'while' condition in parenthesis"); return false; } parser->breaks.push_back(label); parser->continues.push_back(label); rv = parse_while_go(parser, block, out); if (label) mem_d(label); if (parser->breaks.back() != label || parser->continues.back() != label) { parseerror(parser, "internal error: label stack corrupted"); rv = false; ast_delete(*out); *out = NULL; } else { parser->breaks.pop_back(); parser->continues.pop_back(); } return rv; } static bool parse_while_go(parser_t *parser, ast_block *block, ast_expression **out) { ast_loop *aloop; ast_expression *cond, *ontrue; bool ifnot = false; lex_ctx_t ctx = parser_ctx(parser); (void)block; /* not touching */ /* parse into the expression */ if (!parser_next(parser)) { parseerror(parser, "expected 'while' condition after opening paren"); return false; } /* parse the condition */ cond = parse_expression_leave(parser, false, true, false); if (!cond) return false; /* closing paren */ if (parser->tok != ')') { parseerror(parser, "expected closing paren after 'while' condition"); ast_unref(cond); return false; } /* parse into the 'then' branch */ if (!parser_next(parser)) { parseerror(parser, "expected while-loop body"); ast_unref(cond); return false; } if (!parse_statement_or_block(parser, &ontrue)) { ast_unref(cond); return false; } cond = process_condition(parser, cond, &ifnot); if (!cond) { ast_unref(ontrue); return false; } aloop = ast_loop_new(ctx, NULL, cond, ifnot, NULL, false, NULL, ontrue); *out = (ast_expression*)aloop; return true; } static bool parse_dowhile_go(parser_t *parser, ast_block *block, ast_expression **out); static bool parse_dowhile(parser_t *parser, ast_block *block, ast_expression **out) { bool rv; char *label = NULL; /* skip the 'do' and get the body */ if (!parser_next(parser)) { if (OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "expected loop label or body"); else parseerror(parser, "expected loop body"); return false; } if (parser->tok == ':') { if (!OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "labeled loops not activated, try using -floop-labels"); if (!parser_next(parser) || parser->tok != TOKEN_IDENT) { parseerror(parser, "expected loop label"); return false; } label = util_strdup(parser_tokval(parser)); if (!parser_next(parser)) { mem_d(label); parseerror(parser, "expected loop body"); return false; } } parser->breaks.push_back(label); parser->continues.push_back(label); rv = parse_dowhile_go(parser, block, out); if (label) mem_d(label); if (parser->breaks.back() != label || parser->continues.back() != label) { parseerror(parser, "internal error: label stack corrupted"); rv = false; /* * Test for NULL otherwise ast_delete dereferences null pointer * and boom. */ if (*out) ast_delete(*out); *out = NULL; } else { parser->breaks.pop_back(); parser->continues.pop_back(); } return rv; } static bool parse_dowhile_go(parser_t *parser, ast_block *block, ast_expression **out) { ast_loop *aloop; ast_expression *cond, *ontrue; bool ifnot = false; lex_ctx_t ctx = parser_ctx(parser); (void)block; /* not touching */ if (!parse_statement_or_block(parser, &ontrue)) return false; /* expect the "while" */ if (parser->tok != TOKEN_KEYWORD || strcmp(parser_tokval(parser), "while")) { parseerror(parser, "expected 'while' and condition"); ast_delete(ontrue); return false; } /* skip the 'while' and check for opening paren */ if (!parser_next(parser) || parser->tok != '(') { parseerror(parser, "expected 'while' condition in parenthesis"); ast_delete(ontrue); return false; } /* parse into the expression */ if (!parser_next(parser)) { parseerror(parser, "expected 'while' condition after opening paren"); ast_delete(ontrue); return false; } /* parse the condition */ cond = parse_expression_leave(parser, false, true, false); if (!cond) return false; /* closing paren */ if (parser->tok != ')') { parseerror(parser, "expected closing paren after 'while' condition"); ast_delete(ontrue); ast_unref(cond); return false; } /* parse on */ if (!parser_next(parser) || parser->tok != ';') { parseerror(parser, "expected semicolon after condition"); ast_delete(ontrue); ast_unref(cond); return false; } if (!parser_next(parser)) { parseerror(parser, "parse error"); ast_delete(ontrue); ast_unref(cond); return false; } cond = process_condition(parser, cond, &ifnot); if (!cond) { ast_delete(ontrue); return false; } aloop = ast_loop_new(ctx, NULL, NULL, false, cond, ifnot, NULL, ontrue); *out = (ast_expression*)aloop; return true; } static bool parse_for_go(parser_t *parser, ast_block *block, ast_expression **out); static bool parse_for(parser_t *parser, ast_block *block, ast_expression **out) { bool rv; char *label = NULL; /* skip the 'for' and check for opening paren */ if (!parser_next(parser)) { if (OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "expected loop label or 'for' expressions in parenthesis"); else parseerror(parser, "expected 'for' expressions in parenthesis"); return false; } if (parser->tok == ':') { if (!OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "labeled loops not activated, try using -floop-labels"); if (!parser_next(parser) || parser->tok != TOKEN_IDENT) { parseerror(parser, "expected loop label"); return false; } label = util_strdup(parser_tokval(parser)); if (!parser_next(parser)) { mem_d(label); parseerror(parser, "expected 'for' expressions in parenthesis"); return false; } } if (parser->tok != '(') { parseerror(parser, "expected 'for' expressions in parenthesis"); return false; } parser->breaks.push_back(label); parser->continues.push_back(label); rv = parse_for_go(parser, block, out); if (label) mem_d(label); if (parser->breaks.back() != label || parser->continues.back() != label) { parseerror(parser, "internal error: label stack corrupted"); rv = false; ast_delete(*out); *out = NULL; } else { parser->breaks.pop_back(); parser->continues.pop_back(); } return rv; } static bool parse_for_go(parser_t *parser, ast_block *block, ast_expression **out) { ast_loop *aloop; ast_expression *initexpr, *cond, *increment, *ontrue; ast_value *typevar; bool ifnot = false; lex_ctx_t ctx = parser_ctx(parser); parser_enterblock(parser); initexpr = NULL; cond = NULL; increment = NULL; ontrue = NULL; /* parse into the expression */ if (!parser_next(parser)) { parseerror(parser, "expected 'for' initializer after opening paren"); goto onerr; } typevar = NULL; if (parser->tok == TOKEN_IDENT) typevar = parser_find_typedef(parser, parser_tokval(parser), 0); if (typevar || parser->tok == TOKEN_TYPENAME) { if (!parse_variable(parser, block, true, CV_VAR, typevar, false, false, 0, NULL)) goto onerr; } else if (parser->tok != ';') { initexpr = parse_expression_leave(parser, false, false, false); if (!initexpr) goto onerr; /* move on to condition */ if (parser->tok != ';') { parseerror(parser, "expected semicolon after for-loop initializer"); goto onerr; } if (!parser_next(parser)) { parseerror(parser, "expected for-loop condition"); goto onerr; } } /* parse the condition */ if (parser->tok != ';') { cond = parse_expression_leave(parser, false, true, false); if (!cond) goto onerr; } /* move on to incrementor */ if (parser->tok != ';') { parseerror(parser, "expected semicolon after for-loop initializer"); goto onerr; } if (!parser_next(parser)) { parseerror(parser, "expected for-loop condition"); goto onerr; } /* parse the incrementor */ if (parser->tok != ')') { lex_ctx_t condctx = parser_ctx(parser); increment = parse_expression_leave(parser, false, false, false); if (!increment) goto onerr; if (!ast_side_effects(increment)) { if (compile_warning(condctx, WARN_EFFECTLESS_STATEMENT, "statement has no effect")) goto onerr; } } /* closing paren */ if (parser->tok != ')') { parseerror(parser, "expected closing paren after 'for-loop' incrementor"); goto onerr; } /* parse into the 'then' branch */ if (!parser_next(parser)) { parseerror(parser, "expected for-loop body"); goto onerr; } if (!parse_statement_or_block(parser, &ontrue)) goto onerr; if (cond) { cond = process_condition(parser, cond, &ifnot); if (!cond) goto onerr; } aloop = ast_loop_new(ctx, initexpr, cond, ifnot, NULL, false, increment, ontrue); *out = (ast_expression*)aloop; if (!parser_leaveblock(parser)) { ast_delete(aloop); return false; } return true; onerr: if (initexpr) ast_unref(initexpr); if (cond) ast_unref(cond); if (increment) ast_unref(increment); (void)!parser_leaveblock(parser); return false; } static bool parse_return(parser_t *parser, ast_block *block, ast_expression **out) { ast_expression *exp = NULL; ast_expression *var = NULL; ast_return *ret = NULL; ast_value *retval = parser->function->return_value; ast_value *expected = parser->function->vtype; lex_ctx_t ctx = parser_ctx(parser); (void)block; /* not touching */ if (!parser_next(parser)) { parseerror(parser, "expected return expression"); return false; } /* return assignments */ if (parser->tok == '=') { if (!OPTS_FLAG(RETURN_ASSIGNMENTS)) { parseerror(parser, "return assignments not activated, try using -freturn-assigments"); return false; } if (type_store_instr[expected->expression.next->vtype] == VINSTR_END) { char ty1[1024]; ast_type_to_string(expected->expression.next, ty1, sizeof(ty1)); parseerror(parser, "invalid return type: `%s'", ty1); return false; } if (!parser_next(parser)) { parseerror(parser, "expected return assignment expression"); return false; } if (!(exp = parse_expression_leave(parser, false, false, false))) return false; /* prepare the return value */ if (!retval) { retval = ast_value_new(ctx, "#LOCAL_RETURN", TYPE_VOID); ast_type_adopt(retval, expected->expression.next); parser->function->return_value = retval; } if (!ast_compare_type(exp, (ast_expression*)retval)) { char ty1[1024], ty2[1024]; ast_type_to_string(exp, ty1, sizeof(ty1)); ast_type_to_string(&retval->expression, ty2, sizeof(ty2)); parseerror(parser, "invalid type for return value: `%s', expected `%s'", ty1, ty2); } /* store to 'return' local variable */ var = (ast_expression*)ast_store_new( ctx, type_store_instr[expected->expression.next->vtype], (ast_expression*)retval, exp); if (!var) { ast_unref(exp); return false; } if (parser->tok != ';') parseerror(parser, "missing semicolon after return assignment"); else if (!parser_next(parser)) parseerror(parser, "parse error after return assignment"); *out = var; return true; } if (parser->tok != ';') { exp = parse_expression(parser, false, false); if (!exp) return false; if (exp->vtype != TYPE_NIL && exp->vtype != ((ast_expression*)expected)->next->vtype) { parseerror(parser, "return with invalid expression"); } ret = ast_return_new(ctx, exp); if (!ret) { ast_unref(exp); return false; } } else { if (!parser_next(parser)) parseerror(parser, "parse error"); if (!retval && expected->expression.next->vtype != TYPE_VOID) { (void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value"); } ret = ast_return_new(ctx, (ast_expression*)retval); } *out = (ast_expression*)ret; return true; } static bool parse_break_continue(parser_t *parser, ast_block *block, ast_expression **out, bool is_continue) { size_t i; unsigned int levels = 0; lex_ctx_t ctx = parser_ctx(parser); auto &loops = (is_continue ? parser->continues : parser->breaks); (void)block; /* not touching */ if (!parser_next(parser)) { parseerror(parser, "expected semicolon or loop label"); return false; } if (loops.empty()) { if (is_continue) parseerror(parser, "`continue` can only be used inside loops"); else parseerror(parser, "`break` can only be used inside loops or switches"); } if (parser->tok == TOKEN_IDENT) { if (!OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "labeled loops not activated, try using -floop-labels"); i = loops.size(); while (i--) { if (loops[i] && !strcmp(loops[i], parser_tokval(parser))) break; if (!i) { parseerror(parser, "no such loop to %s: `%s`", (is_continue ? "continue" : "break out of"), parser_tokval(parser)); return false; } ++levels; } if (!parser_next(parser)) { parseerror(parser, "expected semicolon"); return false; } } if (parser->tok != ';') { parseerror(parser, "expected semicolon"); return false; } if (!parser_next(parser)) parseerror(parser, "parse error"); *out = (ast_expression*)ast_breakcont_new(ctx, is_continue, levels); return true; } /* returns true when it was a variable qualifier, false otherwise! * on error, cvq is set to CV_WRONG */ struct attribute_t { const char *name; size_t flag; }; static bool parse_qualifiers(parser_t *parser, bool with_local, int *cvq, bool *noref, bool *is_static, uint32_t *_flags, char **message) { bool had_const = false; bool had_var = false; bool had_noref = false; bool had_attrib = false; bool had_static = false; uint32_t flags = 0; static attribute_t attributes[] = { { "noreturn", AST_FLAG_NORETURN }, { "inline", AST_FLAG_INLINE }, { "eraseable", AST_FLAG_ERASEABLE }, { "accumulate", AST_FLAG_ACCUMULATE }, { "last", AST_FLAG_FINAL_DECL } }; *cvq = CV_NONE; for (;;) { size_t i; if (parser->tok == TOKEN_ATTRIBUTE_OPEN) { had_attrib = true; /* parse an attribute */ if (!parser_next(parser)) { parseerror(parser, "expected attribute after `[[`"); *cvq = CV_WRONG; return false; } for (i = 0; i < GMQCC_ARRAY_COUNT(attributes); i++) { if (!strcmp(parser_tokval(parser), attributes[i].name)) { flags |= attributes[i].flag; if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) { parseerror(parser, "`%s` attribute has no parameters, expected `]]`", attributes[i].name); *cvq = CV_WRONG; return false; } break; } } if (i != GMQCC_ARRAY_COUNT(attributes)) goto leave; if (!strcmp(parser_tokval(parser), "noref")) { had_noref = true; if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) { parseerror(parser, "`noref` attribute has no parameters, expected `]]`"); *cvq = CV_WRONG; return false; } } else if (!strcmp(parser_tokval(parser), "alias") && !(flags & AST_FLAG_ALIAS)) { flags |= AST_FLAG_ALIAS; *message = NULL; if (!parser_next(parser)) { parseerror(parser, "parse error in attribute"); goto argerr; } if (parser->tok == '(') { if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) { parseerror(parser, "`alias` attribute missing parameter"); goto argerr; } *message = util_strdup(parser_tokval(parser)); if (!parser_next(parser)) { parseerror(parser, "parse error in attribute"); goto argerr; } if (parser->tok != ')') { parseerror(parser, "`alias` attribute expected `)` after parameter"); goto argerr; } if (!parser_next(parser)) { parseerror(parser, "parse error in attribute"); goto argerr; } } if (parser->tok != TOKEN_ATTRIBUTE_CLOSE) { parseerror(parser, "`alias` attribute expected `]]`"); goto argerr; } } else if (!strcmp(parser_tokval(parser), "deprecated") && !(flags & AST_FLAG_DEPRECATED)) { flags |= AST_FLAG_DEPRECATED; *message = NULL; if (!parser_next(parser)) { parseerror(parser, "parse error in attribute"); goto argerr; } if (parser->tok == '(') { if (!parser_next(parser) || parser->tok != TOKEN_STRINGCONST) { parseerror(parser, "`deprecated` attribute missing parameter"); goto argerr; } *message = util_strdup(parser_tokval(parser)); if (!parser_next(parser)) { parseerror(parser, "parse error in attribute"); goto argerr; } if(parser->tok != ')') { parseerror(parser, "`deprecated` attribute expected `)` after parameter"); goto argerr; } if (!parser_next(parser)) { parseerror(parser, "parse error in attribute"); goto argerr; } } /* no message */ if (parser->tok != TOKEN_ATTRIBUTE_CLOSE) { parseerror(parser, "`deprecated` attribute expected `]]`"); argerr: /* ugly */ if (*message) mem_d(*message); *message = NULL; *cvq = CV_WRONG; return false; } } else if (!strcmp(parser_tokval(parser), "coverage") && !(flags & AST_FLAG_COVERAGE)) { flags |= AST_FLAG_COVERAGE; if (!parser_next(parser)) { error_in_coverage: parseerror(parser, "parse error in coverage attribute"); *cvq = CV_WRONG; return false; } if (parser->tok == '(') { if (!parser_next(parser)) { bad_coverage_arg: parseerror(parser, "invalid parameter for coverage() attribute\n" "valid are: block"); *cvq = CV_WRONG; return false; } if (parser->tok != ')') { do { if (parser->tok != TOKEN_IDENT) goto bad_coverage_arg; if (!strcmp(parser_tokval(parser), "block")) flags |= AST_FLAG_BLOCK_COVERAGE; else if (!strcmp(parser_tokval(parser), "none")) flags &= ~(AST_FLAG_COVERAGE_MASK); else goto bad_coverage_arg; if (!parser_next(parser)) goto error_in_coverage; if (parser->tok == ',') { if (!parser_next(parser)) goto error_in_coverage; } } while (parser->tok != ')'); } if (parser->tok != ')' || !parser_next(parser)) goto error_in_coverage; } else { /* without parameter [[coverage]] equals [[coverage(block)]] */ flags |= AST_FLAG_BLOCK_COVERAGE; } } else { /* Skip tokens until we hit a ]] */ (void)!parsewarning(parser, WARN_UNKNOWN_ATTRIBUTE, "unknown attribute starting with `%s`", parser_tokval(parser)); while (parser->tok != TOKEN_ATTRIBUTE_CLOSE) { if (!parser_next(parser)) { parseerror(parser, "error inside attribute"); *cvq = CV_WRONG; return false; } } } } else if (with_local && !strcmp(parser_tokval(parser), "static")) had_static = true; else if (!strcmp(parser_tokval(parser), "const")) had_const = true; else if (!strcmp(parser_tokval(parser), "var")) had_var = true; else if (with_local && !strcmp(parser_tokval(parser), "local")) had_var = true; else if (!strcmp(parser_tokval(parser), "noref")) had_noref = true; else if (!had_const && !had_var && !had_noref && !had_attrib && !had_static && !flags) { return false; } else break; leave: if (!parser_next(parser)) goto onerr; } if (had_const) *cvq = CV_CONST; else if (had_var) *cvq = CV_VAR; else *cvq = CV_NONE; *noref = had_noref; *is_static = had_static; *_flags = flags; return true; onerr: parseerror(parser, "parse error after variable qualifier"); *cvq = CV_WRONG; return true; } static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression **out); static bool parse_switch(parser_t *parser, ast_block *block, ast_expression **out) { bool rv; char *label = NULL; /* skip the 'while' and get the body */ if (!parser_next(parser)) { if (OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "expected loop label or 'switch' operand in parenthesis"); else parseerror(parser, "expected 'switch' operand in parenthesis"); return false; } if (parser->tok == ':') { if (!OPTS_FLAG(LOOP_LABELS)) parseerror(parser, "labeled loops not activated, try using -floop-labels"); if (!parser_next(parser) || parser->tok != TOKEN_IDENT) { parseerror(parser, "expected loop label"); return false; } label = util_strdup(parser_tokval(parser)); if (!parser_next(parser)) { mem_d(label); parseerror(parser, "expected 'switch' operand in parenthesis"); return false; } } if (parser->tok != '(') { parseerror(parser, "expected 'switch' operand in parenthesis"); return false; } parser->breaks.push_back(label); rv = parse_switch_go(parser, block, out); if (label) mem_d(label); if (parser->breaks.back() != label) { parseerror(parser, "internal error: label stack corrupted"); rv = false; ast_delete(*out); *out = NULL; } else { parser->breaks.pop_back(); } return rv; } static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression **out) { ast_expression *operand; ast_value *opval; ast_value *typevar; ast_switch *switchnode; ast_switch_case swcase; int cvq; bool noref, is_static; uint32_t qflags = 0; lex_ctx_t ctx = parser_ctx(parser); (void)block; /* not touching */ (void)opval; /* parse into the expression */ if (!parser_next(parser)) { parseerror(parser, "expected switch operand"); return false; } /* parse the operand */ operand = parse_expression_leave(parser, false, false, false); if (!operand) return false; switchnode = ast_switch_new(ctx, operand); /* closing paren */ if (parser->tok != ')') { ast_delete(switchnode); parseerror(parser, "expected closing paren after 'switch' operand"); return false; } /* parse over the opening paren */ if (!parser_next(parser) || parser->tok != '{') { ast_delete(switchnode); parseerror(parser, "expected list of cases"); return false; } if (!parser_next(parser)) { ast_delete(switchnode); parseerror(parser, "expected 'case' or 'default'"); return false; } /* new block; allow some variables to be declared here */ parser_enterblock(parser); while (true) { typevar = NULL; if (parser->tok == TOKEN_IDENT) typevar = parser_find_typedef(parser, parser_tokval(parser), 0); if (typevar || parser->tok == TOKEN_TYPENAME) { if (!parse_variable(parser, block, true, CV_NONE, typevar, false, false, 0, NULL)) { ast_delete(switchnode); return false; } continue; } if (parse_qualifiers(parser, true, &cvq, &noref, &is_static, &qflags, NULL)) { if (cvq == CV_WRONG) { ast_delete(switchnode); return false; } if (!parse_variable(parser, block, true, cvq, NULL, noref, is_static, qflags, NULL)) { ast_delete(switchnode); return false; } continue; } break; } /* case list! */ while (parser->tok != '}') { ast_block *caseblock; if (!strcmp(parser_tokval(parser), "case")) { if (!parser_next(parser)) { ast_delete(switchnode); parseerror(parser, "expected expression for case"); return false; } swcase.value = parse_expression_leave(parser, false, false, false); if (!swcase.value) { ast_delete(switchnode); parseerror(parser, "expected expression for case"); return false; } if (!OPTS_FLAG(RELAXED_SWITCH)) { if (!ast_istype(swcase.value, ast_value)) { /* || ((ast_value*)swcase.value)->cvq != CV_CONST) { */ parseerror(parser, "case on non-constant values need to be explicitly enabled via -frelaxed-switch"); ast_unref(operand); return false; } } } else if (!strcmp(parser_tokval(parser), "default")) { swcase.value = NULL; if (!parser_next(parser)) { ast_delete(switchnode); parseerror(parser, "expected colon"); return false; } } else { ast_delete(switchnode); parseerror(parser, "expected 'case' or 'default'"); return false; } /* Now the colon and body */ if (parser->tok != ':') { if (swcase.value) ast_unref(swcase.value); ast_delete(switchnode); parseerror(parser, "expected colon"); return false; } if (!parser_next(parser)) { if (swcase.value) ast_unref(swcase.value); ast_delete(switchnode); parseerror(parser, "expected statements or case"); return false; } caseblock = ast_block_new(parser_ctx(parser)); if (!caseblock) { if (swcase.value) ast_unref(swcase.value); ast_delete(switchnode); return false; } swcase.code = (ast_expression*)caseblock; switchnode->cases.push_back(swcase); while (true) { ast_expression *expr; if (parser->tok == '}') break; if (parser->tok == TOKEN_KEYWORD) { if (!strcmp(parser_tokval(parser), "case") || !strcmp(parser_tokval(parser), "default")) { break; } } if (!parse_statement(parser, caseblock, &expr, true)) { ast_delete(switchnode); return false; } if (!expr) continue; if (!ast_block_add_expr(caseblock, expr)) { ast_delete(switchnode); return false; } } } parser_leaveblock(parser); /* closing paren */ if (parser->tok != '}') { ast_delete(switchnode); parseerror(parser, "expected closing paren of case list"); return false; } if (!parser_next(parser)) { ast_delete(switchnode); parseerror(parser, "parse error after switch"); return false; } *out = (ast_expression*)switchnode; return true; } /* parse computed goto sides */ static ast_expression *parse_goto_computed(parser_t *parser, ast_expression **side) { ast_expression *on_true; ast_expression *on_false; ast_expression *cond; if (!*side) return NULL; if (ast_istype(*side, ast_ternary)) { ast_ternary *tern = (ast_ternary*)*side; on_true = parse_goto_computed(parser, &tern->on_true); on_false = parse_goto_computed(parser, &tern->on_false); if (!on_true || !on_false) { parseerror(parser, "expected label or expression in ternary"); if (on_true) ast_unref(on_true); if (on_false) ast_unref(on_false); return NULL; } cond = tern->cond; tern->cond = NULL; ast_delete(tern); *side = NULL; return (ast_expression*)ast_ifthen_new(parser_ctx(parser), cond, on_true, on_false); } else if (ast_istype(*side, ast_label)) { ast_goto *gt = ast_goto_new(parser_ctx(parser), ((ast_label*)*side)->name); ast_goto_set_label(gt, ((ast_label*)*side)); *side = NULL; return (ast_expression*)gt; } return NULL; } static bool parse_goto(parser_t *parser, ast_expression **out) { ast_goto *gt = NULL; ast_expression *lbl; if (!parser_next(parser)) return false; if (parser->tok != TOKEN_IDENT) { ast_expression *expression; /* could be an expression i.e computed goto :-) */ if (parser->tok != '(') { parseerror(parser, "expected label name after `goto`"); return false; } /* failed to parse expression for goto */ if (!(expression = parse_expression(parser, false, true)) || !(*out = parse_goto_computed(parser, &expression))) { parseerror(parser, "invalid goto expression"); if(expression) ast_unref(expression); return false; } return true; } /* not computed goto */ gt = ast_goto_new(parser_ctx(parser), parser_tokval(parser)); lbl = parser_find_label(parser, gt->name); if (lbl) { if (!ast_istype(lbl, ast_label)) { parseerror(parser, "internal error: label is not an ast_label"); ast_delete(gt); return false; } ast_goto_set_label(gt, (ast_label*)lbl); } else parser->gotos.push_back(gt); if (!parser_next(parser) || parser->tok != ';') { parseerror(parser, "semicolon expected after goto label"); return false; } if (!parser_next(parser)) { parseerror(parser, "parse error after goto"); return false; } *out = (ast_expression*)gt; return true; } static bool parse_skipwhite(parser_t *parser) { do { if (!parser_next(parser)) return false; } while (parser->tok == TOKEN_WHITE && parser->tok < TOKEN_ERROR); return parser->tok < TOKEN_ERROR; } static bool parse_eol(parser_t *parser) { if (!parse_skipwhite(parser)) return false; return parser->tok == TOKEN_EOL; } static bool parse_pragma_do(parser_t *parser) { if (!parser_next(parser) || parser->tok != TOKEN_IDENT || strcmp(parser_tokval(parser), "pragma")) { parseerror(parser, "expected `pragma` keyword after `#`, got `%s`", parser_tokval(parser)); return false; } if (!parse_skipwhite(parser) || parser->tok != TOKEN_IDENT) { parseerror(parser, "expected pragma, got `%s`", parser_tokval(parser)); return false; } if (!strcmp(parser_tokval(parser), "noref")) { if (!parse_skipwhite(parser) || parser->tok != TOKEN_INTCONST) { parseerror(parser, "`noref` pragma requires an argument: 0 or 1"); return false; } parser->noref = !!parser_token(parser)->constval.i; if (!parse_eol(parser)) { parseerror(parser, "parse error after `noref` pragma"); return false; } } else { (void)!parsewarning(parser, WARN_UNKNOWN_PRAGMAS, "ignoring #pragma %s", parser_tokval(parser)); /* skip to eol */ while (!parse_eol(parser)) { parser_next(parser); } return true; } return true; } static bool parse_pragma(parser_t *parser) { bool rv; parser->lex->flags.preprocessing = true; parser->lex->flags.mergelines = true; rv = parse_pragma_do(parser); if (parser->tok != TOKEN_EOL) { parseerror(parser, "junk after pragma"); rv = false; } parser->lex->flags.preprocessing = false; parser->lex->flags.mergelines = false; if (!parser_next(parser)) { parseerror(parser, "parse error after pragma"); rv = false; } return rv; } static bool parse_statement(parser_t *parser, ast_block *block, ast_expression **out, bool allow_cases) { bool noref, is_static; int cvq = CV_NONE; uint32_t qflags = 0; ast_value *typevar = NULL; char *vstring = NULL; *out = NULL; if (parser->tok == TOKEN_IDENT) typevar = parser_find_typedef(parser, parser_tokval(parser), 0); if (typevar || parser->tok == TOKEN_TYPENAME || parser->tok == '.' || parser->tok == TOKEN_DOTS) { /* local variable */ if (!block) { parseerror(parser, "cannot declare a variable from here"); return false; } if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) { if (parsewarning(parser, WARN_EXTENSIONS, "missing 'local' keyword when declaring a local variable")) return false; } if (!parse_variable(parser, block, false, CV_NONE, typevar, false, false, 0, NULL)) return false; return true; } else if (parse_qualifiers(parser, !!block, &cvq, &noref, &is_static, &qflags, &vstring)) { if (cvq == CV_WRONG) return false; return parse_variable(parser, block, false, cvq, NULL, noref, is_static, qflags, vstring); } else if (parser->tok == TOKEN_KEYWORD) { if (!strcmp(parser_tokval(parser), "__builtin_debug_printtype")) { char ty[1024]; ast_value *tdef; if (!parser_next(parser)) { parseerror(parser, "parse error after __builtin_debug_printtype"); return false; } if (parser->tok == TOKEN_IDENT && (tdef = parser_find_typedef(parser, parser_tokval(parser), 0))) { ast_type_to_string((ast_expression*)tdef, ty, sizeof(ty)); con_out("__builtin_debug_printtype: `%s`=`%s`\n", tdef->name, ty); if (!parser_next(parser)) { parseerror(parser, "parse error after __builtin_debug_printtype typename argument"); return false; } } else { if (!parse_statement(parser, block, out, allow_cases)) return false; if (!*out) con_out("__builtin_debug_printtype: got no output node\n"); else { ast_type_to_string(*out, ty, sizeof(ty)); con_out("__builtin_debug_printtype: `%s`\n", ty); } } return true; } else if (!strcmp(parser_tokval(parser), "return")) { return parse_return(parser, block, out); } else if (!strcmp(parser_tokval(parser), "if")) { return parse_if(parser, block, out); } else if (!strcmp(parser_tokval(parser), "while")) { return parse_while(parser, block, out); } else if (!strcmp(parser_tokval(parser), "do")) { return parse_dowhile(parser, block, out); } else if (!strcmp(parser_tokval(parser), "for")) { if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) { if (parsewarning(parser, WARN_EXTENSIONS, "for loops are not recognized in the original Quake C standard, to enable try an alternate standard --std=?")) return false; } return parse_for(parser, block, out); } else if (!strcmp(parser_tokval(parser), "break")) { return parse_break_continue(parser, block, out, false); } else if (!strcmp(parser_tokval(parser), "continue")) { return parse_break_continue(parser, block, out, true); } else if (!strcmp(parser_tokval(parser), "switch")) { return parse_switch(parser, block, out); } else if (!strcmp(parser_tokval(parser), "case") || !strcmp(parser_tokval(parser), "default")) { if (!allow_cases) { parseerror(parser, "unexpected 'case' label"); return false; } return true; } else if (!strcmp(parser_tokval(parser), "goto")) { return parse_goto(parser, out); } else if (!strcmp(parser_tokval(parser), "typedef")) { if (!parser_next(parser)) { parseerror(parser, "expected type definition after 'typedef'"); return false; } return parse_typedef(parser); } parseerror(parser, "Unexpected keyword: `%s'", parser_tokval(parser)); return false; } else if (parser->tok == '{') { ast_block *inner; inner = parse_block(parser); if (!inner) return false; *out = (ast_expression*)inner; return true; } else if (parser->tok == ':') { size_t i; ast_label *label; if (!parser_next(parser)) { parseerror(parser, "expected label name"); return false; } if (parser->tok != TOKEN_IDENT) { parseerror(parser, "label must be an identifier"); return false; } label = (ast_label*)parser_find_label(parser, parser_tokval(parser)); if (label) { if (!label->undefined) { parseerror(parser, "label `%s` already defined", label->name); return false; } label->undefined = false; } else { label = ast_label_new(parser_ctx(parser), parser_tokval(parser), false); parser->labels.push_back(label); } *out = (ast_expression*)label; if (!parser_next(parser)) { parseerror(parser, "parse error after label"); return false; } for (i = 0; i < parser->gotos.size(); ++i) { if (!strcmp(parser->gotos[i]->name, label->name)) { ast_goto_set_label(parser->gotos[i], label); parser->gotos.erase(parser->gotos.begin() + i); --i; } } return true; } else if (parser->tok == ';') { if (!parser_next(parser)) { parseerror(parser, "parse error after empty statement"); return false; } return true; } else { lex_ctx_t ctx = parser_ctx(parser); ast_expression *exp = parse_expression(parser, false, false); if (!exp) return false; *out = exp; if (!ast_side_effects(exp)) { if (compile_warning(ctx, WARN_EFFECTLESS_STATEMENT, "statement has no effect")) return false; } return true; } } static bool parse_enum(parser_t *parser) { bool flag = false; bool reverse = false; qcfloat_t num = 0; ast_value **values = NULL; ast_value *var = NULL; ast_value *asvalue; ast_expression *old; if (!parser_next(parser) || (parser->tok != '{' && parser->tok != ':')) { parseerror(parser, "expected `{` or `:` after `enum` keyword"); return false; } /* enumeration attributes (can add more later) */ if (parser->tok == ':') { if (!parser_next(parser) || parser->tok != TOKEN_IDENT){ parseerror(parser, "expected `flag` or `reverse` for enumeration attribute"); return false; } /* attributes? */ if (!strcmp(parser_tokval(parser), "flag")) { num = 1; flag = true; } else if (!strcmp(parser_tokval(parser), "reverse")) { reverse = true; } else { parseerror(parser, "invalid attribute `%s` for enumeration", parser_tokval(parser)); return false; } if (!parser_next(parser) || parser->tok != '{') { parseerror(parser, "expected `{` after enum attribute "); return false; } } while (true) { if (!parser_next(parser) || parser->tok != TOKEN_IDENT) { if (parser->tok == '}') { /* allow an empty enum */ break; } parseerror(parser, "expected identifier or `}`"); goto onerror; } old = parser_find_field(parser, parser_tokval(parser)); if (!old) old = parser_find_global(parser, parser_tokval(parser)); if (old) { parseerror(parser, "value `%s` has already been declared here: %s:%i", parser_tokval(parser), ast_ctx(old).file, ast_ctx(old).line); goto onerror; } var = ast_value_new(parser_ctx(parser), parser_tokval(parser), TYPE_FLOAT); vec_push(values, var); var->cvq = CV_CONST; var->hasvalue = true; /* for flagged enumerations increment in POTs of TWO */ var->constval.vfloat = (flag) ? (num *= 2) : (num ++); parser_addglobal(parser, var->name, (ast_expression*)var); if (!parser_next(parser)) { parseerror(parser, "expected `=`, `}` or comma after identifier"); goto onerror; } if (parser->tok == ',') continue; if (parser->tok == '}') break; if (parser->tok != '=') { parseerror(parser, "expected `=`, `}` or comma after identifier"); goto onerror; } if (!parser_next(parser)) { parseerror(parser, "expected expression after `=`"); goto onerror; } /* We got a value! */ old = parse_expression_leave(parser, true, false, false); asvalue = (ast_value*)old; if (!ast_istype(old, ast_value) || asvalue->cvq != CV_CONST || !asvalue->hasvalue) { compile_error(ast_ctx(var), "constant value or expression expected"); goto onerror; } num = (var->constval.vfloat = asvalue->constval.vfloat) + 1; if (parser->tok == '}') break; if (parser->tok != ',') { parseerror(parser, "expected `}` or comma after expression"); goto onerror; } } /* patch them all (for reversed attribute) */ if (reverse) { size_t i; for (i = 0; i < vec_size(values); i++) values[i]->constval.vfloat = vec_size(values) - i - 1; } if (parser->tok != '}') { parseerror(parser, "internal error: breaking without `}`"); goto onerror; } if (!parser_next(parser) || parser->tok != ';') { parseerror(parser, "expected semicolon after enumeration"); goto onerror; } if (!parser_next(parser)) { parseerror(parser, "parse error after enumeration"); goto onerror; } vec_free(values); return true; onerror: vec_free(values); return false; } static bool parse_block_into(parser_t *parser, ast_block *block) { bool retval = true; parser_enterblock(parser); if (!parser_next(parser)) { /* skip the '{' */ parseerror(parser, "expected function body"); goto cleanup; } while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR) { ast_expression *expr = NULL; if (parser->tok == '}') break; if (!parse_statement(parser, block, &expr, false)) { /* parseerror(parser, "parse error"); */ block = NULL; goto cleanup; } if (!expr) continue; if (!ast_block_add_expr(block, expr)) { ast_delete(block); block = NULL; goto cleanup; } } if (parser->tok != '}') { block = NULL; } else { (void)parser_next(parser); } cleanup: if (!parser_leaveblock(parser)) retval = false; return retval && !!block; } static ast_block* parse_block(parser_t *parser) { ast_block *block; block = ast_block_new(parser_ctx(parser)); if (!block) return NULL; if (!parse_block_into(parser, block)) { ast_block_delete(block); return NULL; } return block; } static bool parse_statement_or_block(parser_t *parser, ast_expression **out) { if (parser->tok == '{') { *out = (ast_expression*)parse_block(parser); return !!*out; } return parse_statement(parser, NULL, out, false); } static bool create_vector_members(ast_value *var, ast_member **me) { size_t i; size_t len = strlen(var->name); for (i = 0; i < 3; ++i) { char *name = (char*)mem_a(len+3); memcpy(name, var->name, len); name[len+0] = '_'; name[len+1] = 'x'+i; name[len+2] = 0; me[i] = ast_member_new(ast_ctx(var), (ast_expression*)var, i, name); mem_d(name); if (!me[i]) break; } if (i == 3) return true; /* unroll */ do { ast_member_delete(me[--i]); } while(i); return false; } static bool parse_function_body(parser_t *parser, ast_value *var) { ast_block *block = NULL; ast_function *func; ast_function *old; ast_expression *framenum = NULL; ast_expression *nextthink = NULL; /* None of the following have to be deleted */ ast_expression *fld_think = NULL, *fld_nextthink = NULL, *fld_frame = NULL; ast_expression *gbl_time = NULL, *gbl_self = NULL; bool has_frame_think; bool retval = true; has_frame_think = false; old = parser->function; if (var->expression.flags & AST_FLAG_ALIAS) { parseerror(parser, "function aliases cannot have bodies"); return false; } if (parser->gotos.size() || parser->labels.size()) { parseerror(parser, "gotos/labels leaking"); return false; } if (!OPTS_FLAG(VARIADIC_ARGS) && var->expression.flags & AST_FLAG_VARIADIC) { if (parsewarning(parser, WARN_VARIADIC_FUNCTION, "variadic function with implementation will not be able to access additional parameters (try -fvariadic-args)")) { return false; } } if (parser->tok == '[') { /* got a frame definition: [ framenum, nextthink ] * this translates to: * self.frame = framenum; * self.nextthink = time + 0.1; * self.think = nextthink; */ nextthink = NULL; fld_think = parser_find_field(parser, "think"); fld_nextthink = parser_find_field(parser, "nextthink"); fld_frame = parser_find_field(parser, "frame"); if (!fld_think || !fld_nextthink || !fld_frame) { parseerror(parser, "cannot use [frame,think] notation without the required fields"); parseerror(parser, "please declare the following entityfields: `frame`, `think`, `nextthink`"); return false; } gbl_time = parser_find_global(parser, "time"); gbl_self = parser_find_global(parser, "self"); if (!gbl_time || !gbl_self) { parseerror(parser, "cannot use [frame,think] notation without the required globals"); parseerror(parser, "please declare the following globals: `time`, `self`"); return false; } if (!parser_next(parser)) return false; framenum = parse_expression_leave(parser, true, false, false); if (!framenum) { parseerror(parser, "expected a framenumber constant in[frame,think] notation"); return false; } if (!ast_istype(framenum, ast_value) || !( (ast_value*)framenum )->hasvalue) { ast_unref(framenum); parseerror(parser, "framenumber in [frame,think] notation must be a constant"); return false; } if (parser->tok != ',') { ast_unref(framenum); parseerror(parser, "expected comma after frame number in [frame,think] notation"); parseerror(parser, "Got a %i\n", parser->tok); return false; } if (!parser_next(parser)) { ast_unref(framenum); return false; } if (parser->tok == TOKEN_IDENT && !parser_find_var(parser, parser_tokval(parser))) { /* qc allows the use of not-yet-declared functions here * - this automatically creates a prototype */ ast_value *thinkfunc; ast_expression *functype = fld_think->next; thinkfunc = ast_value_new(parser_ctx(parser), parser_tokval(parser), functype->vtype); if (!thinkfunc) { /* || !ast_type_adopt(thinkfunc, functype)*/ ast_unref(framenum); parseerror(parser, "failed to create implicit prototype for `%s`", parser_tokval(parser)); return false; } ast_type_adopt(thinkfunc, functype); if (!parser_next(parser)) { ast_unref(framenum); ast_delete(thinkfunc); return false; } parser_addglobal(parser, thinkfunc->name, (ast_expression*)thinkfunc); nextthink = (ast_expression*)thinkfunc; } else { nextthink = parse_expression_leave(parser, true, false, false); if (!nextthink) { ast_unref(framenum); parseerror(parser, "expected a think-function in [frame,think] notation"); return false; } } if (!ast_istype(nextthink, ast_value)) { parseerror(parser, "think-function in [frame,think] notation must be a constant"); retval = false; } if (retval && parser->tok != ']') { parseerror(parser, "expected closing `]` for [frame,think] notation"); retval = false; } if (retval && !parser_next(parser)) { retval = false; } if (retval && parser->tok != '{') { parseerror(parser, "a function body has to be declared after a [frame,think] declaration"); retval = false; } if (!retval) { ast_unref(nextthink); ast_unref(framenum); return false; } has_frame_think = true; } block = ast_block_new(parser_ctx(parser)); if (!block) { parseerror(parser, "failed to allocate block"); if (has_frame_think) { ast_unref(nextthink); ast_unref(framenum); } return false; } if (has_frame_think) { if (!OPTS_FLAG(EMULATE_STATE)) { ast_state *state_op = ast_state_new(parser_ctx(parser), framenum, nextthink); if (!ast_block_add_expr(block, (ast_expression*)state_op)) { parseerror(parser, "failed to generate state op for [frame,think]"); ast_unref(nextthink); ast_unref(framenum); ast_delete(block); return false; } } else { /* emulate OP_STATE in code: */ lex_ctx_t ctx; ast_expression *self_frame; ast_expression *self_nextthink; ast_expression *self_think; ast_expression *time_plus_1; ast_store *store_frame; ast_store *store_nextthink; ast_store *store_think; float frame_delta = 1.0f / (float)OPTS_OPTION_U32(OPTION_STATE_FPS); ctx = parser_ctx(parser); self_frame = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_frame); self_nextthink = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_nextthink); self_think = (ast_expression*)ast_entfield_new(ctx, gbl_self, fld_think); time_plus_1 = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, gbl_time, (ast_expression*)fold_constgen_float(parser->fold, frame_delta, false)); if (!self_frame || !self_nextthink || !self_think || !time_plus_1) { if (self_frame) ast_delete(self_frame); if (self_nextthink) ast_delete(self_nextthink); if (self_think) ast_delete(self_think); if (time_plus_1) ast_delete(time_plus_1); retval = false; } if (retval) { store_frame = ast_store_new(ctx, INSTR_STOREP_F, self_frame, framenum); store_nextthink = ast_store_new(ctx, INSTR_STOREP_F, self_nextthink, time_plus_1); store_think = ast_store_new(ctx, INSTR_STOREP_FNC, self_think, nextthink); if (!store_frame) { ast_delete(self_frame); retval = false; } if (!store_nextthink) { ast_delete(self_nextthink); retval = false; } if (!store_think) { ast_delete(self_think); retval = false; } if (!retval) { if (store_frame) ast_delete(store_frame); if (store_nextthink) ast_delete(store_nextthink); if (store_think) ast_delete(store_think); retval = false; } if (!ast_block_add_expr(block, (ast_expression*)store_frame) || !ast_block_add_expr(block, (ast_expression*)store_nextthink) || !ast_block_add_expr(block, (ast_expression*)store_think)) { retval = false; } } if (!retval) { parseerror(parser, "failed to generate code for [frame,think]"); ast_unref(nextthink); ast_unref(framenum); ast_delete(block); return false; } } } if (var->hasvalue) { if (!(var->expression.flags & AST_FLAG_ACCUMULATE)) { parseerror(parser, "function `%s` declared with multiple bodies", var->name); ast_block_delete(block); goto enderr; } func = var->constval.vfunc; if (!func) { parseerror(parser, "internal error: NULL function: `%s`", var->name); ast_block_delete(block); goto enderr; } } else { func = ast_function_new(ast_ctx(var), var->name, var); if (!func) { parseerror(parser, "failed to allocate function for `%s`", var->name); ast_block_delete(block); goto enderr; } parser->functions.push_back(func); } parser_enterblock(parser); for (auto &it : var->expression.params) { size_t e; ast_member *me[3]; if (it->expression.vtype != TYPE_VECTOR && (it->expression.vtype != TYPE_FIELD || it->expression.next->vtype != TYPE_VECTOR)) { continue; } if (!create_vector_members(it, me)) { ast_block_delete(block); goto enderrfn; } for (e = 0; e < 3; ++e) { parser_addlocal(parser, me[e]->name, (ast_expression*)me[e]); ast_block_collect(block, (ast_expression*)me[e]); } } if (var->argcounter && !func->argc) { ast_value *argc = ast_value_new(ast_ctx(var), var->argcounter, TYPE_FLOAT); parser_addlocal(parser, argc->name, (ast_expression*)argc); func->argc = argc; } if (OPTS_FLAG(VARIADIC_ARGS) && var->expression.flags & AST_FLAG_VARIADIC && !func->varargs) { char name[1024]; ast_value *varargs = ast_value_new(ast_ctx(var), "reserved:va_args", TYPE_ARRAY); varargs->expression.flags |= AST_FLAG_IS_VARARG; varargs->expression.next = (ast_expression*)ast_value_new(ast_ctx(var), NULL, TYPE_VECTOR); varargs->expression.count = 0; util_snprintf(name, sizeof(name), "%s##va##SET", var->name); if (!parser_create_array_setter_proto(parser, varargs, name)) { ast_delete(varargs); ast_block_delete(block); goto enderrfn; } util_snprintf(name, sizeof(name), "%s##va##GET", var->name); if (!parser_create_array_getter_proto(parser, varargs, varargs->expression.next, name)) { ast_delete(varargs); ast_block_delete(block); goto enderrfn; } func->varargs = varargs; func->fixedparams = (ast_value*)fold_constgen_float(parser->fold, var->expression.params.size(), false); } parser->function = func; if (!parse_block_into(parser, block)) { ast_block_delete(block); goto enderrfn; } func->blocks.push_back(block); parser->function = old; if (!parser_leaveblock(parser)) retval = false; if (vec_size(parser->variables) != PARSER_HT_LOCALS) { parseerror(parser, "internal error: local scopes left"); retval = false; } if (parser->tok == ';') return parser_next(parser); else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) parseerror(parser, "missing semicolon after function body (mandatory with -std=qcc)"); return retval; enderrfn: (void)!parser_leaveblock(parser); parser->functions.pop_back(); ast_function_delete(func); var->constval.vfunc = NULL; enderr: parser->function = old; return false; } static ast_expression *array_accessor_split( parser_t *parser, ast_value *array, ast_value *index, size_t middle, ast_expression *left, ast_expression *right ) { ast_ifthen *ifthen; ast_binary *cmp; lex_ctx_t ctx = ast_ctx(array); if (!left || !right) { if (left) ast_delete(left); if (right) ast_delete(right); return NULL; } cmp = ast_binary_new(ctx, INSTR_LT, (ast_expression*)index, (ast_expression*)fold_constgen_float(parser->fold, middle, false)); if (!cmp) { ast_delete(left); ast_delete(right); parseerror(parser, "internal error: failed to create comparison for array setter"); return NULL; } ifthen = ast_ifthen_new(ctx, (ast_expression*)cmp, left, right); if (!ifthen) { ast_delete(cmp); /* will delete left and right */ parseerror(parser, "internal error: failed to create conditional jump for array setter"); return NULL; } return (ast_expression*)ifthen; } static ast_expression *array_setter_node(parser_t *parser, ast_value *array, ast_value *index, ast_value *value, size_t from, size_t afterend) { lex_ctx_t ctx = ast_ctx(array); if (from+1 == afterend) { /* set this value */ ast_block *block; ast_return *ret; ast_array_index *subscript; ast_store *st; int assignop = type_store_instr[value->expression.vtype]; if (value->expression.vtype == TYPE_FIELD && value->expression.next->vtype == TYPE_VECTOR) assignop = INSTR_STORE_V; subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)fold_constgen_float(parser->fold, from, false)); if (!subscript) return NULL; st = ast_store_new(ctx, assignop, (ast_expression*)subscript, (ast_expression*)value); if (!st) { ast_delete(subscript); return NULL; } block = ast_block_new(ctx); if (!block) { ast_delete(st); return NULL; } if (!ast_block_add_expr(block, (ast_expression*)st)) { ast_delete(block); return NULL; } ret = ast_return_new(ctx, NULL); if (!ret) { ast_delete(block); return NULL; } if (!ast_block_add_expr(block, (ast_expression*)ret)) { ast_delete(block); return NULL; } return (ast_expression*)block; } else { ast_expression *left, *right; size_t diff = afterend - from; size_t middle = from + diff/2; left = array_setter_node(parser, array, index, value, from, middle); right = array_setter_node(parser, array, index, value, middle, afterend); return array_accessor_split(parser, array, index, middle, left, right); } } static ast_expression *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) { lex_ctx_t ctx = ast_ctx(array); if (from+1 == afterend) { /* set this value */ ast_block *block; ast_return *ret; ast_entfield *entfield; ast_array_index *subscript; ast_store *st; int assignop = type_storep_instr[value->expression.vtype]; if (value->expression.vtype == TYPE_FIELD && value->expression.next->vtype == TYPE_VECTOR) assignop = INSTR_STOREP_V; subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)fold_constgen_float(parser->fold, from, false)); if (!subscript) return NULL; subscript->expression.next = ast_type_copy(ast_ctx(subscript), (ast_expression*)subscript); subscript->expression.vtype = TYPE_FIELD; entfield = ast_entfield_new_force(ctx, (ast_expression*)entity, (ast_expression*)subscript, (ast_expression*)subscript); if (!entfield) { ast_delete(subscript); return NULL; } st = ast_store_new(ctx, assignop, (ast_expression*)entfield, (ast_expression*)value); if (!st) { ast_delete(entfield); return NULL; } block = ast_block_new(ctx); if (!block) { ast_delete(st); return NULL; } if (!ast_block_add_expr(block, (ast_expression*)st)) { ast_delete(block); return NULL; } ret = ast_return_new(ctx, NULL); if (!ret) { ast_delete(block); return NULL; } if (!ast_block_add_expr(block, (ast_expression*)ret)) { ast_delete(block); return NULL; } return (ast_expression*)block; } else { ast_expression *left, *right; size_t diff = afterend - from; size_t middle = from + diff/2; left = array_field_setter_node(parser, array, entity, index, value, from, middle); right = array_field_setter_node(parser, array, entity, index, value, middle, afterend); return array_accessor_split(parser, array, index, middle, left, right); } } static ast_expression *array_getter_node(parser_t *parser, ast_value *array, ast_value *index, size_t from, size_t afterend) { lex_ctx_t ctx = ast_ctx(array); if (from+1 == afterend) { ast_return *ret; ast_array_index *subscript; subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)fold_constgen_float(parser->fold, from, false)); if (!subscript) return NULL; ret = ast_return_new(ctx, (ast_expression*)subscript); if (!ret) { ast_delete(subscript); return NULL; } return (ast_expression*)ret; } else { ast_expression *left, *right; size_t diff = afterend - from; size_t middle = from + diff/2; left = array_getter_node(parser, array, index, from, middle); right = array_getter_node(parser, array, index, middle, afterend); return array_accessor_split(parser, array, index, middle, left, right); } } static bool parser_create_array_accessor(parser_t *parser, ast_value *array, const char *funcname, ast_value **out) { ast_function *func = NULL; ast_value *fval = NULL; ast_block *body = NULL; fval = ast_value_new(ast_ctx(array), funcname, TYPE_FUNCTION); if (!fval) { parseerror(parser, "failed to create accessor function value"); return false; } fval->expression.flags &= ~(AST_FLAG_COVERAGE_MASK); func = ast_function_new(ast_ctx(array), funcname, fval); if (!func) { ast_delete(fval); parseerror(parser, "failed to create accessor function node"); return false; } body = ast_block_new(ast_ctx(array)); if (!body) { parseerror(parser, "failed to create block for array accessor"); ast_delete(fval); ast_delete(func); return false; } func->blocks.push_back(body); *out = fval; parser->accessors.push_back(fval); return true; } static ast_value* parser_create_array_setter_proto(parser_t *parser, ast_value *array, const char *funcname) { ast_value *index = NULL; ast_value *value = NULL; ast_function *func; ast_value *fval; if (!ast_istype(array->expression.next, ast_value)) { parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type"); return NULL; } if (!parser_create_array_accessor(parser, array, funcname, &fval)) return NULL; func = fval->constval.vfunc; fval->expression.next = (ast_expression*)ast_value_new(ast_ctx(array), "", TYPE_VOID); index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT); value = ast_value_copy((ast_value*)array->expression.next); if (!index || !value) { parseerror(parser, "failed to create locals for array accessor"); goto cleanup; } (void)!ast_value_set_name(value, "value"); /* not important */ fval->expression.params.push_back(index); fval->expression.params.push_back(value); array->setter = fval; return fval; cleanup: if (index) ast_delete(index); if (value) ast_delete(value); ast_delete(func); ast_delete(fval); return NULL; } static bool parser_create_array_setter_impl(parser_t *parser, ast_value *array) { ast_expression *root = NULL; root = array_setter_node(parser, array, array->setter->expression.params[0], array->setter->expression.params[1], 0, array->expression.count); if (!root) { parseerror(parser, "failed to build accessor search tree"); return false; } if (!ast_block_add_expr(array->setter->constval.vfunc->blocks[0], root)) { ast_delete(root); return false; } return true; } static bool parser_create_array_setter(parser_t *parser, ast_value *array, const char *funcname) { if (!parser_create_array_setter_proto(parser, array, funcname)) return false; return parser_create_array_setter_impl(parser, array); } static bool parser_create_array_field_setter(parser_t *parser, ast_value *array, const char *funcname) { ast_expression *root = NULL; ast_value *entity = NULL; ast_value *index = NULL; ast_value *value = NULL; ast_function *func; ast_value *fval; if (!ast_istype(array->expression.next, ast_value)) { parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type"); return false; } if (!parser_create_array_accessor(parser, array, funcname, &fval)) return false; func = fval->constval.vfunc; fval->expression.next = (ast_expression*)ast_value_new(ast_ctx(array), "", TYPE_VOID); entity = ast_value_new(ast_ctx(array), "entity", TYPE_ENTITY); index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT); value = ast_value_copy((ast_value*)array->expression.next); if (!entity || !index || !value) { parseerror(parser, "failed to create locals for array accessor"); goto cleanup; } (void)!ast_value_set_name(value, "value"); /* not important */ fval->expression.params.push_back(entity); fval->expression.params.push_back(index); fval->expression.params.push_back(value); root = array_field_setter_node(parser, array, entity, index, value, 0, array->expression.count); if (!root) { parseerror(parser, "failed to build accessor search tree"); goto cleanup; } array->setter = fval; return ast_block_add_expr(func->blocks[0], root); cleanup: if (entity) ast_delete(entity); if (index) ast_delete(index); if (value) ast_delete(value); if (root) ast_delete(root); ast_delete(func); ast_delete(fval); return false; } static ast_value* parser_create_array_getter_proto(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname) { ast_value *index = NULL; ast_value *fval; ast_function *func; /* NOTE: checking array->expression.next rather than elemtype since * for fields elemtype is a temporary fieldtype. */ if (!ast_istype(array->expression.next, ast_value)) { parseerror(parser, "internal error: array accessor needs to build an ast_value with a copy of the element type"); return NULL; } if (!parser_create_array_accessor(parser, array, funcname, &fval)) return NULL; func = fval->constval.vfunc; fval->expression.next = ast_type_copy(ast_ctx(array), elemtype); index = ast_value_new(ast_ctx(array), "index", TYPE_FLOAT); if (!index) { parseerror(parser, "failed to create locals for array accessor"); goto cleanup; } fval->expression.params.push_back(index); array->getter = fval; return fval; cleanup: if (index) ast_delete(index); ast_delete(func); ast_delete(fval); return NULL; } static bool parser_create_array_getter_impl(parser_t *parser, ast_value *array) { ast_expression *root = NULL; root = array_getter_node(parser, array, array->getter->expression.params[0], 0, array->expression.count); if (!root) { parseerror(parser, "failed to build accessor search tree"); return false; } if (!ast_block_add_expr(array->getter->constval.vfunc->blocks[0], root)) { ast_delete(root); return false; } return true; } static bool parser_create_array_getter(parser_t *parser, ast_value *array, const ast_expression *elemtype, const char *funcname) { if (!parser_create_array_getter_proto(parser, array, elemtype, funcname)) return false; return parser_create_array_getter_impl(parser, array); } static ast_value *parse_parameter_list(parser_t *parser, ast_value *var) { lex_ctx_t ctx = parser_ctx(parser); std::vector params; ast_value *fval; bool first = true; bool variadic = false; ast_value *varparam = NULL; char *argcounter = NULL; /* for the sake of less code we parse-in in this function */ if (!parser_next(parser)) { ast_delete(var); parseerror(parser, "expected parameter list"); return NULL; } /* parse variables until we hit a closing paren */ while (parser->tok != ')') { bool is_varargs = false; if (!first) { /* there must be commas between them */ if (parser->tok != ',') { parseerror(parser, "expected comma or end of parameter list"); goto on_error; } if (!parser_next(parser)) { parseerror(parser, "expected parameter"); goto on_error; } } first = false; ast_value *param = parse_typename(parser, NULL, NULL, &is_varargs); if (!param && !is_varargs) goto on_error; if (is_varargs) { /* '...' indicates a varargs function */ variadic = true; if (parser->tok != ')' && parser->tok != TOKEN_IDENT) { parseerror(parser, "`...` must be the last parameter of a variadic function declaration"); goto on_error; } if (parser->tok == TOKEN_IDENT) { argcounter = util_strdup(parser_tokval(parser)); if (!parser_next(parser) || parser->tok != ')') { parseerror(parser, "`...` must be the last parameter of a variadic function declaration"); goto on_error; } } } else { params.push_back(param); if (param->expression.vtype >= TYPE_VARIANT) { char tname[1024]; /* typename is reserved in C++ */ ast_type_to_string((ast_expression*)param, tname, sizeof(tname)); parseerror(parser, "type not supported as part of a parameter list: %s", tname); goto on_error; } /* type-restricted varargs */ if (parser->tok == TOKEN_DOTS) { variadic = true; varparam = params.back(); params.pop_back(); if (!parser_next(parser) || (parser->tok != ')' && parser->tok != TOKEN_IDENT)) { parseerror(parser, "`...` must be the last parameter of a variadic function declaration"); goto on_error; } if (parser->tok == TOKEN_IDENT) { argcounter = util_strdup(parser_tokval(parser)); ast_value_set_name(param, argcounter); if (!parser_next(parser) || parser->tok != ')') { parseerror(parser, "`...` must be the last parameter of a variadic function declaration"); goto on_error; } } } if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_FTEQCC && param->name[0] == '<') { parseerror(parser, "parameter name omitted"); goto on_error; } } } if (params.size() == 1 && params[0]->expression.vtype == TYPE_VOID) params.clear(); /* sanity check */ if (params.size() > 8 && OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) (void)!parsewarning(parser, WARN_EXTENSIONS, "more than 8 parameters are not supported by this standard"); /* parse-out */ if (!parser_next(parser)) { parseerror(parser, "parse error after typename"); goto on_error; } /* now turn 'var' into a function type */ fval = ast_value_new(ctx, "", TYPE_FUNCTION); fval->expression.next = (ast_expression*)var; if (variadic) fval->expression.flags |= AST_FLAG_VARIADIC; var = fval; var->expression.params = params; var->expression.varparam = (ast_expression*)varparam; var->argcounter = argcounter; return var; on_error: if (argcounter) mem_d(argcounter); if (varparam) ast_delete(varparam); ast_delete(var); for (auto &it : params) ast_delete(it); return NULL; } static ast_value *parse_arraysize(parser_t *parser, ast_value *var) { ast_expression *cexp; ast_value *cval, *tmp; lex_ctx_t ctx; ctx = parser_ctx(parser); if (!parser_next(parser)) { ast_delete(var); parseerror(parser, "expected array-size"); return NULL; } if (parser->tok != ']') { cexp = parse_expression_leave(parser, true, false, false); if (!cexp || !ast_istype(cexp, ast_value)) { if (cexp) ast_unref(cexp); ast_delete(var); parseerror(parser, "expected array-size as constant positive integer"); return NULL; } cval = (ast_value*)cexp; } else { cexp = NULL; cval = NULL; } tmp = ast_value_new(ctx, "", TYPE_ARRAY); tmp->expression.next = (ast_expression*)var; var = tmp; if (cval) { if (cval->expression.vtype == TYPE_INTEGER) tmp->expression.count = cval->constval.vint; else if (cval->expression.vtype == TYPE_FLOAT) tmp->expression.count = cval->constval.vfloat; else { ast_unref(cexp); ast_delete(var); parseerror(parser, "array-size must be a positive integer constant"); return NULL; } ast_unref(cexp); } else { var->expression.count = -1; var->expression.flags |= AST_FLAG_ARRAY_INIT; } if (parser->tok != ']') { ast_delete(var); parseerror(parser, "expected ']' after array-size"); return NULL; } if (!parser_next(parser)) { ast_delete(var); parseerror(parser, "error after parsing array size"); return NULL; } return var; } /* Parse a complete typename. * for single-variables (ie. function parameters or typedefs) storebase should be NULL * but when parsing variables separated by comma * 'storebase' should point to where the base-type should be kept. * The base type makes up every bit of type information which comes *before* the * variable name. * * NOTE: The value must either be named, have a NULL name, or a name starting * with '<'. In the first case, this will be the actual variable or type * name, in the other cases it is assumed that the name will appear * later, and an error is generated otherwise. * * The following will be parsed in its entirety: * void() foo() * The 'basetype' in this case is 'void()' * and if there's a comma after it, say: * void() foo(), bar * then the type-information 'void()' can be stored in 'storebase' */ static ast_value *parse_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef, bool *is_vararg) { ast_value *var, *tmp; lex_ctx_t ctx; const char *name = NULL; bool isfield = false; bool wasarray = false; size_t morefields = 0; bool vararg = (parser->tok == TOKEN_DOTS); ctx = parser_ctx(parser); /* types may start with a dot */ if (parser->tok == '.' || parser->tok == TOKEN_DOTS) { isfield = true; if (parser->tok == TOKEN_DOTS) morefields += 2; /* if we parsed a dot we need a typename now */ if (!parser_next(parser)) { parseerror(parser, "expected typename for field definition"); return NULL; } /* Further dots are handled seperately because they won't be part of the * basetype */ while (true) { if (parser->tok == '.') ++morefields; else if (parser->tok == TOKEN_DOTS) morefields += 3; else break; vararg = false; if (!parser_next(parser)) { parseerror(parser, "expected typename for field definition"); return NULL; } } } if (parser->tok == TOKEN_IDENT) cached_typedef = parser_find_typedef(parser, parser_tokval(parser), 0); if (!cached_typedef && parser->tok != TOKEN_TYPENAME) { if (vararg && is_vararg) { *is_vararg = true; return NULL; } parseerror(parser, "expected typename"); return NULL; } /* generate the basic type value */ if (cached_typedef) { var = ast_value_copy(cached_typedef); ast_value_set_name(var, ""); } else var = ast_value_new(ctx, "", parser_token(parser)->constval.t); for (; morefields; --morefields) { tmp = ast_value_new(ctx, "<.type>", TYPE_FIELD); tmp->expression.next = (ast_expression*)var; var = tmp; } /* do not yet turn into a field - remember: * .void() foo; is a field too * .void()() foo; is a function */ /* parse on */ if (!parser_next(parser)) { ast_delete(var); parseerror(parser, "parse error after typename"); return NULL; } /* an opening paren now starts the parameter-list of a function * this is where original-QC has parameter lists. * We allow a single parameter list here. * Much like fteqcc we don't allow `float()() x` */ if (parser->tok == '(') { var = parse_parameter_list(parser, var); if (!var) return NULL; } /* store the base if requested */ if (storebase) { *storebase = ast_value_copy(var); if (isfield) { tmp = ast_value_new(ctx, "", TYPE_FIELD); tmp->expression.next = (ast_expression*)*storebase; *storebase = tmp; } } /* there may be a name now */ if (parser->tok == TOKEN_IDENT || parser->tok == TOKEN_KEYWORD) { if (!strcmp(parser_tokval(parser), "break")) (void)!parsewarning(parser, WARN_BREAKDEF, "break definition ignored (suggest removing it)"); else if (parser->tok == TOKEN_KEYWORD) goto leave; name = util_strdup(parser_tokval(parser)); /* parse on */ if (!parser_next(parser)) { ast_delete(var); mem_d(name); parseerror(parser, "error after variable or field declaration"); return NULL; } } leave: /* now this may be an array */ if (parser->tok == '[') { wasarray = true; var = parse_arraysize(parser, var); if (!var) { if (name) mem_d(name); return NULL; } } /* This is the point where we can turn it into a field */ if (isfield) { /* turn it into a field if desired */ tmp = ast_value_new(ctx, "", TYPE_FIELD); tmp->expression.next = (ast_expression*)var; var = tmp; } /* now there may be function parens again */ if (parser->tok == '(' && OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) parseerror(parser, "C-style function syntax is not allowed in -std=qcc"); if (parser->tok == '(' && wasarray) parseerror(parser, "arrays as part of a return type is not supported"); while (parser->tok == '(') { var = parse_parameter_list(parser, var); if (!var) { if (name) mem_d(name); return NULL; } } /* finally name it */ if (name) { if (!ast_value_set_name(var, name)) { ast_delete(var); mem_d(name); parseerror(parser, "internal error: failed to set name"); return NULL; } /* free the name, ast_value_set_name duplicates */ mem_d(name); } return var; } static bool parse_typedef(parser_t *parser) { ast_value *typevar, *oldtype; ast_expression *old; typevar = parse_typename(parser, NULL, NULL, NULL); if (!typevar) return false; /* while parsing types, the ast_value's get named '' */ if (!typevar->name || typevar->name[0] == '<') { parseerror(parser, "missing name in typedef"); ast_delete(typevar); return false; } if ( (old = parser_find_var(parser, typevar->name)) ) { parseerror(parser, "cannot define a type with the same name as a variable: %s\n" " -> `%s` has been declared here: %s:%i", typevar->name, ast_ctx(old).file, ast_ctx(old).line); ast_delete(typevar); return false; } if ( (oldtype = parser_find_typedef(parser, typevar->name, vec_last(parser->_blocktypedefs))) ) { parseerror(parser, "type `%s` has already been declared here: %s:%i", typevar->name, ast_ctx(oldtype).file, ast_ctx(oldtype).line); ast_delete(typevar); return false; } vec_push(parser->_typedefs, typevar); util_htset(vec_last(parser->typedefs), typevar->name, typevar); if (parser->tok != ';') { parseerror(parser, "expected semicolon after typedef"); return false; } if (!parser_next(parser)) { parseerror(parser, "parse error after typedef"); return false; } return true; } static const char *cvq_to_str(int cvq) { switch (cvq) { case CV_NONE: return "none"; case CV_VAR: return "`var`"; case CV_CONST: return "`const`"; default: return ""; } } static bool parser_check_qualifiers(parser_t *parser, const ast_value *var, const ast_value *proto) { bool av, ao; if (proto->cvq != var->cvq) { if (!(proto->cvq == CV_CONST && var->cvq == CV_NONE && !OPTS_FLAG(INITIALIZED_NONCONSTANTS) && parser->tok == '=')) { return !parsewarning(parser, WARN_DIFFERENT_QUALIFIERS, "`%s` declared with different qualifiers: %s\n" " -> previous declaration here: %s:%i uses %s", var->name, cvq_to_str(var->cvq), ast_ctx(proto).file, ast_ctx(proto).line, cvq_to_str(proto->cvq)); } } av = (var ->expression.flags & AST_FLAG_NORETURN); ao = (proto->expression.flags & AST_FLAG_NORETURN); if (!av != !ao) { return !parsewarning(parser, WARN_DIFFERENT_ATTRIBUTES, "`%s` declared with different attributes%s\n" " -> previous declaration here: %s:%i", var->name, (av ? ": noreturn" : ""), ast_ctx(proto).file, ast_ctx(proto).line, (ao ? ": noreturn" : "")); } return true; } static bool create_array_accessors(parser_t *parser, ast_value *var) { char name[1024]; util_snprintf(name, sizeof(name), "%s##SET", var->name); if (!parser_create_array_setter(parser, var, name)) return false; util_snprintf(name, sizeof(name), "%s##GET", var->name); if (!parser_create_array_getter(parser, var, var->expression.next, name)) return false; return true; } static bool parse_array(parser_t *parser, ast_value *array) { size_t i; if (array->initlist.size()) { parseerror(parser, "array already initialized elsewhere"); return false; } if (!parser_next(parser)) { parseerror(parser, "parse error in array initializer"); return false; } i = 0; while (parser->tok != '}') { ast_value *v = (ast_value*)parse_expression_leave(parser, true, false, false); if (!v) return false; if (!ast_istype(v, ast_value) || !v->hasvalue || v->cvq != CV_CONST) { ast_unref(v); parseerror(parser, "initializing element must be a compile time constant"); return false; } array->initlist.push_back(v->constval); if (v->expression.vtype == TYPE_STRING) { array->initlist[i].vstring = util_strdupe(array->initlist[i].vstring); ++i; } ast_unref(v); if (parser->tok == '}') break; if (parser->tok != ',' || !parser_next(parser)) { parseerror(parser, "expected comma or '}' in element list"); return false; } } if (!parser_next(parser) || parser->tok != ';') { parseerror(parser, "expected semicolon after initializer, got %s"); return false; } /* if (!parser_next(parser)) { parseerror(parser, "parse error after initializer"); return false; } */ if (array->expression.flags & AST_FLAG_ARRAY_INIT) { if (array->expression.count != (size_t)-1) { parseerror(parser, "array `%s' has already been initialized with %u elements", array->name, (unsigned)array->expression.count); } array->expression.count = array->initlist.size(); if (!create_array_accessors(parser, array)) return false; } return true; } 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) { ast_value *var; ast_value *proto; ast_expression *old; bool was_end; size_t i; ast_value *basetype = NULL; bool retval = true; bool isparam = false; bool isvector = false; bool cleanvar = true; bool wasarray = false; ast_member *me[3] = { NULL, NULL, NULL }; ast_member *last_me[3] = { NULL, NULL, NULL }; if (!localblock && is_static) parseerror(parser, "`static` qualifier is not supported in global scope"); /* get the first complete variable */ var = parse_typename(parser, &basetype, cached_typedef, NULL); if (!var) { if (basetype) ast_delete(basetype); return false; } /* while parsing types, the ast_value's get named '' */ if (!var->name || var->name[0] == '<') { parseerror(parser, "declaration does not declare anything"); if (basetype) ast_delete(basetype); return false; } while (true) { proto = NULL; wasarray = false; /* Part 0: finish the type */ if (parser->tok == '(') { if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) parseerror(parser, "C-style function syntax is not allowed in -std=qcc"); var = parse_parameter_list(parser, var); if (!var) { retval = false; goto cleanup; } } /* we only allow 1-dimensional arrays */ if (parser->tok == '[') { wasarray = true; var = parse_arraysize(parser, var); if (!var) { retval = false; goto cleanup; } } if (parser->tok == '(' && wasarray) { parseerror(parser, "arrays as part of a return type is not supported"); /* we'll still parse the type completely for now */ } /* for functions returning functions */ while (parser->tok == '(') { if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) parseerror(parser, "C-style function syntax is not allowed in -std=qcc"); var = parse_parameter_list(parser, var); if (!var) { retval = false; goto cleanup; } } var->cvq = qualifier; if (qflags & AST_FLAG_COVERAGE) /* specified in QC, drop our default */ var->expression.flags &= ~(AST_FLAG_COVERAGE_MASK); var->expression.flags |= qflags; /* * store the vstring back to var for alias and * deprecation messages. */ if (var->expression.flags & AST_FLAG_DEPRECATED || var->expression.flags & AST_FLAG_ALIAS) var->desc = vstring; if (parser_find_global(parser, var->name) && var->expression.flags & AST_FLAG_ALIAS) { parseerror(parser, "function aliases cannot be forward declared"); retval = false; goto cleanup; } /* Part 1: * check for validity: (end_sys_..., multiple-definitions, prototypes, ...) * Also: if there was a prototype, `var` will be deleted and set to `proto` which * is then filled with the previous definition and the parameter-names replaced. */ if (!strcmp(var->name, "nil")) { if (OPTS_FLAG(UNTYPED_NIL)) { if (!localblock || !OPTS_FLAG(PERMISSIVE)) parseerror(parser, "name `nil` not allowed (try -fpermissive)"); } else (void)!parsewarning(parser, WARN_RESERVED_NAMES, "variable name `nil` is reserved"); } if (!localblock) { /* Deal with end_sys_ vars */ was_end = false; if (!strcmp(var->name, "end_sys_globals")) { var->uses++; parser->crc_globals = parser->globals.size(); was_end = true; } else if (!strcmp(var->name, "end_sys_fields")) { var->uses++; parser->crc_fields = parser->fields.size(); was_end = true; } if (was_end && var->expression.vtype == TYPE_FIELD) { if (parsewarning(parser, WARN_END_SYS_FIELDS, "global '%s' hint should not be a field", parser_tokval(parser))) { retval = false; goto cleanup; } } if (!nofields && var->expression.vtype == TYPE_FIELD) { /* deal with field declarations */ old = parser_find_field(parser, var->name); if (old) { if (parsewarning(parser, WARN_FIELD_REDECLARED, "field `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, (int)ast_ctx(old).line)) { retval = false; goto cleanup; } ast_delete(var); var = NULL; goto skipvar; /* parseerror(parser, "field `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, ast_ctx(old).line); retval = false; goto cleanup; */ } if ((OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC || OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_FTEQCC) && (old = parser_find_global(parser, var->name))) { parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc"); parseerror(parser, "field `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, ast_ctx(old).line); retval = false; goto cleanup; } } else { /* deal with other globals */ old = parser_find_global(parser, var->name); if (old && var->expression.vtype == TYPE_FUNCTION && old->vtype == TYPE_FUNCTION) { /* This is a function which had a prototype */ if (!ast_istype(old, ast_value)) { parseerror(parser, "internal error: prototype is not an ast_value"); retval = false; goto cleanup; } proto = (ast_value*)old; proto->desc = var->desc; if (!ast_compare_type((ast_expression*)proto, (ast_expression*)var)) { parseerror(parser, "conflicting types for `%s`, previous declaration was here: %s:%i", proto->name, ast_ctx(proto).file, ast_ctx(proto).line); retval = false; goto cleanup; } /* we need the new parameter-names */ for (i = 0; i < proto->expression.params.size(); ++i) ast_value_set_name(proto->expression.params[i], var->expression.params[i]->name); if (!parser_check_qualifiers(parser, var, proto)) { retval = false; if (proto->desc) mem_d(proto->desc); proto = NULL; goto cleanup; } proto->expression.flags |= var->expression.flags; ast_delete(var); var = proto; } else { /* other globals */ if (old) { if (parsewarning(parser, WARN_DOUBLE_DECLARATION, "global `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, ast_ctx(old).line)) { retval = false; goto cleanup; } if (old->flags & AST_FLAG_FINAL_DECL) { parseerror(parser, "cannot redeclare variable `%s`, declared final here: %s:%i", var->name, ast_ctx(old).file, ast_ctx(old).line); retval = false; goto cleanup; } proto = (ast_value*)old; if (!ast_istype(old, ast_value)) { parseerror(parser, "internal error: not an ast_value"); retval = false; proto = NULL; goto cleanup; } if (!parser_check_qualifiers(parser, var, proto)) { retval = false; proto = NULL; goto cleanup; } proto->expression.flags |= var->expression.flags; /* copy the context for finals, * so the error can show where it was actually made 'final' */ if (proto->expression.flags & AST_FLAG_FINAL_DECL) ast_ctx(old) = ast_ctx(var); ast_delete(var); var = proto; } if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC && (old = parser_find_field(parser, var->name))) { parseerror(parser, "cannot declare a field and a global of the same name with -std=qcc"); parseerror(parser, "global `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, ast_ctx(old).line); retval = false; goto cleanup; } } } } else /* it's not a global */ { old = parser_find_local(parser, var->name, vec_size(parser->variables)-1, &isparam); if (old && !isparam) { parseerror(parser, "local `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, (int)ast_ctx(old).line); retval = false; goto cleanup; } /* doing this here as the above is just for a single scope */ old = parser_find_local(parser, var->name, 0, &isparam); if (old && isparam) { if (parsewarning(parser, WARN_LOCAL_SHADOWS, "local `%s` is shadowing a parameter", var->name)) { parseerror(parser, "local `%s` already declared here: %s:%i", var->name, ast_ctx(old).file, (int)ast_ctx(old).line); retval = false; goto cleanup; } if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_GMQCC) { ast_delete(var); if (ast_istype(old, ast_value)) var = proto = (ast_value*)old; else { var = NULL; goto skipvar; } } } } /* in a noref section we simply bump the usecount */ if (noref || parser->noref) var->uses++; /* Part 2: * Create the global/local, and deal with vector types. */ if (!proto) { if (var->expression.vtype == TYPE_VECTOR) isvector = true; else if (var->expression.vtype == TYPE_FIELD && var->expression.next->vtype == TYPE_VECTOR) isvector = true; if (isvector) { if (!create_vector_members(var, me)) { retval = false; goto cleanup; } } if (!localblock) { /* deal with global variables, fields, functions */ if (!nofields && var->expression.vtype == TYPE_FIELD && parser->tok != '=') { var->isfield = true; parser->fields.push_back((ast_expression*)var); util_htset(parser->htfields, var->name, var); if (isvector) { for (i = 0; i < 3; ++i) { parser->fields.push_back((ast_expression*)me[i]); util_htset(parser->htfields, me[i]->name, me[i]); } } } else { if (!(var->expression.flags & AST_FLAG_ALIAS)) { parser_addglobal(parser, var->name, (ast_expression*)var); if (isvector) { for (i = 0; i < 3; ++i) { parser_addglobal(parser, me[i]->name, (ast_expression*)me[i]); } } } else { ast_expression *find = parser_find_global(parser, var->desc); if (!find) { compile_error(parser_ctx(parser), "undeclared variable `%s` for alias `%s`", var->desc, var->name); return false; } if (!ast_compare_type((ast_expression*)var, find)) { char ty1[1024]; char ty2[1024]; ast_type_to_string(find, ty1, sizeof(ty1)); ast_type_to_string((ast_expression*)var, ty2, sizeof(ty2)); compile_error(parser_ctx(parser), "incompatible types `%s` and `%s` for alias `%s`", ty1, ty2, var->name ); return false; } util_htset(parser->aliases, var->name, find); /* generate aliases for vector components */ if (isvector) { char *buffer[3]; util_asprintf(&buffer[0], "%s_x", var->desc); util_asprintf(&buffer[1], "%s_y", var->desc); util_asprintf(&buffer[2], "%s_z", var->desc); util_htset(parser->aliases, me[0]->name, parser_find_global(parser, buffer[0])); util_htset(parser->aliases, me[1]->name, parser_find_global(parser, buffer[1])); util_htset(parser->aliases, me[2]->name, parser_find_global(parser, buffer[2])); mem_d(buffer[0]); mem_d(buffer[1]); mem_d(buffer[2]); } } } } else { if (is_static) { /* a static adds itself to be generated like any other global * but is added to the local namespace instead */ char *defname = NULL; size_t prefix_len, ln; size_t sn, sn_size; ln = strlen(parser->function->name); vec_append(defname, ln, parser->function->name); vec_append(defname, 2, "::"); /* remember the length up to here */ prefix_len = vec_size(defname); /* Add it to the local scope */ util_htset(vec_last(parser->variables), var->name, (void*)var); /* now rename the global */ ln = strlen(var->name); vec_append(defname, ln, var->name); /* if a variable of that name already existed, add the * counter value. * The counter is incremented either way. */ sn_size = vec_size(parser->function->static_names); for (sn = 0; sn != sn_size; ++sn) { if (strcmp(parser->function->static_names[sn], var->name) == 0) break; } if (sn != sn_size) { char *num = NULL; int len = util_asprintf(&num, "#%u", parser->function->static_count); vec_append(defname, len, num); mem_d(num); } else vec_push(parser->function->static_names, util_strdup(var->name)); parser->function->static_count++; ast_value_set_name(var, defname); /* push it to the to-be-generated globals */ parser->globals.push_back((ast_expression*)var); /* same game for the vector members */ if (isvector) { for (i = 0; i < 3; ++i) { util_htset(vec_last(parser->variables), me[i]->name, (void*)(me[i])); vec_shrinkto(defname, prefix_len); ln = strlen(me[i]->name); vec_append(defname, ln, me[i]->name); ast_member_set_name(me[i], defname); parser->globals.push_back((ast_expression*)me[i]); } } vec_free(defname); } else { localblock->locals.push_back(var); parser_addlocal(parser, var->name, (ast_expression*)var); if (isvector) { for (i = 0; i < 3; ++i) { parser_addlocal(parser, me[i]->name, (ast_expression*)me[i]); ast_block_collect(localblock, (ast_expression*)me[i]); } } } } } memcpy(last_me, me, sizeof(me)); me[0] = me[1] = me[2] = NULL; cleanvar = false; /* Part 2.2 * deal with arrays */ if (var->expression.vtype == TYPE_ARRAY) { if (var->expression.count != (size_t)-1) { if (!create_array_accessors(parser, var)) goto cleanup; } } else if (!localblock && !nofields && var->expression.vtype == TYPE_FIELD && var->expression.next->vtype == TYPE_ARRAY) { char name[1024]; ast_expression *telem; ast_value *tfield; ast_value *array = (ast_value*)var->expression.next; if (!ast_istype(var->expression.next, ast_value)) { parseerror(parser, "internal error: field element type must be an ast_value"); goto cleanup; } util_snprintf(name, sizeof(name), "%s##SETF", var->name); if (!parser_create_array_field_setter(parser, array, name)) goto cleanup; telem = ast_type_copy(ast_ctx(var), array->expression.next); tfield = ast_value_new(ast_ctx(var), "<.type>", TYPE_FIELD); tfield->expression.next = telem; util_snprintf(name, sizeof(name), "%s##GETFP", var->name); if (!parser_create_array_getter(parser, array, (ast_expression*)tfield, name)) { ast_delete(tfield); goto cleanup; } ast_delete(tfield); } skipvar: if (parser->tok == ';') { ast_delete(basetype); if (!parser_next(parser)) { parseerror(parser, "error after variable declaration"); return false; } return true; } if (parser->tok == ',') goto another; /* if (!var || (!localblock && !nofields && basetype->expression.vtype == TYPE_FIELD)) { */ if (!var) { parseerror(parser, "missing comma or semicolon while parsing variables"); break; } if (localblock && OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) { if (parsewarning(parser, WARN_LOCAL_CONSTANTS, "initializing expression turns variable `%s` into a constant in this standard", var->name) ) { break; } } if (parser->tok != '{' || var->expression.vtype != TYPE_FUNCTION) { if (parser->tok != '=') { parseerror(parser, "missing semicolon or initializer, got: `%s`", parser_tokval(parser)); break; } if (!parser_next(parser)) { parseerror(parser, "error parsing initializer"); break; } } else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) { parseerror(parser, "expected '=' before function body in this standard"); } if (parser->tok == '#') { ast_function *func = NULL; ast_value *number = NULL; float fractional; float integral; int builtin_num; if (localblock) { parseerror(parser, "cannot declare builtins within functions"); break; } if (var->expression.vtype != TYPE_FUNCTION) { parseerror(parser, "unexpected builtin number, '%s' is not a function", var->name); break; } if (!parser_next(parser)) { parseerror(parser, "expected builtin number"); break; } if (OPTS_FLAG(EXPRESSIONS_FOR_BUILTINS)) { number = (ast_value*)parse_expression_leave(parser, true, false, false); if (!number) { parseerror(parser, "builtin number expected"); break; } if (!ast_istype(number, ast_value) || !number->hasvalue || number->cvq != CV_CONST) { ast_unref(number); parseerror(parser, "builtin number must be a compile time constant"); break; } if (number->expression.vtype == TYPE_INTEGER) builtin_num = number->constval.vint; else if (number->expression.vtype == TYPE_FLOAT) builtin_num = number->constval.vfloat; else { ast_unref(number); parseerror(parser, "builtin number must be an integer constant"); break; } ast_unref(number); fractional = modff(builtin_num, &integral); if (builtin_num < 0 || fractional != 0) { parseerror(parser, "builtin number must be an integer greater than zero"); break; } /* we only want the integral part anyways */ builtin_num = integral; } else if (parser->tok == TOKEN_INTCONST) { builtin_num = parser_token(parser)->constval.i; } else { parseerror(parser, "builtin number must be a compile time constant"); break; } if (var->hasvalue) { (void)!parsewarning(parser, WARN_DOUBLE_DECLARATION, "builtin `%s` has already been defined\n" " -> previous declaration here: %s:%i", var->name, ast_ctx(var).file, (int)ast_ctx(var).line); } else { func = ast_function_new(ast_ctx(var), var->name, var); if (!func) { parseerror(parser, "failed to allocate function for `%s`", var->name); break; } parser->functions.push_back(func); func->builtin = -builtin_num-1; } if (OPTS_FLAG(EXPRESSIONS_FOR_BUILTINS) ? (parser->tok != ',' && parser->tok != ';') : (!parser_next(parser))) { parseerror(parser, "expected comma or semicolon"); if (func) ast_function_delete(func); var->constval.vfunc = NULL; break; } } else if (var->expression.vtype == TYPE_ARRAY && parser->tok == '{') { if (localblock) { /* Note that fteqcc and most others don't even *have* * local arrays, so this is not a high priority. */ parseerror(parser, "TODO: initializers for local arrays"); break; } var->hasvalue = true; if (!parse_array(parser, var)) break; } else if (var->expression.vtype == TYPE_FUNCTION && (parser->tok == '{' || parser->tok == '[')) { if (localblock) { parseerror(parser, "cannot declare functions within functions"); break; } if (proto) ast_ctx(proto) = parser_ctx(parser); if (!parse_function_body(parser, var)) break; ast_delete(basetype); for (auto &it : parser->gotos) parseerror(parser, "undefined label: `%s`", it->name); return true; } else { ast_expression *cexp; ast_value *cval; bool folded_const = false; cexp = parse_expression_leave(parser, true, false, false); if (!cexp) break; cval = ast_istype(cexp, ast_value) ? (ast_value*)cexp : NULL; /* deal with foldable constants: */ if (localblock && var->cvq == CV_CONST && cval && cval->hasvalue && cval->cvq == CV_CONST && !cval->isfield) { /* remove it from the current locals */ if (isvector) { for (i = 0; i < 3; ++i) { vec_pop(parser->_locals); localblock->collect.pop_back(); } } /* do sanity checking, this function really needs refactoring */ if (vec_last(parser->_locals) != (ast_expression*)var) parseerror(parser, "internal error: unexpected change in local variable handling"); else vec_pop(parser->_locals); if (localblock->locals.back() != var) parseerror(parser, "internal error: unexpected change in local variable handling (2)"); else localblock->locals.pop_back(); /* push it to the to-be-generated globals */ parser->globals.push_back((ast_expression*)var); if (isvector) for (i = 0; i < 3; ++i) parser->globals.push_back((ast_expression*)last_me[i]); folded_const = true; } if (folded_const || !localblock || is_static) { if (cval != parser->nil && (!cval || ((!cval->hasvalue || cval->cvq != CV_CONST) && !cval->isfield)) ) { parseerror(parser, "initializer is non constant"); } else { if (!is_static && !OPTS_FLAG(INITIALIZED_NONCONSTANTS) && qualifier != CV_VAR) { var->cvq = CV_CONST; } if (cval == parser->nil) var->expression.flags |= AST_FLAG_INITIALIZED; else { var->hasvalue = true; if (cval->expression.vtype == TYPE_STRING) var->constval.vstring = parser_strdup(cval->constval.vstring); else if (cval->expression.vtype == TYPE_FIELD) var->constval.vfield = cval; else memcpy(&var->constval, &cval->constval, sizeof(var->constval)); ast_unref(cval); } } } else { int cvq; shunt sy = { NULL, NULL, NULL, NULL }; cvq = var->cvq; var->cvq = CV_NONE; vec_push(sy.out, syexp(ast_ctx(var), (ast_expression*)var)); vec_push(sy.out, syexp(ast_ctx(cexp), (ast_expression*)cexp)); vec_push(sy.ops, syop(ast_ctx(var), parser->assign_op)); if (!parser_sy_apply_operator(parser, &sy)) ast_unref(cexp); else { if (vec_size(sy.out) != 1 && vec_size(sy.ops) != 0) parseerror(parser, "internal error: leaked operands"); if (!ast_block_add_expr(localblock, (ast_expression*)sy.out[0].out)) break; } vec_free(sy.out); vec_free(sy.ops); vec_free(sy.argc); var->cvq = cvq; } /* a constant initialized to an inexact value should be marked inexact: * const float x = ; should propagate the inexact flag */ if (var->cvq == CV_CONST && var->expression.vtype == TYPE_FLOAT) { if (cval && cval->hasvalue && cval->cvq == CV_CONST) var->inexact = cval->inexact; } } another: if (parser->tok == ',') { if (!parser_next(parser)) { parseerror(parser, "expected another variable"); break; } if (parser->tok != TOKEN_IDENT) { parseerror(parser, "expected another variable"); break; } var = ast_value_copy(basetype); cleanvar = true; ast_value_set_name(var, parser_tokval(parser)); if (!parser_next(parser)) { parseerror(parser, "error parsing variable declaration"); break; } continue; } if (parser->tok != ';') { parseerror(parser, "missing semicolon after variables"); break; } if (!parser_next(parser)) { parseerror(parser, "parse error after variable declaration"); break; } ast_delete(basetype); return true; } if (cleanvar && var) ast_delete(var); ast_delete(basetype); return false; cleanup: ast_delete(basetype); if (cleanvar && var) ast_delete(var); if (me[0]) ast_member_delete(me[0]); if (me[1]) ast_member_delete(me[1]); if (me[2]) ast_member_delete(me[2]); return retval; } static bool parser_global_statement(parser_t *parser) { int cvq = CV_WRONG; bool noref = false; bool is_static = false; uint32_t qflags = 0; ast_value *istype = NULL; char *vstring = NULL; if (parser->tok == TOKEN_IDENT) istype = parser_find_typedef(parser, parser_tokval(parser), 0); if (istype || parser->tok == TOKEN_TYPENAME || parser->tok == '.' || parser->tok == TOKEN_DOTS) { return parse_variable(parser, NULL, false, CV_NONE, istype, false, false, 0, NULL); } else if (parse_qualifiers(parser, false, &cvq, &noref, &is_static, &qflags, &vstring)) { if (cvq == CV_WRONG) return false; return parse_variable(parser, NULL, false, cvq, NULL, noref, is_static, qflags, vstring); } else if (parser->tok == TOKEN_IDENT && !strcmp(parser_tokval(parser), "enum")) { return parse_enum(parser); } else if (parser->tok == TOKEN_KEYWORD) { if (!strcmp(parser_tokval(parser), "typedef")) { if (!parser_next(parser)) { parseerror(parser, "expected type definition after 'typedef'"); return false; } return parse_typedef(parser); } parseerror(parser, "unrecognized keyword `%s`", parser_tokval(parser)); return false; } else if (parser->tok == '#') { return parse_pragma(parser); } else if (parser->tok == '$') { if (!parser_next(parser)) { parseerror(parser, "parse error"); return false; } } else { parseerror(parser, "unexpected token: `%s`", parser->lex->tok.value); return false; } return true; } static uint16_t progdefs_crc_sum(uint16_t old, const char *str) { return util_crc16(old, str, strlen(str)); } static void progdefs_crc_file(const char *str) { /* write to progdefs.h here */ (void)str; } static uint16_t progdefs_crc_both(uint16_t old, const char *str) { old = progdefs_crc_sum(old, str); progdefs_crc_file(str); return old; } static void generate_checksum(parser_t *parser, ir_builder *ir) { uint16_t crc = 0xFFFF; size_t i; ast_value *value; crc = progdefs_crc_both(crc, "\n/* file generated by qcc, do not modify */\n\ntypedef struct\n{"); crc = progdefs_crc_sum(crc, "\tint\tpad[28];\n"); /* progdefs_crc_file("\tint\tpad;\n"); progdefs_crc_file("\tint\tofs_return[3];\n"); progdefs_crc_file("\tint\tofs_parm0[3];\n"); progdefs_crc_file("\tint\tofs_parm1[3];\n"); progdefs_crc_file("\tint\tofs_parm2[3];\n"); progdefs_crc_file("\tint\tofs_parm3[3];\n"); progdefs_crc_file("\tint\tofs_parm4[3];\n"); progdefs_crc_file("\tint\tofs_parm5[3];\n"); progdefs_crc_file("\tint\tofs_parm6[3];\n"); progdefs_crc_file("\tint\tofs_parm7[3];\n"); */ for (i = 0; i < parser->crc_globals; ++i) { if (!ast_istype(parser->globals[i], ast_value)) continue; value = (ast_value*)(parser->globals[i]); switch (value->expression.vtype) { case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break; case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break; case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break; case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break; default: crc = progdefs_crc_both(crc, "\tint\t"); break; } crc = progdefs_crc_both(crc, value->name); crc = progdefs_crc_both(crc, ";\n"); } crc = progdefs_crc_both(crc, "} globalvars_t;\n\ntypedef struct\n{\n"); for (i = 0; i < parser->crc_fields; ++i) { if (!ast_istype(parser->fields[i], ast_value)) continue; value = (ast_value*)(parser->fields[i]); switch (value->expression.next->vtype) { case TYPE_FLOAT: crc = progdefs_crc_both(crc, "\tfloat\t"); break; case TYPE_VECTOR: crc = progdefs_crc_both(crc, "\tvec3_t\t"); break; case TYPE_STRING: crc = progdefs_crc_both(crc, "\tstring_t\t"); break; case TYPE_FUNCTION: crc = progdefs_crc_both(crc, "\tfunc_t\t"); break; default: crc = progdefs_crc_both(crc, "\tint\t"); break; } crc = progdefs_crc_both(crc, value->name); crc = progdefs_crc_both(crc, ";\n"); } crc = progdefs_crc_both(crc, "} entvars_t;\n\n"); ir->code->crc = crc; } parser_t *parser_create() { parser_t *parser; lex_ctx_t empty_ctx; size_t i; parser = (parser_t*)mem_a(sizeof(parser_t)); if (!parser) return NULL; memset(parser, 0, sizeof(*parser)); // TODO: remove new (parser) parser_t(); for (i = 0; i < operator_count; ++i) { if (operators[i].id == opid1('=')) { parser->assign_op = operators+i; break; } } if (!parser->assign_op) { con_err("internal error: initializing parser: failed to find assign operator\n"); mem_d(parser); return NULL; } vec_push(parser->variables, parser->htfields = util_htnew(PARSER_HT_SIZE)); vec_push(parser->variables, parser->htglobals = util_htnew(PARSER_HT_SIZE)); vec_push(parser->typedefs, util_htnew(TYPEDEF_HT_SIZE)); vec_push(parser->_blocktypedefs, 0); parser->aliases = util_htnew(PARSER_HT_SIZE); empty_ctx.file = ""; empty_ctx.line = 0; empty_ctx.column = 0; parser->nil = ast_value_new(empty_ctx, "nil", TYPE_NIL); parser->nil->cvq = CV_CONST; if (OPTS_FLAG(UNTYPED_NIL)) util_htset(parser->htglobals, "nil", (void*)parser->nil); parser->max_param_count = 1; parser->const_vec[0] = ast_value_new(empty_ctx, "", TYPE_NOEXPR); parser->const_vec[1] = ast_value_new(empty_ctx, "", TYPE_NOEXPR); parser->const_vec[2] = ast_value_new(empty_ctx, "", TYPE_NOEXPR); if (OPTS_OPTION_BOOL(OPTION_ADD_INFO)) { parser->reserved_version = ast_value_new(empty_ctx, "reserved:version", TYPE_STRING); parser->reserved_version->cvq = CV_CONST; parser->reserved_version->hasvalue = true; parser->reserved_version->expression.flags |= AST_FLAG_INCLUDE_DEF; parser->reserved_version->constval.vstring = util_strdup(GMQCC_FULL_VERSION_STRING); } else { parser->reserved_version = NULL; } parser->fold = fold_init (parser); parser->intrin = intrin_init(parser); return parser; } static bool parser_compile(parser_t *parser) { /* initial lexer/parser state */ parser->lex->flags.noops = true; if (parser_next(parser)) { while (parser->tok != TOKEN_EOF && parser->tok < TOKEN_ERROR) { if (!parser_global_statement(parser)) { if (parser->tok == TOKEN_EOF) parseerror(parser, "unexpected end of file"); else if (compile_errors) parseerror(parser, "there have been errors, bailing out"); lex_close(parser->lex); parser->lex = NULL; return false; } } } else { parseerror(parser, "parse error"); lex_close(parser->lex); parser->lex = NULL; return false; } lex_close(parser->lex); parser->lex = NULL; return !compile_errors; } bool parser_compile_file(parser_t *parser, const char *filename) { parser->lex = lex_open(filename); if (!parser->lex) { con_err("failed to open file \"%s\"\n", filename); return false; } return parser_compile(parser); } bool parser_compile_string(parser_t *parser, const char *name, const char *str, size_t len) { parser->lex = lex_open_string(str, len, name); if (!parser->lex) { con_err("failed to create lexer for string \"%s\"\n", name); return false; } return parser_compile(parser); } static void parser_remove_ast(parser_t *parser) { size_t i; if (parser->ast_cleaned) return; parser->ast_cleaned = true; for (auto &it : parser->accessors) { ast_delete(it->constval.vfunc); it->constval.vfunc = nullptr; ast_delete(it); } for (auto &it : parser->functions) ast_delete(it); for (auto &it : parser->globals) ast_delete(it); for (auto &it : parser->fields) ast_delete(it); for (i = 0; i < vec_size(parser->variables); ++i) util_htdel(parser->variables[i]); vec_free(parser->variables); vec_free(parser->_blocklocals); vec_free(parser->_locals); for (i = 0; i < vec_size(parser->_typedefs); ++i) ast_delete(parser->_typedefs[i]); vec_free(parser->_typedefs); for (i = 0; i < vec_size(parser->typedefs); ++i) util_htdel(parser->typedefs[i]); vec_free(parser->typedefs); vec_free(parser->_blocktypedefs); vec_free(parser->_block_ctx); ast_value_delete(parser->nil); ast_value_delete(parser->const_vec[0]); ast_value_delete(parser->const_vec[1]); ast_value_delete(parser->const_vec[2]); if (parser->reserved_version) ast_value_delete(parser->reserved_version); util_htdel(parser->aliases); fold_cleanup(parser->fold); intrin_cleanup(parser->intrin); } void parser_cleanup(parser_t *parser) { parser_remove_ast(parser); mem_d(parser); } static bool parser_set_coverage_func(parser_t *parser, ir_builder *ir) { ast_expression *expr; ast_value *cov; ast_function *func; if (!OPTS_OPTION_BOOL(OPTION_COVERAGE)) return true; func = nullptr; for (auto &it : parser->functions) { if (!strcmp(it->name, "coverage")) { func = it; break; } } if (!func) { if (OPTS_OPTION_BOOL(OPTION_COVERAGE)) { con_out("coverage support requested but no coverage() builtin declared\n"); ir_builder_delete(ir); return false; } return true; } cov = func->vtype; expr = (ast_expression*)cov; if (expr->vtype != TYPE_FUNCTION || expr->params.size()) { char ty[1024]; ast_type_to_string(expr, ty, sizeof(ty)); con_out("invalid type for coverage(): %s\n", ty); ir_builder_delete(ir); return false; } ir->coverage_func = func->ir_func->value; return true; } bool parser_finish(parser_t *parser, const char *output) { ir_builder *ir; bool retval = true; if (compile_errors) { con_out("*** there were compile errors\n"); return false; } ir = ir_builder_new("gmqcc_out"); if (!ir) { con_out("failed to allocate builder\n"); return false; } for (auto &it : parser->fields) { bool hasvalue; if (!ast_istype(it, ast_value)) continue; ast_value *field = (ast_value*)it; hasvalue = field->hasvalue; field->hasvalue = false; if (!ast_global_codegen((ast_value*)field, ir, true)) { con_out("failed to generate field %s\n", field->name); ir_builder_delete(ir); return false; } if (hasvalue) { ir_value *ifld; ast_expression *subtype; field->hasvalue = true; subtype = field->expression.next; ifld = ir_builder_create_field(ir, field->name, subtype->vtype); if (subtype->vtype == TYPE_FIELD) ifld->fieldtype = subtype->next->vtype; else if (subtype->vtype == TYPE_FUNCTION) ifld->outtype = subtype->next->vtype; (void)!ir_value_set_field(field->ir_v, ifld); } } for (auto &it : parser->globals) { ast_value *asvalue; if (!ast_istype(it, ast_value)) continue; asvalue = (ast_value*)it; if (!asvalue->uses && !asvalue->hasvalue && asvalue->expression.vtype != TYPE_FUNCTION) { retval = retval && !compile_warning(ast_ctx(asvalue), WARN_UNUSED_VARIABLE, "unused global: `%s`", asvalue->name); } if (!ast_global_codegen(asvalue, ir, false)) { con_out("failed to generate global %s\n", asvalue->name); ir_builder_delete(ir); return false; } } /* Build function vararg accessor ast tree now before generating * immediates, because the accessors may add new immediates */ for (auto &f : parser->functions) { if (f->varargs) { if (parser->max_param_count > f->vtype->expression.params.size()) { f->varargs->expression.count = parser->max_param_count - f->vtype->expression.params.size(); if (!parser_create_array_setter_impl(parser, f->varargs)) { con_out("failed to generate vararg setter for %s\n", f->name); ir_builder_delete(ir); return false; } if (!parser_create_array_getter_impl(parser, f->varargs)) { con_out("failed to generate vararg getter for %s\n", f->name); ir_builder_delete(ir); return false; } } else { ast_delete(f->varargs); f->varargs = NULL; } } } /* Now we can generate immediates */ if (!fold_generate(parser->fold, ir)) return false; /* before generating any functions we need to set the coverage_func */ if (!parser_set_coverage_func(parser, ir)) return false; for (auto &it : parser->globals) { if (!ast_istype(it, ast_value)) continue; ast_value *asvalue = (ast_value*)it; if (!(asvalue->expression.flags & AST_FLAG_INITIALIZED)) { if (asvalue->cvq == CV_CONST && !asvalue->hasvalue) (void)!compile_warning(ast_ctx(asvalue), WARN_UNINITIALIZED_CONSTANT, "uninitialized constant: `%s`", asvalue->name); else if ((asvalue->cvq == CV_NONE || asvalue->cvq == CV_CONST) && !asvalue->hasvalue) (void)!compile_warning(ast_ctx(asvalue), WARN_UNINITIALIZED_GLOBAL, "uninitialized global: `%s`", asvalue->name); } if (!ast_generate_accessors(asvalue, ir)) { ir_builder_delete(ir); return false; } } for (auto &it : parser->fields) { ast_value *asvalue = (ast_value*)it->next; if (!ast_istype((ast_expression*)asvalue, ast_value)) continue; if (asvalue->expression.vtype != TYPE_ARRAY) continue; if (!ast_generate_accessors(asvalue, ir)) { ir_builder_delete(ir); return false; } } if (parser->reserved_version && !ast_global_codegen(parser->reserved_version, ir, false)) { con_out("failed to generate reserved::version"); ir_builder_delete(ir); return false; } for (auto &f : parser->functions) { if (!ast_function_codegen(f, ir)) { con_out("failed to generate function %s\n", f->name); ir_builder_delete(ir); return false; } } generate_checksum(parser, ir); if (OPTS_OPTION_BOOL(OPTION_DUMP)) ir_builder_dump(ir, con_out); for (auto &it : parser->functions) { if (!ir_function_finalize(it->ir_func)) { con_out("failed to finalize function %s\n", it->name); ir_builder_delete(ir); return false; } } parser_remove_ast(parser); if (compile_Werrors) { con_out("*** there were warnings treated as errors\n"); compile_show_werrors(); retval = false; } if (retval) { if (OPTS_OPTION_BOOL(OPTION_DUMPFIN)) ir_builder_dump(ir, con_out); if (!ir_builder_generate(ir, output)) { con_out("*** failed to generate output file\n"); ir_builder_delete(ir); return false; } } ir_builder_delete(ir); return retval; }