/*
* Copyright (C) 2012, 2013
* Wolfgang Bumiller
* Dale Weiler
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
* of the Software, and to permit persons to whom the Software is furnished to do
* so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include <stdio.h>
#include <stdarg.h>
#include "gmqcc.h"
#include "lexer.h"
#include "ast.h"
/* beginning of locals */
#define PARSER_HT_LOCALS 2
#define PARSER_HT_SIZE 128
#define TYPEDEF_HT_SIZE 16
typedef struct {
lex_file *lex;
int tok;
ast_expression **globals;
ast_expression **fields;
ast_function **functions;
ast_value **imm_float;
ast_value **imm_string;
ast_value **imm_vector;
size_t translated;
/* must be deleted first, they reference immediates and values */
ast_value **accessors;
ast_value *imm_float_zero;
ast_value *imm_float_one;
ast_value *imm_float_neg_one;
ast_value *imm_vector_zero;
ast_value *nil;
ast_value *reserved_version;
size_t crc_globals;
size_t crc_fields;
ast_function *function;
ht aliases;
/* All the labels the function defined...
* Should they be in ast_function instead?
*/
ast_label **labels;
ast_goto **gotos;
const char **breaks;
const char **continues;
/* A list of hashtables for each scope */
ht *variables;
ht htfields;
ht htglobals;
ht *typedefs;
/* same as above but for the spelling corrector */
correct_trie_t **correct_variables;
size_t ***correct_variables_score; /* vector of vector of size_t* */
/* not to be used directly, we use the hash table */
ast_expression **_locals;
size_t *_blocklocals;
ast_value **_typedefs;
size_t *_blocktypedefs;
lex_ctx *_block_ctx;
/* we store the '=' operator info */
const oper_info *assign_op;
/* magic values */
ast_value *const_vec[3];
/* pragma flags */
bool noref;
/* collected information */
size_t max_param_count;
} parser_t;
static ast_expression * const intrinsic_debug_typestring = (ast_expression*)0x1;
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);
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;
}
/**********************************************************************
* some maths used for constant folding
*/
vector vec3_add(vector a, vector b)
{
vector out;
out.x = a.x + b.x;
out.y = a.y + b.y;
out.z = a.z + b.z;
return out;
}
vector vec3_sub(vector a, vector b)
{
vector out;
out.x = a.x - b.x;
out.y = a.y - b.y;
out.z = a.z - b.z;
return out;
}
qcfloat vec3_mulvv(vector a, vector b)
{
return (a.x * b.x + a.y * b.y + a.z * b.z);
}
vector vec3_mulvf(vector a, float b)
{
vector out;
out.x = a.x * b;
out.y = a.y * b;
out.z = a.z * b;
return out;
}
/**********************************************************************
* parsing
*/
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))
#define parser_ctx(p) ((p)->lex->tok.ctx)
static ast_value* parser_const_float(parser_t *parser, double d)
{
size_t i;
ast_value *out;
lex_ctx ctx;
for (i = 0; i < vec_size(parser->imm_float); ++i) {
const double compare = parser->imm_float[i]->constval.vfloat;
if (memcmp((const void*)&compare, (const void *)&d, sizeof(double)) == 0)
return parser->imm_float[i];
}
if (parser->lex)
ctx = parser_ctx(parser);
else {
memset(&ctx, 0, sizeof(ctx));
}
out = ast_value_new(ctx, "#IMMEDIATE", TYPE_FLOAT);
out->cvq = CV_CONST;
out->hasvalue = true;
out->constval.vfloat = d;
vec_push(parser->imm_float, out);
return out;
}
static ast_value* parser_const_float_0(parser_t *parser)
{
if (!parser->imm_float_zero)
parser->imm_float_zero = parser_const_float(parser, 0);
return parser->imm_float_zero;
}
static ast_value* parser_const_float_neg1(parser_t *parser) {
if (!parser->imm_float_neg_one)
parser->imm_float_neg_one = parser_const_float(parser, -1);
return parser->imm_float_neg_one;
}
static ast_value* parser_const_float_1(parser_t *parser)
{
if (!parser->imm_float_one)
parser->imm_float_one = parser_const_float(parser, 1);
return parser->imm_float_one;
}
static 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_value* parser_const_string(parser_t *parser, const char *str, bool dotranslate)
{
size_t i;
ast_value *out;
for (i = 0; i < vec_size(parser->imm_string); ++i) {
if (!strcmp(parser->imm_string[i]->constval.vstring, str))
return parser->imm_string[i];
}
if (dotranslate) {
char name[32];
snprintf(name, sizeof(name), "dotranslate_%lu", (unsigned long)(parser->translated++));
out = ast_value_new(parser_ctx(parser), name, TYPE_STRING);
} else
out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_STRING);
out->cvq = CV_CONST;
out->hasvalue = true;
out->constval.vstring = parser_strdup(str);
vec_push(parser->imm_string, out);
return out;
}
static ast_value* parser_const_vector(parser_t *parser, vector v)
{
size_t i;
ast_value *out;
for (i = 0; i < vec_size(parser->imm_vector); ++i) {
if (!memcmp(&parser->imm_vector[i]->constval.vvec, &v, sizeof(v)))
return parser->imm_vector[i];
}
out = ast_value_new(parser_ctx(parser), "#IMMEDIATE", TYPE_VECTOR);
out->cvq = CV_CONST;
out->hasvalue = true;
out->constval.vvec = v;
vec_push(parser->imm_vector, out);
return out;
}
static ast_value* parser_const_vector_f(parser_t *parser, float x, float y, float z)
{
vector v;
v.x = x;
v.y = y;
v.z = z;
return parser_const_vector(parser, v);
}
static ast_value* parser_const_vector_0(parser_t *parser)
{
if (!parser->imm_vector_zero)
parser->imm_vector_zero = parser_const_vector_f(parser, 0, 0, 0);
return parser->imm_vector_zero;
}
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)
{
size_t i;
for(i = 0; i < vec_size(parser->labels); i++)
if (!strcmp(parser->labels[i]->name, name))
return (ast_expression*)parser->labels[i];
return NULL;
}
static 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)
{
size_t i;
ast_value *fun;
if (!parser->function)
return NULL;
fun = parser->function->vtype;
for (i = 0; i < vec_size(fun->expression.params); ++i) {
if (!strcmp(fun->expression.params[i]->name, name))
return (ast_expression*)(fun->expression.params[i]);
}
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;
}
typedef struct
{
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 ctx;
} sy_elem;
enum {
PAREN_EXPR,
PAREN_FUNC,
PAREN_INDEX,
PAREN_TERNARY1,
PAREN_TERNARY2
};
typedef struct
{
sy_elem *out;
sy_elem *ops;
size_t *argc;
unsigned int *paren;
} shunt;
static sy_elem syexp(lex_ctx 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 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 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 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 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 immediate_is_true(lex_ctx ctx, ast_value *v)
{
switch (v->expression.vtype) {
case TYPE_FLOAT:
return !!v->constval.vfloat;
case TYPE_INTEGER:
return !!v->constval.vint;
case TYPE_VECTOR:
if (OPTS_FLAG(CORRECT_LOGIC))
return v->constval.vvec.x &&
v->constval.vvec.y &&
v->constval.vvec.z;
else
return !!(v->constval.vvec.x);
case TYPE_STRING:
if (!v->constval.vstring)
return false;
if (v->constval.vstring && OPTS_FLAG(TRUE_EMPTY_STRINGS))
return true;
return !!v->constval.vstring[0];
default:
compile_error(ctx, "internal error: immediate_is_true on invalid type");
return !!v->constval.vfunc;
}
}
static bool parser_sy_apply_operator(parser_t *parser, shunt *sy)
{
const oper_info *op;
lex_ctx ctx;
ast_expression *out = NULL;
ast_expression *exprs[3];
ast_block *blocks[3];
ast_value *asvalue[3];
ast_binstore *asbinstore;
size_t i, assignop, addop, subop;
qcint 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) {
compile_error(ctx, "internal error: not enough operands: %i (operator %s (%i))", vec_size(sy->out),
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;
asvalue[i] = (ast_value*)exprs[i];
if (exprs[i]->expression.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] && !vec_size(blocks[0]->exprs) && op->id != opid1(',')) {
compile_error(ctx, "internal error: operator cannot be applied on empty blocks");
return false;
}
#define NotSameType(T) \
(exprs[0]->expression.vtype != exprs[1]->expression.vtype || \
exprs[0]->expression.vtype != T)
#define CanConstFold1(A) \
(ast_istype((A), ast_value) && ((ast_value*)(A))->hasvalue && (((ast_value*)(A))->cvq == CV_CONST) &&\
(A)->expression.vtype != TYPE_FUNCTION)
#define CanConstFold(A, B) \
(CanConstFold1(A) && CanConstFold1(B))
#define ConstV(i) (asvalue[(i)]->constval.vvec)
#define ConstF(i) (asvalue[(i)]->constval.vfloat)
#define ConstS(i) (asvalue[(i)]->constval.vstring)
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]->expression.vtype == TYPE_VECTOR &&
exprs[1]->expression.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]->expression.vtype == TYPE_ENTITY) {
if (exprs[1]->expression.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]->expression.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]->expression.vtype != TYPE_ARRAY &&
!(exprs[0]->expression.vtype == TYPE_FIELD &&
exprs[0]->expression.next->expression.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]->expression.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]);
if (rotate_entfield_array_index_nodes(&out))
{
#if 0
/* This is not broken in fteqcc anymore */
if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_GMQCC) {
/* this error doesn't need to make us bail out */
(void)!parsewarning(parser, WARN_EXTENSIONS,
"accessing array-field members of an entity without parenthesis\n"
" -> this is an extension from -std=gmqcc");
}
#endif
}
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'):
switch (exprs[0]->expression.vtype) {
case TYPE_FLOAT:
if (CanConstFold1(exprs[0]))
out = (ast_expression*)parser_const_float(parser, -ConstF(0));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F,
(ast_expression*)parser_const_float_0(parser),
exprs[0]);
break;
case TYPE_VECTOR:
if (CanConstFold1(exprs[0]))
out = (ast_expression*)parser_const_vector_f(parser,
-ConstV(0).x, -ConstV(0).y, -ConstV(0).z);
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_V,
(ast_expression*)parser_const_vector_0(parser),
exprs[0]);
break;
default:
compile_error(ctx, "invalid types used in expression: cannot negate type %s",
type_name[exprs[0]->expression.vtype]);
return false;
}
break;
case opid2('!','P'):
switch (exprs[0]->expression.vtype) {
case TYPE_FLOAT:
if (CanConstFold1(exprs[0]))
out = (ast_expression*)parser_const_float(parser, !ConstF(0));
else
out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_F, exprs[0]);
break;
case TYPE_VECTOR:
if (CanConstFold1(exprs[0]))
out = (ast_expression*)parser_const_float(parser,
(!ConstV(0).x && !ConstV(0).y && !ConstV(0).z));
else
out = (ast_expression*)ast_unary_new(ctx, INSTR_NOT_V, exprs[0]);
break;
case TYPE_STRING:
if (CanConstFold1(exprs[0])) {
if (OPTS_FLAG(TRUE_EMPTY_STRINGS))
out = (ast_expression*)parser_const_float(parser, !ConstS(0));
else
out = (ast_expression*)parser_const_float(parser, !ConstS(0) || !*ConstS(0));
} else {
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]->expression.vtype]);
return false;
}
break;
case opid1('+'):
if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
(exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
{
compile_error(ctx, "invalid types used in expression: cannot add type %s and %s",
type_name[exprs[0]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
}
switch (exprs[0]->expression.vtype) {
case TYPE_FLOAT:
if (CanConstFold(exprs[0], exprs[1]))
{
out = (ast_expression*)parser_const_float(parser, ConstF(0) + ConstF(1));
}
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_ADD_F, exprs[0], exprs[1]);
break;
case TYPE_VECTOR:
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_vector(parser, vec3_add(ConstV(0), ConstV(1)));
else
out = (ast_expression*)ast_binary_new(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]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
};
break;
case opid1('-'):
if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
(exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.vtype != TYPE_FLOAT) )
{
compile_error(ctx, "invalid types used in expression: cannot subtract type %s from %s",
type_name[exprs[1]->expression.vtype],
type_name[exprs[0]->expression.vtype]);
return false;
}
switch (exprs[0]->expression.vtype) {
case TYPE_FLOAT:
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_float(parser, ConstF(0) - ConstF(1));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, exprs[0], exprs[1]);
break;
case TYPE_VECTOR:
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_vector(parser, vec3_sub(ConstV(0), ConstV(1)));
else
out = (ast_expression*)ast_binary_new(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]->expression.vtype],
type_name[exprs[0]->expression.vtype]);
return false;
};
break;
case opid1('*'):
if (exprs[0]->expression.vtype != exprs[1]->expression.vtype &&
!(exprs[0]->expression.vtype == TYPE_VECTOR &&
exprs[1]->expression.vtype == TYPE_FLOAT) &&
!(exprs[1]->expression.vtype == TYPE_VECTOR &&
exprs[0]->expression.vtype == TYPE_FLOAT)
)
{
compile_error(ctx, "invalid types used in expression: cannot multiply types %s and %s",
type_name[exprs[1]->expression.vtype],
type_name[exprs[0]->expression.vtype]);
return false;
}
switch (exprs[0]->expression.vtype) {
case TYPE_FLOAT:
if (exprs[1]->expression.vtype == TYPE_VECTOR)
{
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(1), ConstF(0)));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_FV, exprs[0], exprs[1]);
}
else
{
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_float(parser, ConstF(0) * ConstF(1));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, exprs[0], exprs[1]);
}
break;
case TYPE_VECTOR:
if (exprs[1]->expression.vtype == TYPE_FLOAT)
{
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(0), ConstF(1)));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], exprs[1]);
}
else
{
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_float(parser, vec3_mulvv(ConstV(0), ConstV(1)));
else if (OPTS_OPTIMIZATION(OPTIM_VECTOR_COMPONENTS) && CanConstFold1(exprs[0])) {
vector vec = ConstV(0);
if (!vec.y && !vec.z) { /* 'n 0 0' * v */
++opts_optimizationcount[OPTIM_VECTOR_COMPONENTS];
out = (ast_expression*)ast_member_new(ctx, exprs[1], 0, NULL);
out->expression.node.keep = false;
((ast_member*)out)->rvalue = true;
if (vec.x != 1)
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, (ast_expression*)parser_const_float(parser, vec.x), out);
}
else if (!vec.x && !vec.z) { /* '0 n 0' * v */
++opts_optimizationcount[OPTIM_VECTOR_COMPONENTS];
out = (ast_expression*)ast_member_new(ctx, exprs[1], 1, NULL);
out->expression.node.keep = false;
((ast_member*)out)->rvalue = true;
if (vec.y != 1)
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, (ast_expression*)parser_const_float(parser, vec.y), out);
}
else if (!vec.x && !vec.y) { /* '0 n 0' * v */
++opts_optimizationcount[OPTIM_VECTOR_COMPONENTS];
out = (ast_expression*)ast_member_new(ctx, exprs[1], 2, NULL);
out->expression.node.keep = false;
((ast_member*)out)->rvalue = true;
if (vec.z != 1)
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, (ast_expression*)parser_const_float(parser, vec.z), out);
}
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
}
else if (OPTS_OPTIMIZATION(OPTIM_VECTOR_COMPONENTS) && CanConstFold1(exprs[1])) {
vector vec = ConstV(1);
if (!vec.y && !vec.z) { /* v * 'n 0 0' */
++opts_optimizationcount[OPTIM_VECTOR_COMPONENTS];
out = (ast_expression*)ast_member_new(ctx, exprs[0], 0, NULL);
out->expression.node.keep = false;
((ast_member*)out)->rvalue = true;
if (vec.x != 1)
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, out, (ast_expression*)parser_const_float(parser, vec.x));
}
else if (!vec.x && !vec.z) { /* v * '0 n 0' */
++opts_optimizationcount[OPTIM_VECTOR_COMPONENTS];
out = (ast_expression*)ast_member_new(ctx, exprs[0], 1, NULL);
out->expression.node.keep = false;
((ast_member*)out)->rvalue = true;
if (vec.y != 1)
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, out, (ast_expression*)parser_const_float(parser, vec.y));
}
else if (!vec.x && !vec.y) { /* v * '0 n 0' */
++opts_optimizationcount[OPTIM_VECTOR_COMPONENTS];
out = (ast_expression*)ast_member_new(ctx, exprs[0], 2, NULL);
out->expression.node.keep = false;
((ast_member*)out)->rvalue = true;
if (vec.z != 1)
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_F, out, (ast_expression*)parser_const_float(parser, vec.z));
}
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_V, exprs[0], exprs[1]);
}
else
out = (ast_expression*)ast_binary_new(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]->expression.vtype],
type_name[exprs[0]->expression.vtype]);
return false;
};
break;
case opid1('/'):
if (exprs[1]->expression.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 tyeps %s and %s", ty1, ty2);
return false;
}
if (exprs[0]->expression.vtype == TYPE_FLOAT) {
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_float(parser, ConstF(0) / ConstF(1));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F, exprs[0], exprs[1]);
}
else if (exprs[0]->expression.vtype == TYPE_VECTOR) {
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_vector(parser, vec3_mulvf(ConstV(0), 1.0/ConstF(1)));
else {
if (CanConstFold1(exprs[1])) {
out = (ast_expression*)parser_const_float(parser, 1.0 / ConstF(1));
} else {
out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F,
(ast_expression*)parser_const_float_1(parser),
exprs[1]);
}
if (!out) {
compile_error(ctx, "internal error: failed to generate division");
return false;
}
out = (ast_expression*)ast_binary_new(ctx, INSTR_MUL_VF, exprs[0], out);
}
}
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 tyeps %s and %s", ty1, ty2);
return false;
}
break;
case opid1('%'):
case opid2('%','='):
compile_error(ctx, "qc does not have a modulo operator");
return false;
case opid1('|'):
case opid1('&'):
if (NotSameType(TYPE_FLOAT)) {
compile_error(ctx, "invalid types used in expression: cannot perform bit operations between types %s and %s",
type_name[exprs[0]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
}
if (CanConstFold(exprs[0], exprs[1]))
out = (ast_expression*)parser_const_float(parser,
(op->id == opid1('|') ? (float)( ((qcint)ConstF(0)) | ((qcint)ConstF(1)) ) :
(float)( ((qcint)ConstF(0)) & ((qcint)ConstF(1)) ) ));
else
out = (ast_expression*)ast_binary_new(ctx,
(op->id == opid1('|') ? INSTR_BITOR : INSTR_BITAND),
exprs[0], exprs[1]);
break;
case opid1('^'):
compile_error(ast_ctx(exprs[0]), "Not Yet Implemented: bit-xor via ^");
return false;
case opid2('<','<'):
case opid2('>','>'):
if (CanConstFold(exprs[0], exprs[1]) && ! NotSameType(TYPE_FLOAT)) {
if (op->id == opid2('<','<'))
out = (ast_expression*)parser_const_float(parser, (double)((int)(ConstF(0)) << (int)(ConstF(1))));
else
out = (ast_expression*)parser_const_float(parser, (double)((int)(ConstF(0)) >> (int)(ConstF(1))));
break;
}
case opid3('<','<','='):
case opid3('>','>','='):
compile_error(ast_ctx(exprs[0]), "Not Yet Implemented: bit-shifts");
return false;
case opid2('|','|'):
generated_op += 1; /* INSTR_OR */
case opid2('&','&'):
generated_op += INSTR_AND;
if (CanConstFold(exprs[0], exprs[1]))
{
if (OPTS_FLAG(PERL_LOGIC)) {
if (immediate_is_true(ctx, asvalue[0]))
out = exprs[1];
}
else
out = (ast_expression*)parser_const_float(parser,
( (generated_op == INSTR_OR)
? (immediate_is_true(ctx, asvalue[0]) || immediate_is_true(ctx, asvalue[1]))
: (immediate_is_true(ctx, asvalue[0]) && immediate_is_true(ctx, asvalue[1])) )
? 1 : 0);
}
else
{
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]->expression.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]->expression.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 = (ast_expression*)ast_binary_new(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 (CanConstFold1(exprs[0]))
out = (immediate_is_true(ctx, asvalue[0]) ? exprs[1] : exprs[2]);
else
out = (ast_expression*)ast_ternary_new(ctx, exprs[0], exprs[1], exprs[2]);
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 (CanConstFold(exprs[0], exprs[1])) {
if (ConstF(0) < ConstF(1))
out = (ast_expression*)parser_const_float_neg1(parser);
else if (ConstF(0) == ConstF(1))
out = (ast_expression*)parser_const_float_0(parser);
else if (ConstF(0) > ConstF(1))
out = (ast_expression*)parser_const_float_1(parser);
} else {
ast_binary *eq = ast_binary_new(ctx, INSTR_EQ_F, exprs[0], exprs[1]);
eq->refs = (ast_binary_ref)false; /* references nothing */
/* 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_const_float_neg1(parser),
/* } else { */
/* if (eq) { */
(ast_expression*)ast_ternary_new(ctx, (ast_expression*)eq,
/* out = 0 */
(ast_expression*)parser_const_float_0(parser),
/* } else { */
/* out = 1 */
(ast_expression*)parser_const_float_1(parser)
/* } */
)
/* } */
);
}
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]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
}
out = (ast_expression*)ast_binary_new(ctx, generated_op, exprs[0], exprs[1]);
break;
case opid2('!', '='):
if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
type_name[exprs[0]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
}
out = (ast_expression*)ast_binary_new(ctx, type_ne_instr[exprs[0]->expression.vtype], exprs[0], exprs[1]);
break;
case opid2('=', '='):
if (exprs[0]->expression.vtype != exprs[1]->expression.vtype) {
compile_error(ctx, "invalid types used in expression: cannot perform comparison between types %s and %s",
type_name[exprs[0]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
}
out = (ast_expression*)ast_binary_new(ctx, type_eq_instr[exprs[0]->expression.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]->expression.vtype == TYPE_FIELD &&
exprs[0]->expression.next->expression.vtype == TYPE_VECTOR)
{
assignop = type_storep_instr[TYPE_VECTOR];
}
else
assignop = type_storep_instr[exprs[0]->expression.vtype];
if (assignop == VINSTR_END || !ast_compare_type(field->expression.next, exprs[1]))
{
ast_type_to_string(field->expression.next, ty1, sizeof(ty1));
ast_type_to_string(exprs[1], ty2, sizeof(ty2));
if (OPTS_FLAG(ASSIGN_FUNCTION_TYPES) &&
field->expression.next->expression.vtype == TYPE_FUNCTION &&
exprs[1]->expression.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]->expression.vtype == TYPE_FIELD &&
exprs[0]->expression.next->expression.vtype == TYPE_VECTOR)
{
assignop = type_store_instr[TYPE_VECTOR];
}
else {
assignop = type_store_instr[exprs[0]->expression.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]->expression.vtype == TYPE_FUNCTION &&
exprs[1]->expression.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);
}
}
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ctx, "assignment to constant `%s`", asvalue[0]->name);
}
out = (ast_expression*)ast_store_new(ctx, assignop, exprs[0], exprs[1]);
break;
case opid3('+','+','P'):
case opid3('-','-','P'):
/* prefix ++ */
if (exprs[0]->expression.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;
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ast_ctx(exprs[0]), "assignment to constant `%s`", asvalue[0]->name);
}
if (ast_istype(exprs[0], ast_entfield)) {
out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
exprs[0],
(ast_expression*)parser_const_float_1(parser));
} else {
out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
exprs[0],
(ast_expression*)parser_const_float_1(parser));
}
break;
case opid3('S','+','+'):
case opid3('S','-','-'):
/* prefix ++ */
if (exprs[0]->expression.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;
}
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ast_ctx(exprs[0]), "assignment to constant `%s`", asvalue[0]->name);
}
if (ast_istype(exprs[0], ast_entfield)) {
out = (ast_expression*)ast_binstore_new(ctx, INSTR_STOREP_F, addop,
exprs[0],
(ast_expression*)parser_const_float_1(parser));
} else {
out = (ast_expression*)ast_binstore_new(ctx, INSTR_STORE_F, addop,
exprs[0],
(ast_expression*)parser_const_float_1(parser));
}
if (!out)
return false;
out = (ast_expression*)ast_binary_new(ctx, subop,
out,
(ast_expression*)parser_const_float_1(parser));
break;
case opid2('+','='):
case opid2('-','='):
if (exprs[0]->expression.vtype != exprs[1]->expression.vtype ||
(exprs[0]->expression.vtype != TYPE_VECTOR && exprs[0]->expression.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;
}
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ctx, "assignment to constant `%s`", asvalue[0]->name);
}
if (ast_istype(exprs[0], ast_entfield))
assignop = type_storep_instr[exprs[0]->expression.vtype];
else
assignop = type_store_instr[exprs[0]->expression.vtype];
switch (exprs[0]->expression.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]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
};
break;
case opid2('*','='):
case opid2('/','='):
if (exprs[1]->expression.vtype != TYPE_FLOAT ||
!(exprs[0]->expression.vtype == TYPE_FLOAT ||
exprs[0]->expression.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;
}
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ctx, "assignment to constant `%s`", asvalue[0]->name);
}
if (ast_istype(exprs[0], ast_entfield))
assignop = type_storep_instr[exprs[0]->expression.vtype];
else
assignop = type_store_instr[exprs[0]->expression.vtype];
switch (exprs[0]->expression.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 {
/* there's no DIV_VF */
if (CanConstFold1(exprs[1])) {
out = (ast_expression*)parser_const_float(parser, 1.0 / ConstF(1));
} else {
out = (ast_expression*)ast_binary_new(ctx, INSTR_DIV_F,
(ast_expression*)parser_const_float_1(parser),
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]->expression.vtype],
type_name[exprs[1]->expression.vtype]);
return false;
};
break;
case opid2('&','='):
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 expression: %s and %s",
ty1, ty2);
return false;
}
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ctx, "assignment to constant `%s`", asvalue[0]->name);
}
if (ast_istype(exprs[0], ast_entfield))
assignop = type_storep_instr[exprs[0]->expression.vtype];
else
assignop = type_store_instr[exprs[0]->expression.vtype];
out = (ast_expression*)ast_binstore_new(ctx, assignop,
(op->id == opid2('&','=') ? INSTR_BITAND : INSTR_BITOR),
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)) {
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]->expression.vtype];
else
assignop = type_store_instr[exprs[0]->expression.vtype];
out = (ast_expression*)ast_binary_new(ctx, INSTR_BITAND, exprs[0], exprs[1]);
if (!out)
return false;
if (ast_istype(exprs[0], ast_value) && asvalue[0]->cvq == CV_CONST) {
compile_error(ctx, "assignment to constant `%s`", asvalue[0]->name);
}
asbinstore = ast_binstore_new(ctx, assignop, INSTR_SUB_F, exprs[0], out);
asbinstore->keep_dest = true;
out = (ast_expression*)asbinstore;
break;
case opid2('~', 'P'):
if (exprs[0]->expression.vtype != TYPE_FLOAT) {
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(CanConstFold1(exprs[0]))
out = (ast_expression*)parser_const_float(parser, ~(qcint)ConstF(0));
else
out = (ast_expression*)ast_binary_new(ctx, INSTR_SUB_F, (ast_expression*)parser_const_float_neg1(parser), 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;
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;
}
fun = sy->out[fid].out;
if (fun == intrinsic_debug_typestring) {
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*)parser_const_string(parser, ty, false));
vec_shrinkby(sy->out, 1);
return true;
}
call = ast_call_new(sy->ops[vec_size(sy->ops)].ctx, fun);
if (!call)
return false;
if (fid+1 < vec_size(sy->out))
++paramcount;
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)
vec_push(call->params, sy->out[fid+1 + i].out);
vec_shrinkby(sy->out, paramcount);
(void)!ast_call_check_types(call);
if (parser->max_param_count < paramcount)
parser->max_param_count = paramcount;
if (ast_istype(fun, ast_value)) {
funval = (ast_value*)fun;
if ((fun->expression.flags & AST_FLAG_VARIADIC) &&
!(/*funval->cvq == CV_CONST && */ funval->hasvalue && funval->constval.vfunc->builtin))
{
call->va_count = (ast_expression*)parser_const_float(parser, (double)paramcount);
}
}
/* overwrite fid, the function, with a call */
sy->out[fid] = syexp(call->expression.node.context, (ast_expression*)call);
if (fun->expression.vtype != TYPE_FUNCTION) {
parseerror(parser, "not a function (%s)", type_name[fun->expression.vtype]);
return false;
}
if (!fun->expression.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->expression.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 (vec_size(fun->expression.params) != paramcount &&
!((fun->expression.flags & AST_FLAG_VARIADIC) &&
vec_size(fun->expression.params) < paramcount))
{
const char *fewmany = (vec_size(fun->expression.params) > 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)vec_size(fun->expression.params), (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)vec_size(fun->expression.params), (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;
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 ctx = parser_ctx(parser);
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 != ',') {
ast_unref(idx);
parseerror(parser, "expected comma after parameter index");
return NULL;
}
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);
if (!typevar) {
ast_unref(idx);
return NULL;
}
if (parser->tok != ')') {
ast_unref(idx);
ast_delete(typevar);
parseerror(parser, "expected closing paren");
return NULL;
}
#if 0
if (!parser_next(parser)) {
ast_unref(idx);
ast_delete(typevar);
parseerror(parser, "parse error after vararg");
return NULL;
}
#endif
if (!parser->function->varargs) {
ast_unref(idx);
ast_delete(typevar);
parseerror(parser, "function has no variable argument list");
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;
}
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 = parser_const_string(parser, 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_value *val;
val = parser_const_float(parser, (parser_token(parser)->constval.f));
if (!val)
return false;
vec_push(sy->out, syexp(parser_ctx(parser), (ast_expression*)val));
return true;
}
else if (parser->tok == TOKEN_INTCONST || parser->tok == TOKEN_CHARCONST) {
ast_value *val;
val = parser_const_float(parser, (double)(parser_token(parser)->constval.i));
if (!val)
return false;
vec_push(sy->out, syexp(parser_ctx(parser), (ast_expression*)val));
return true;
}
else if (parser->tok == TOKEN_STRINGCONST) {
ast_value *val;
val = parser_const_string(parser, parser_tokval(parser), false);
if (!val)
return false;
vec_push(sy->out, syexp(parser_ctx(parser), (ast_expression*)val));
return true;
}
else if (parser->tok == TOKEN_VECTORCONST) {
ast_value *val;
val = parser_const_vector(parser, parser_token(parser)->constval.v);
if (!val)
return false;
vec_push(sy->out, syexp(parser_ctx(parser), (ast_expression*)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('.') ||
(prev >= intrinsic_debug_typestring &&
prev <= intrinsic_debug_typestring))
{
/* When adding more intrinsics, fix the above condition */
prev = NULL;
}
if (prev && prev->expression.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;
vec_push(parser->labels, lbl);
}
}
if (!var) {
/* intrinsics */
if (!strcmp(parser_tokval(parser), "__builtin_debug_typestring")) {
var = (ast_expression*)intrinsic_debug_typestring;
}
if (!var) {
char *correct = NULL;
size_t i;
/*
* 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)) {
for (i = 0; i < sizeof(ftepp_predefs)/sizeof(*ftepp_predefs); i++) {
if (!strcmp(ftepp_predefs[i].name, parser_tokval(parser))) {
parseerror(parser, "unexpected ident: %s (use -fftepp-predef to enable pre-defined macros)", parser_tokval(parser));
return false;
}
}
}
/*
* TODO: determine the best score for the identifier: be it
* a variable, a field.
*
* We should also consider adding correction tables for
* other things as well.
*/
if (OPTS_OPTION_BOOL(OPTION_CORRECTION)) {
correction_t corr;
correct_init(&corr);
for (i = 0; i < vec_size(parser->correct_variables); i++) {
correct = correct_str(&corr, parser->correct_variables[i], parser_tokval(parser));
if (strcmp(correct, parser_tokval(parser))) {
break;
} else if (correct) {
mem_d(correct);
correct = NULL;
}
}
correct_free(&corr);
if (correct) {
parseerror(parser, "unexpected ident: %s (did you mean %s?)", parser_tokval(parser), correct);
mem_d(correct);
return false;
}
}
parseerror(parser, "unexpected ident: %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_truthvalue = 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");
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), "unknown 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];
#define IsAssignOp(x) (\
(x) == opid1('=') || \
(x) == opid2('+','=') || \
(x) == opid2('-','=') || \
(x) == opid2('*','=') || \
(x) == opid2('/','=') || \
(x) == opid2('%','=') || \
(x) == opid2('&','=') || \
(x) == opid2('|','=') || \
(x) == opid3('&','~','=') \
)
if (warn_truthvalue) {
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_truthvalue = 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;
#if 0
if (vec_size(sy.paren) && vec_last(sy.ops).isparen && vec_last(sy.paren) == PAREN_FUNC)
vec_last(sy.argc)++;
#endif
wantop = true;
}
else {
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)) {
parseerror(parser, "empty expression");
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));
/* corrector */
vec_push(parser->correct_variables, correct_trie_new());
vec_push(parser->correct_variables_score, NULL);
}
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));
correct_del(vec_last(parser->correct_variables), vec_last(parser->correct_variables_score));
vec_pop(parser->variables);
vec_pop(parser->correct_variables);
vec_pop(parser->correct_variables_score);
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);
/* corrector */
correct_add (
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
name
);
}
static void parser_addglobal(parser_t *parser, const char *name, ast_expression *e)
{
vec_push(parser->globals, e);
util_htset(parser->htglobals, name, e);
/* corrector */
correct_add (
parser->correct_variables[0],
&parser->correct_variables_score[0],
name
);
}
static ast_expression* process_condition(parser_t *parser, ast_expression *cond, bool *_ifnot)
{
bool ifnot = false;
ast_unary *unary;
ast_expression *prev;
if (cond->expression.vtype == TYPE_VOID || cond->expression.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->expression.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->expression.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;
while (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 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_delete(cond);
return false;
}
/* parse into the 'then' branch */
if (!parser_next(parser)) {
parseerror(parser, "expected statement for on-true branch of 'if'");
ast_delete(cond);
return false;
}
if (!parse_statement_or_block(parser, &ontrue)) {
ast_delete(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_delete(cond);
return false;
}
if (!parse_statement_or_block(parser, &onfalse)) {
ast_delete(ontrue);
ast_delete(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;
}
vec_push(parser->breaks, label);
vec_push(parser->continues, label);
rv = parse_while_go(parser, block, out);
if (label)
mem_d(label);
if (vec_last(parser->breaks) != label || vec_last(parser->continues) != label) {
parseerror(parser, "internal error: label stack corrupted");
rv = false;
ast_delete(*out);
*out = NULL;
}
else {
vec_pop(parser->breaks);
vec_pop(parser->continues);
}
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 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_delete(cond);
return false;
}
/* parse into the 'then' branch */
if (!parser_next(parser)) {
parseerror(parser, "expected while-loop body");
ast_delete(cond);
return false;
}
if (!parse_statement_or_block(parser, &ontrue)) {
ast_delete(cond);
return false;
}
cond = process_condition(parser, cond, &ifnot);
if (!cond) {
ast_delete(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;
}
}
vec_push(parser->breaks, label);
vec_push(parser->continues, label);
rv = parse_dowhile_go(parser, block, out);
if (label)
mem_d(label);
if (vec_last(parser->breaks) != label || vec_last(parser->continues) != label) {
parseerror(parser, "internal error: label stack corrupted");
rv = false;
ast_delete(*out);
*out = NULL;
}
else {
vec_pop(parser->breaks);
vec_pop(parser->continues);
}
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 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_delete(cond);
return false;
}
/* parse on */
if (!parser_next(parser) || parser->tok != ';') {
parseerror(parser, "expected semicolon after condition");
ast_delete(ontrue);
ast_delete(cond);
return false;
}
if (!parser_next(parser)) {
parseerror(parser, "parse error");
ast_delete(ontrue);
ast_delete(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;
}
vec_push(parser->breaks, label);
vec_push(parser->continues, label);
rv = parse_for_go(parser, block, out);
if (label)
mem_d(label);
if (vec_last(parser->breaks) != label || vec_last(parser->continues) != label) {
parseerror(parser, "internal error: label stack corrupted");
rv = false;
ast_delete(*out);
*out = NULL;
}
else {
vec_pop(parser->breaks);
vec_pop(parser->continues);
}
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 retval = true;
bool ifnot = false;
lex_ctx 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 0
if (OPTS_OPTION_U32(OPTION_STANDARD) != COMPILER_GMQCC) {
if (parsewarning(parser, WARN_EXTENSIONS,
"current standard does not allow variable declarations in for-loop initializers"))
goto onerr;
}
#endif
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 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))
retval = false;
return retval;
onerr:
if (initexpr) ast_delete(initexpr);
if (cond) ast_delete(cond);
if (increment) ast_delete(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_return *ret = NULL;
ast_value *expected = parser->function->vtype;
lex_ctx ctx = parser_ctx(parser);
(void)block; /* not touching */
if (!parser_next(parser)) {
parseerror(parser, "expected return expression");
return false;
}
if (parser->tok != ';') {
exp = parse_expression(parser, false, false);
if (!exp)
return false;
if (exp->expression.vtype != TYPE_NIL &&
exp->expression.vtype != expected->expression.next->expression.vtype)
{
parseerror(parser, "return with invalid expression");
}
ret = ast_return_new(ctx, exp);
if (!ret) {
ast_delete(exp);
return false;
}
} else {
if (!parser_next(parser))
parseerror(parser, "parse error");
if (expected->expression.next->expression.vtype != TYPE_VOID) {
(void)!parsewarning(parser, WARN_MISSING_RETURN_VALUES, "return without value");
}
ret = ast_return_new(ctx, NULL);
}
*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 ctx = parser_ctx(parser);
const char **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 (!vec_size(loops)) {
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 = vec_size(loops);
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
*/
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;
*cvq = CV_NONE;
for (;;) {
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;
}
if (!strcmp(parser_tokval(parser), "noreturn")) {
flags |= AST_FLAG_NORETURN;
if (!parser_next(parser) || parser->tok != TOKEN_ATTRIBUTE_CLOSE) {
parseerror(parser, "`noreturn` attribute has no parameters, expected `]]`");
*cvq = CV_WRONG;
return false;
}
}
else 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), "inline")) {
flags |= AST_FLAG_INLINE;
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
{
/* 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;
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;
}
vec_push(parser->breaks, label);
rv = parse_switch_go(parser, block, out);
if (label)
mem_d(label);
if (vec_last(parser->breaks) != label) {
parseerror(parser, "internal error: label stack corrupted");
rv = false;
ast_delete(*out);
*out = NULL;
}
else {
vec_pop(parser->breaks);
}
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 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, false, 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, false, 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;
vec_push(switchnode->cases, 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");
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
vec_push(parser->gotos, 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));
return false;
}
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 == '.')
{
/* 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, true, 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");
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);
vec_push(parser->labels, label);
}
*out = (ast_expression*)label;
if (!parser_next(parser)) {
parseerror(parser, "parse error after label");
return false;
}
for (i = 0; i < vec_size(parser->gotos); ++i) {
if (!strcmp(parser->gotos[i]->name, label->name)) {
ast_goto_set_label(parser->gotos[i], label);
vec_remove(parser->gotos, i, 1);
--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 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)
{
qcfloat num = 0;
ast_value **values = NULL;
ast_value *var = NULL;
ast_value *asvalue;
ast_expression *old;
if (!parser_next(parser) || parser->tok != '{') {
parseerror(parser, "expected `{` after `enum` keyword");
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;
var->constval.vfloat = 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;
}
}
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;
size_t parami;
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 (vec_size(parser->gotos) || vec_size(parser->labels)) {
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->expression.next;
thinkfunc = ast_value_new(parser_ctx(parser), parser_tokval(parser), functype->expression.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) {
lex_ctx 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;
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*)parser_const_float(parser, 0.1));
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) {
parseerror(parser, "function `%s` declared with multiple bodies", var->name);
ast_block_delete(block);
goto enderr;
}
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;
}
vec_push(parser->functions, func);
parser_enterblock(parser);
for (parami = 0; parami < vec_size(var->expression.params); ++parami) {
size_t e;
ast_value *param = var->expression.params[parami];
ast_member *me[3];
if (param->expression.vtype != TYPE_VECTOR &&
(param->expression.vtype != TYPE_FIELD ||
param->expression.next->expression.vtype != TYPE_VECTOR))
{
continue;
}
if (!create_vector_members(param, 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) {
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) {
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;
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;
}
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 = parser_const_float(parser, vec_size(var->expression.params));
}
parser->function = func;
if (!parse_block_into(parser, block)) {
ast_block_delete(block);
goto enderrfn;
}
vec_push(func->blocks, 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);
vec_pop(parser->functions);
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 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*)parser_const_float(parser, middle));
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 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->expression.vtype == TYPE_VECTOR)
assignop = INSTR_STORE_V;
subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
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 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->expression.vtype == TYPE_VECTOR)
assignop = INSTR_STOREP_V;
subscript = ast_array_index_new(ctx, (ast_expression*)array, (ast_expression*)parser_const_float(parser, from));
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 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*)parser_const_float(parser, from));
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;
}
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;
}
vec_push(func->blocks, body);
*out = fval;
vec_push(parser->accessors, 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), "<void>", 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 */
vec_push(fval->expression.params, index);
vec_push(fval->expression.params, 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), "<void>", 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 */
vec_push(fval->expression.params, entity);
vec_push(fval->expression.params, index);
vec_push(fval->expression.params, 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;
}
vec_push(fval->expression.params, 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_typename(parser_t *parser, ast_value **storebase, ast_value *cached_typedef);
static ast_value *parse_parameter_list(parser_t *parser, ast_value *var)
{
lex_ctx ctx;
size_t i;
ast_value **params;
ast_value *param;
ast_value *fval;
bool first = true;
bool variadic = false;
ast_value *varparam = NULL;
char *argcounter = NULL;
ctx = parser_ctx(parser);
/* for the sake of less code we parse-in in this function */
if (!parser_next(parser)) {
parseerror(parser, "expected parameter list");
return NULL;
}
params = NULL;
/* parse variables until we hit a closing paren */
while (parser->tok != ')') {
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;
if (parser->tok == TOKEN_DOTS) {
/* '...' indicates a varargs function */
variadic = true;
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));
if (!parser_next(parser) || parser->tok != ')') {
parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
goto on_error;
}
}
}
else
{
/* for anything else just parse a typename */
param = parse_typename(parser, NULL, NULL);
if (!param)
goto on_error;
vec_push(params, 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 = vec_last(params);
vec_pop(params);
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));
if (!parser_next(parser) || parser->tok != ')') {
parseerror(parser, "`...` must be the last parameter of a variadic function declaration");
goto on_error;
}
}
}
}
}
if (vec_size(params) == 1 && params[0]->expression.vtype == TYPE_VOID)
vec_free(params);
/* sanity check */
if (vec_size(params) > 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()>", 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;
params = NULL;
return var;
on_error:
if (argcounter)
mem_d(argcounter);
ast_delete(var);
for (i = 0; i < vec_size(params); ++i)
ast_delete(params[i]);
vec_free(params);
return NULL;
}
static ast_value *parse_arraysize(parser_t *parser, ast_value *var)
{
ast_expression *cexp;
ast_value *cval, *tmp;
lex_ctx ctx;
ctx = parser_ctx(parser);
if (!parser_next(parser)) {
ast_delete(var);
parseerror(parser, "expected array-size");
return NULL;
}
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;
tmp = ast_value_new(ctx, "<type[]>", TYPE_ARRAY);
tmp->expression.next = (ast_expression*)var;
var = tmp;
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);
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.
*
* 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)
{
ast_value *var, *tmp;
lex_ctx ctx;
const char *name = NULL;
bool isfield = false;
bool wasarray = false;
size_t morefields = 0;
ctx = parser_ctx(parser);
/* types may start with a dot */
if (parser->tok == '.') {
isfield = true;
/* 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 (parser->tok == '.') {
++morefields;
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) {
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, "<type(from_def)>");
} else
var = ast_value_new(ctx, "<type>", 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:f>", TYPE_FIELD);
tmp->expression.next = (ast_expression*)*storebase;
*storebase = tmp;
}
}
/* there may be a name now */
if (parser->tok == TOKEN_IDENT) {
name = util_strdup(parser_tokval(parser));
/* parse on */
if (!parser_next(parser)) {
ast_delete(var);
parseerror(parser, "error after variable or field declaration");
return NULL;
}
}
/* now this may be an array */
if (parser->tok == '[') {
wasarray = true;
var = parse_arraysize(parser, var);
if (!var)
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:f>", 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((void*)name);
return NULL;
}
}
/* finally name it */
if (name) {
if (!ast_value_set_name(var, name)) {
ast_delete(var);
parseerror(parser, "internal error: failed to set name");
return NULL;
}
/* free the name, ast_value_set_name duplicates */
mem_d((void*)name);
}
return var;
}
static bool parse_typedef(parser_t *parser)
{
ast_value *typevar, *oldtype;
ast_expression *old;
typevar = parse_typename(parser, NULL, NULL);
if (!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 "<INVALID>";
}
}
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 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];
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);
if (!var) {
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;
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;
/* 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 = vec_size(parser->globals);
was_end = true;
}
else if (!strcmp(var->name, "end_sys_fields")) {
var->uses++;
parser->crc_fields = vec_size(parser->fields);
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->expression.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 < vec_size(proto->expression.params); ++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;
}
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;
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;
}
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);
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->expression.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;
vec_push(parser->fields, (ast_expression*)var);
util_htset(parser->htfields, var->name, var);
if (isvector) {
for (i = 0; i < 3; ++i) {
vec_push(parser->fields, (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 (var->expression.vtype != find->expression.vtype) {
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;
}
/*
* add alias to aliases table and to corrector
* so corrections can apply for aliases as well.
*/
util_htset(parser->aliases, var->name, find);
/*
* add to corrector so corrections can work
* even for aliases too.
*/
correct_add (
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
var->name
);
/* 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]);
/*
* add to corrector so corrections can work
* even for aliases too.
*/
correct_add (
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
me[0]->name
);
correct_add (
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
me[1]->name
);
correct_add (
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
me[2]->name
);
}
}
}
} 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;
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);
/* corrector */
correct_add (
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
var->name
);
/* now rename the global */
ln = strlen(var->name);
vec_append(defname, ln, var->name);
ast_value_set_name(var, defname);
/* push it to the to-be-generated globals */
vec_push(parser->globals, (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]));
/* corrector */
correct_add(
vec_last(parser->correct_variables),
&vec_last(parser->correct_variables_score),
me[i]->name
);
vec_shrinkto(defname, prefix_len);
ln = strlen(me[i]->name);
vec_append(defname, ln, me[i]->name);
ast_member_set_name(me[i], defname);
vec_push(parser->globals, (ast_expression*)me[i]);
}
}
vec_free(defname);
} else {
vec_push(localblock->locals, 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]);
}
}
}
}
}
me[0] = me[1] = me[2] = NULL;
cleanvar = false;
/* Part 2.2
* deal with arrays
*/
if (var->expression.vtype == TYPE_ARRAY) {
char name[1024];
snprintf(name, sizeof(name), "%s##SET", var->name);
if (!parser_create_array_setter(parser, var, name))
goto cleanup;
snprintf(name, sizeof(name), "%s##GET", var->name);
if (!parser_create_array_getter(parser, var, var->expression.next, name))
goto cleanup;
}
else if (!localblock && !nofields &&
var->expression.vtype == TYPE_FIELD &&
var->expression.next->expression.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;
}
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;
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 != '{') {
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;
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 (parser->tok != TOKEN_INTCONST) {
parseerror(parser, "builtin number must be an integer constant");
break;
}
if (parser_token(parser)->constval.i < 0) {
parseerror(parser, "builtin number must be an integer greater than zero");
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;
}
vec_push(parser->functions, func);
func->builtin = -parser_token(parser)->constval.i-1;
}
if (!parser_next(parser)) {
parseerror(parser, "expected comma or semicolon");
if (func)
ast_function_delete(func);
var->constval.vfunc = NULL;
break;
}
}
else if (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 (i = 0; i < vec_size(parser->gotos); ++i)
parseerror(parser, "undefined label: `%s`", parser->gotos[i]->name);
vec_free(parser->gotos);
vec_free(parser->labels);
return true;
} else {
ast_expression *cexp;
ast_value *cval;
cexp = parse_expression_leave(parser, true, false, false);
if (!cexp)
break;
if (!localblock) {
cval = (ast_value*)cexp;
if (cval != parser->nil &&
(!ast_istype(cval, ast_value) || ((!cval->hasvalue || cval->cvq != CV_CONST) && !cval->isfield))
)
{
parseerror(parser, "cannot initialize a global constant variable with a non-constant expression");
}
else
{
if (!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);
var->cvq = cvq;
}
}
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 == '.')
{
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, true, 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)
{
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->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, "} entvars_t;\n\n");
code_crc = crc;
}
static parser_t *parser;
bool parser_init()
{
lex_ctx empty_ctx;
size_t i;
parser = (parser_t*)mem_a(sizeof(parser_t));
if (!parser)
return false;
memset(parser, 0, sizeof(*parser));
for (i = 0; i < operator_count; ++i) {
if (operators[i].id == opid1('=')) {
parser->assign_op = operators+i;
break;
}
}
if (!parser->assign_op) {
printf("internal error: initializing parser: failed to find assign operator\n");
mem_d(parser);
return false;
}
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);
/* corrector */
vec_push(parser->correct_variables, correct_trie_new());
vec_push(parser->correct_variables_score, NULL);
empty_ctx.file = "<internal>";
empty_ctx.line = 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, "<vector.x>", TYPE_NOEXPR);
parser->const_vec[1] = ast_value_new(empty_ctx, "<vector.y>", TYPE_NOEXPR);
parser->const_vec[2] = ast_value_new(empty_ctx, "<vector.z>", 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;
}
return true;
}
bool parser_compile()
{
/* 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 eof");
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(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();
}
bool parser_compile_string(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();
}
void parser_cleanup()
{
size_t i;
for (i = 0; i < vec_size(parser->accessors); ++i) {
ast_delete(parser->accessors[i]->constval.vfunc);
parser->accessors[i]->constval.vfunc = NULL;
ast_delete(parser->accessors[i]);
}
for (i = 0; i < vec_size(parser->functions); ++i) {
ast_delete(parser->functions[i]);
}
for (i = 0; i < vec_size(parser->imm_vector); ++i) {
ast_delete(parser->imm_vector[i]);
}
for (i = 0; i < vec_size(parser->imm_string); ++i) {
ast_delete(parser->imm_string[i]);
}
for (i = 0; i < vec_size(parser->imm_float); ++i) {
ast_delete(parser->imm_float[i]);
}
for (i = 0; i < vec_size(parser->fields); ++i) {
ast_delete(parser->fields[i]);
}
for (i = 0; i < vec_size(parser->globals); ++i) {
ast_delete(parser->globals[i]);
}
vec_free(parser->accessors);
vec_free(parser->functions);
vec_free(parser->imm_vector);
vec_free(parser->imm_string);
vec_free(parser->imm_float);
vec_free(parser->globals);
vec_free(parser->fields);
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);
/* corrector */
for (i = 0; i < vec_size(parser->correct_variables); ++i) {
correct_del(parser->correct_variables[i], parser->correct_variables_score[i]);
}
vec_free(parser->correct_variables);
vec_free(parser->correct_variables_score);
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);
vec_free(parser->labels);
vec_free(parser->gotos);
vec_free(parser->breaks);
vec_free(parser->continues);
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]);
util_htdel(parser->aliases);
mem_d(parser);
}
bool parser_finish(const char *output)
{
size_t i;
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 (i = 0; i < vec_size(parser->fields); ++i) {
ast_value *field;
bool hasvalue;
if (!ast_istype(parser->fields[i], ast_value))
continue;
field = (ast_value*)parser->fields[i];
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->expression.vtype);
if (subtype->expression.vtype == TYPE_FIELD)
ifld->fieldtype = subtype->expression.next->expression.vtype;
else if (subtype->expression.vtype == TYPE_FUNCTION)
ifld->outtype = subtype->expression.next->expression.vtype;
(void)!ir_value_set_field(field->ir_v, ifld);
}
}
for (i = 0; i < vec_size(parser->globals); ++i) {
ast_value *asvalue;
if (!ast_istype(parser->globals[i], ast_value))
continue;
asvalue = (ast_value*)(parser->globals[i]);
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 (i = 0; i < vec_size(parser->functions); ++i) {
ast_function *f = parser->functions[i];
if (f->varargs) {
if (parser->max_param_count > vec_size(f->vtype->expression.params)) {
f->varargs->expression.count = parser->max_param_count - vec_size(f->vtype->expression.params);
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 */
for (i = 0; i < vec_size(parser->imm_float); ++i) {
if (!ast_global_codegen(parser->imm_float[i], ir, false)) {
con_out("failed to generate global %s\n", parser->imm_float[i]->name);
ir_builder_delete(ir);
return false;
}
}
for (i = 0; i < vec_size(parser->imm_string); ++i) {
if (!ast_global_codegen(parser->imm_string[i], ir, false)) {
con_out("failed to generate global %s\n", parser->imm_string[i]->name);
ir_builder_delete(ir);
return false;
}
}
for (i = 0; i < vec_size(parser->imm_vector); ++i) {
if (!ast_global_codegen(parser->imm_vector[i], ir, false)) {
con_out("failed to generate global %s\n", parser->imm_vector[i]->name);
ir_builder_delete(ir);
return false;
}
}
for (i = 0; i < vec_size(parser->globals); ++i) {
ast_value *asvalue;
if (!ast_istype(parser->globals[i], ast_value))
continue;
asvalue = (ast_value*)(parser->globals[i]);
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 (i = 0; i < vec_size(parser->fields); ++i) {
ast_value *asvalue;
asvalue = (ast_value*)(parser->fields[i]->expression.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 (i = 0; i < vec_size(parser->functions); ++i) {
ast_function *f = parser->functions[i];
if (!ast_function_codegen(f, ir)) {
con_out("failed to generate function %s\n", f->name);
ir_builder_delete(ir);
return false;
}
}
if (OPTS_OPTION_BOOL(OPTION_DUMP))
ir_builder_dump(ir, con_out);
for (i = 0; i < vec_size(parser->functions); ++i) {
if (!ir_function_finalize(parser->functions[i]->ir_func)) {
con_out("failed to finalize function %s\n", parser->functions[i]->name);
ir_builder_delete(ir);
return false;
}
}
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);
generate_checksum(parser);
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;
}