quakeforge/tools/qfcc/source/statements.c
Bill Currie 7f42677a34 [qfcc] Clean up the last general uses of expr's next
It's now meant only for ALLOC. Interestingly, when DEBUG_QF_MEMORY is
defined in expr.c, something breaks badly with vkgen (no sniffles out of
valgrind, though), but everything is fine with it not defined. It seems
there may be some unpleasant UB going on somewhere.
2023-10-02 21:38:16 +09:00

2730 lines
68 KiB
C

/*
statements.c
Internal statements
Copyright (C) 2011 Bill Currie <bill@taniwha.org>
Author: Bill Currie <bill@taniwha.org>
Date: 2011/06/18
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to:
Free Software Foundation, Inc.
59 Temple Place - Suite 330
Boston, MA 02111-1307, USA
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#ifdef HAVE_STRING_H
# include <string.h>
#endif
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif
#include <stdlib.h>
#include <inttypes.h>
#include "qfalloca.h"
#include "QF/alloc.h"
#include "QF/mathlib.h"
#include "QF/va.h"
#include "tools/qfcc/include/class.h"
#include "tools/qfcc/include/dags.h"
#include "tools/qfcc/include/defspace.h"
#include "tools/qfcc/include/diagnostic.h"
#include "tools/qfcc/include/dot.h"
#include "tools/qfcc/include/expr.h"
#include "tools/qfcc/include/function.h"
#include "tools/qfcc/include/method.h"
#include "tools/qfcc/include/options.h"
#include "tools/qfcc/include/qfcc.h"
#include "tools/qfcc/include/reloc.h"
#include "tools/qfcc/include/statements.h"
#include "tools/qfcc/include/strpool.h"
#include "tools/qfcc/include/symtab.h"
#include "tools/qfcc/include/type.h"
#include "tools/qfcc/include/value.h"
#include "tools/qfcc/source/qc-parse.h"
const char * const op_type_names[] = {
"op_def",
"op_value",
"op_label",
"op_temp",
"op_alias",
"op_nil",
"op_pseudo",
};
const char * const st_type_names[] = {
"st_none",
"st_alias",
"st_expr",
"st_assign",
"st_ptrassign",
"st_move",
"st_ptrmove",
"st_memset",
"st_ptrmemset",
"st_state",
"st_func",
"st_flow",
"st_address",
};
const char *
optype_str (op_type_e type)
{
if (type > op_temp)
return "<invalid op_type>";
return op_type_names[type];
}
static const char *
tempop_string (operand_t *tmpop)
{
tempop_t *tempop = &tmpop->tempop;
if (tempop->alias) {
return va (0, "<tmp %s %p:%d:%p:%d:%d>",
pr_type_name[tempop->type->type],
tmpop, tempop->users,
tempop->alias,
tempop->offset,
tempop->alias->tempop.users);
}
return va (0, "<tmp %s %p:%d>", pr_type_name[tempop->type->type],
tmpop, tempop->users);
}
const char *
operand_string (operand_t *op)
{
if (!op)
return "";
switch (op->op_type) {
case op_def:
return op->def->name;
case op_value:
return get_value_string (op->value);
case op_label:
return op->label->name;
case op_temp:
return tempop_string (op);
case op_alias:
{
const char *alias = operand_string (op->alias);
char *buf = alloca (strlen (alias) + 1);
strcpy (buf, alias);
return va (0, "alias(%s,%s)", pr_type_name[op->type->type],
buf);
}
case op_nil:
return "nil";
case op_pseudo:
return va (0, "pseudo: %s", op->pseudoop->name);
}
return ("??");
}
static void
_print_operand (operand_t *op)
{
switch (op->op_type) {
case op_def:
printf ("(%s) ", get_type_string (op->type));
printf ("%s", op->def->name);
break;
case op_value:
printf ("(%s) %s", get_type_string (op->type),
get_value_string (op->value));
break;
case op_label:
printf ("block %p", op->label->dest);
break;
case op_temp:
printf ("tmp (%s) %p", get_type_string (op->type), op);
if (op->tempop.def)
printf (" %s:%04x", op->tempop.def->name,
op->tempop.def->offset);
break;
case op_alias:
printf ("alias(%s,", get_type_string (op->type));
_print_operand (op->alias);
printf (")");
break;
case op_nil:
printf ("nil");
break;
case op_pseudo:
printf ("pseudo: %s", op->pseudoop->name);
break;
}
}
void
print_operand (operand_t *op)
{
_print_operand (op);
puts ("");
}
static void
print_operand_chain (const char *name, operand_t *op)
{
if (op) {
printf (" %s:", name);
while (op) {
printf (" ");
_print_operand (op);
op = op->next;
}
printf ("\n");
}
}
void
print_statement (statement_t *s)
{
printf ("(%s, ", s->opcode);
if (s->opa)
_print_operand (s->opa);
printf (", ");
if (s->opb)
_print_operand (s->opb);
printf (", ");
if (s->opc)
_print_operand (s->opc);
printf (")\n");
print_operand_chain ("use", s->use);
print_operand_chain ("def", s->def);
print_operand_chain ("kill", s->kill);
}
ALLOC_STATE (pseudoop_t, pseudoops);
ALLOC_STATE (sblock_t, sblocks);
ALLOC_STATE (statement_t, statements);
ALLOC_STATE (operand_t, operands);
sblock_t *
new_sblock (void)
{
sblock_t *sblock;
ALLOC (256, sblock_t, sblocks, sblock);
sblock->tail = &sblock->statements;
return sblock;
}
void
sblock_add_statement (sblock_t *sblock, statement_t *statement)
{
// this should normally be null, but might be inserting
statement->next = *sblock->tail;
*sblock->tail = statement;
sblock->tail = &statement->next;
}
statement_t *
new_statement (st_type_t type, const char *opcode, const expr_t *expr)
{
statement_t *statement;
ALLOC (256, statement_t, statements, statement);
statement->type = type;
statement->opcode = save_string (opcode);
statement->expr = expr;
statement->number = -1; // indicates flow analysis not done yet
return statement;
}
static void
statement_add_use (statement_t *s, operand_t *use)
{
if (use) {
use->next = s->use;
s->use = use;
}
}
static void
statement_add_def (statement_t *s, operand_t *def)
{
if (def) {
def->next = s->def;
s->def = def;
}
}
static void
statement_add_kill (statement_t *s, operand_t *kill)
{
if (kill) {
kill->next = s->kill;
s->kill = kill;
}
}
static pseudoop_t *
new_pseudoop (const char *name)
{
pseudoop_t *pseudoop;
ALLOC (256, pseudoop_t, pseudoops, pseudoop);
pseudoop->name = save_string (name);
return pseudoop;
}
static operand_t *
new_operand (op_type_e op, const expr_t *expr, void *return_addr)
{
operand_t *operand;
ALLOC (256, operand_t, operands, operand);
operand->op_type = op;
operand->expr = expr;
operand->return_addr = return_addr;
return operand;
}
static operand_t *
copy_operand (operand_t *src)
{
if (!src) {
return 0;
}
operand_t *cpy = new_operand (src->op_type, src->expr, 0);
*cpy = *src;
cpy->return_addr = __builtin_return_address (0);
return cpy;
}
void
free_operand (operand_t *op)
{
FREE (operands, op);
}
static void
free_statement (statement_t *s)
{
// if (s->opa)
// free_operand (s->opa);
// if (s->opb)
// free_operand (s->opb);
// if (s->opc)
// free_operand (s->opc);
FREE (statements, s);
}
void
free_sblock (sblock_t *sblock)
{
while (sblock->statements) {
statement_t *s = sblock->statements;
sblock->statements = s->next;
free_statement (s);
}
FREE (sblocks, sblock);
}
static operand_t *
pseudo_operand (pseudoop_t *pseudoop, const expr_t *expr)
{
operand_t *op;
op = new_operand (op_pseudo, expr, __builtin_return_address (0));
op->pseudoop = pseudoop;
op->size = 1;
return op;
}
operand_t *
nil_operand (type_t *type, const expr_t *expr)
{
operand_t *op;
op = new_operand (op_nil, expr, __builtin_return_address (0));
op->type = type;
op->size = type_size (type);
op->width = type_width (type);
return op;
}
operand_t *
def_operand (def_t *def, type_t *type, const expr_t *expr)
{
operand_t *op;
if (!type)
type = def->type;
op = new_operand (op_def, expr, __builtin_return_address (0));
op->type = type;
op->size = type_size (type);
op->width = type_width (type);
op->def = def;
return op;
}
operand_t *
return_operand (type_t *type, const expr_t *expr)
{
symbol_t *return_symbol;
return_symbol = make_symbol (".return", &type_param, pr.symtab->space,
sc_extern);
if (!return_symbol->table) {
symtab_addsymbol (pr.symtab, return_symbol);
}
def_t *return_def = return_symbol->s.def;
return def_operand (alias_def (return_def, type, 0), 0, expr);
}
operand_t *
value_operand (ex_value_t *value, const expr_t *expr)
{
operand_t *op;
op = new_operand (op_value, expr, __builtin_return_address (0));
op->type = value->type;
op->size = type_size (value->type);
op->width = type_width (value->type);
op->value = value;
return op;
}
operand_t *
temp_operand (const type_t *type, const expr_t *expr)
{
operand_t *op = new_operand (op_temp, expr, __builtin_return_address (0));
op->tempop.type = (type_t *) type;
op->type = (type_t *) type;
op->size = type_size (type);
op->width = type_width (type);
return op;
}
int
tempop_overlap (tempop_t *t1, tempop_t *t2)
{
int offs1 = t1->offset;
int offs2 = t2->offset;
int size1 = type_size (t1->type);
int size2 = type_size (t2->type);
if (t1->alias) {
offs1 += t1->alias->tempop.offset;
}
if (t2->alias) {
offs2 += t2->alias->tempop.offset;
}
if (offs1 <= offs2 && offs1 + size1 >= offs2 + size2)
return 2; // t1 fully overlaps t2
if (offs1 < offs2 + size2 && offs2 < offs1 + size1)
return 1; // t1 and t2 at least partially overlap
return 0;
}
int
tempop_visit_all (tempop_t *tempop, int overlap,
int (*visit) (tempop_t *, void *), void *data)
{
tempop_t *start_tempop = tempop;
operand_t *top;
int ret;
if ((ret = visit (tempop, data)))
return ret;
if (tempop->alias) {
top = tempop->alias;
if (top->op_type != op_temp) {
internal_error (top->expr, "temp alias of non-temp operand");
}
tempop = &top->tempop;
if (!(overlap & 4) && (ret = visit (tempop, data)))
return ret;
overlap &= ~4;
} else {
overlap = 0;
}
for (top = tempop->alias_ops; top; top = top->next) {
if (top->op_type != op_temp) {
internal_error (top->expr, "temp alias of non-temp operand");
}
tempop = &top->tempop;
if (tempop == start_tempop)
continue;
if (overlap && tempop_overlap (tempop, start_tempop) < overlap)
continue;
if ((ret = visit (tempop, data)))
return ret;
}
return 0;
}
operand_t *
offset_alias_operand (type_t *type, int offset, operand_t *aop, const expr_t *expr)
{
operand_t *top;
def_t *def;
if (type_compatible (aop->type, type)) {
if (offset) {
//For types to be compatible, they must be the same size, thus this
//seemingly mismatched error
internal_error (expr, "offset alias of same size type");
}
return aop;
}
if (aop->op_type == op_temp) {
while (aop->tempop.alias) {
aop = aop->tempop.alias;
if (aop->op_type != op_temp)
internal_error (expr, "temp alias of non-temp var");
if (aop->tempop.alias)
bug (expr, "aliased temp alias");
}
for (top = aop->tempop.alias_ops; top; top = top->next) {
if (top->type == type && top->tempop.offset == offset) {
break;
}
}
if (!top) {
top = temp_operand (type, expr);
top->tempop.alias = aop;
top->tempop.offset = offset;
top->next = aop->tempop.alias_ops;
aop->tempop.alias_ops = top;
}
return top;
} else if (aop->op_type == op_def) {
def = aop->def;
while (def->alias)
def = def->alias;
return def_operand (alias_def (def, type, offset), 0, expr);
} else if (aop->op_type == op_value) {
if (!is_ptr (aop->value->type)) {
auto value = offset_alias_value (aop->value, type, offset);
top = value_operand (value, expr);
} else {
// even an offset of 0 will break a pointer value because of
// relocations
if (offset) {
internal_error (expr, "offset alias of pointer value operand");
}
top = value_operand (aop->value, expr);
top->type = type;
}
return top;
} else {
internal_error (expr, "invalid alias target: %s: %s",
optype_str (aop->op_type), operand_string (aop));
}
}
operand_t *
alias_operand (type_t *type, operand_t *op, const expr_t *expr)
{
operand_t *aop;
if (type_size (type) != type_size (op->type)) {
internal_error (op->expr,
"aliasing operand with type of different size: %d, %d",
type_size (type), type_size (op->type));
}
aop = new_operand (op_alias, expr, __builtin_return_address (0));
aop->alias = op;
aop->type = type;
aop->size = type_size (type);
aop->width = type_width (type);
return aop;
}
operand_t *
label_operand (const expr_t *label)
{
operand_t *lop;
if (label->type != ex_label) {
internal_error (label, "not a label expression");
}
lop = new_operand (op_label, label, __builtin_return_address (0));
lop->label = (ex_label_t *) &label->label;
return lop;
}
static operand_t *
short_operand (short short_val, const expr_t *expr)
{
ex_value_t *val = new_short_val (short_val);
return value_operand (val, expr);
}
static const char *
convert_op (int op)
{
switch (op) {
case OR: return "or";
case AND: return "and";
case EQ: return "eq";
case NE: return "ne";
case LE: return "le";
case GE: return "ge";
case LT: return "lt";
case GT: return "gt";
case '+': return "add";
case '-': return "sub";
case '*': return "mul";
case '/': return "div";
case '%': return "rem";
case MOD: return "mod";
case '&': return "bitand";
case '|': return "bitor";
case '^': return "bitxor";
case '~': return "bitnot";
case '!': return "not";
case SHL: return "shl";
case SHR: return "shr";
case '.': return "load";
case CROSS: return "cross";
case WEDGE: return "wedge";
case DOT: return "dot";
case HADAMARD: return "mul";
case SCALE: return "scale";
default:
return 0;
}
}
int
statement_is_cond (statement_t *s)
{
if (!s)
return 0;
return !strncmp (s->opcode, "if", 2);
}
int
statement_is_goto (statement_t *s)
{
if (!s)
return 0;
return !strcmp (s->opcode, "jump") && !s->opb;
}
int
statement_is_jumpb (statement_t *s)
{
if (!s)
return 0;
return !strcmp (s->opcode, "jump") && s->opb;
}
int
statement_is_call (statement_t *s)
{
if (!s)
return 0;
if (!strncmp (s->opcode, "call", 4))
return 1;
if (!strncmp (s->opcode, "rcall", 5))
return 2;
return 0;
}
int
statement_is_return (statement_t *s)
{
if (!s)
return 0;
return !strncmp (s->opcode, "return", 6);
}
static ex_label_t **
statement_get_labelref (statement_t *s)
{
if (statement_is_cond (s)
|| statement_is_goto (s)
|| statement_is_jumpb (s)) {
return &s->opa->label;
}
return 0;
}
sblock_t *
statement_get_target (statement_t *s)
{
ex_label_t **label = statement_get_labelref (s);
return label ? (*label)->dest : 0;
}
sblock_t **
statement_get_targetlist (statement_t *s)
{
sblock_t **target_list;
int count = 0, i;
def_t *table = 0;
element_t *e;
if (statement_is_cond (s)) {
count = 1;
} else if (statement_is_goto (s)) {
count = 1;
} else if (statement_is_jumpb (s)) {
table = s->opa->def;
count = table->type->t.array.size;
}
target_list = malloc ((count + 1) * sizeof (sblock_t *));
target_list[count] = 0;
if (statement_is_cond (s)) {
target_list[0] = statement_get_target (s);
} else if (statement_is_goto (s)) {
target_list[0] = statement_get_target (s);
} else if (statement_is_jumpb (s)) {
if (table->alias)
internal_error (s->opa->expr, "aliased jump table");
e = table->initializer->compound.head; //FIXME check!!!
for (i = 0; i < count; e = e->next, i++)
target_list[i] = e->expr->labelref.label->dest;
}
return target_list;
}
static void
invert_conditional (statement_t *s)
{
if (!strcmp (s->opcode, "ifnz"))
s->opcode = "ifz";
else if (!strcmp (s->opcode, "ifz"))
s->opcode = "ifnz";
else if (!strcmp (s->opcode, "ifbe"))
s->opcode = "ifa";
else if (!strcmp (s->opcode, "ifb"))
s->opcode = "ifae";
else if (!strcmp (s->opcode, "ifae"))
s->opcode = "ifb";
else if (!strcmp (s->opcode, "ifa"))
s->opcode = "ifbe";
}
typedef sblock_t *(*statement_f) (sblock_t *, const expr_t *);
typedef sblock_t *(*expr_f) (sblock_t *, const expr_t *, operand_t **);
static sblock_t *statement_subexpr (sblock_t *sblock, const expr_t *e,
operand_t **op);
static sblock_t *expr_symbol (sblock_t *sblock, const expr_t *e, operand_t **op);
static sblock_t *expr_def (sblock_t *sblock, const expr_t *e, operand_t **op);
static sblock_t *statement_single (sblock_t *sblock, const expr_t *e);
static sblock_t *
expr_address (sblock_t *sblock, const expr_t *e, operand_t **op)
{
statement_t *s;
const expr_t *lvalue = e->address.lvalue;
const expr_t *offset = e->address.offset;
if (lvalue->type == ex_alias && offset && is_constant (offset)) {
lvalue = new_offset_alias_expr (lvalue->alias.type,
lvalue->alias.expr,
expr_int (offset));
offset = 0;
} else if (offset && is_constant (offset)) {
int o = expr_int (offset);
if (o < 32768 && o >= -32768) {
offset = new_short_expr (o);
//offset = expr_file_line (new_short_expr (o), offset);
}
}
s = new_statement (st_address, "lea", e);
sblock = statement_subexpr (sblock, lvalue, &s->opa);
if (offset) {
sblock = statement_subexpr (sblock, offset, &s->opb);
}
s->opc = temp_operand (e->address.type, e);
sblock_add_statement (sblock, s);
*(op) = s->opc;
return sblock;
}
static operand_t *
operand_address (operand_t *reference, const expr_t *e)
{
def_t *def;
type_t *type;
int offset = 0;
type = reference->type;
switch (reference->op_type) {
case op_def:
// assumes aliasing is only one level deep which should be the
// case
def = reference->def;
if (def->alias) {
offset = def->offset;
def = def->alias;
}
return value_operand (new_pointer_val (offset, type, def, 0), e);
case op_temp:
// assumes aliasing is only one level deep which should be the
// case
if (reference->tempop.alias) {
offset = reference->tempop.offset;
reference = reference->tempop.alias;
}
return value_operand (new_pointer_val (offset, type, 0,
reference), e);
case op_alias:
//op_alias comes only from alias_operand and that is called
// by dags, so not expected
case op_value:
case op_label:
case op_nil:
case op_pseudo:
break;
}
internal_error (e, "invalid operand type for operand address: %s",
op_type_names[reference->op_type]);
}
static __attribute__((pure)) int
is_indirect (const expr_t *e)
{
if ((e->type == ex_expr || e->type == ex_uexpr)
&& e->expr.op == '.') {
return 1;
}
return 0;
}
static sblock_t *addressing_mode (sblock_t *sblock, const expr_t *ref,
operand_t **base, operand_t **offset,
pr_ushort_t *mode, operand_t **target);
static statement_t *lea_statement (operand_t *pointer, operand_t *offset,
const expr_t *e);
static statement_t *
assign_statement (operand_t *dst, operand_t *src, const expr_t *e)
{
statement_t *s = new_statement (st_assign, "assign", e);
s->opa = dst;
s->opc = src;
return s;
}
static sblock_t *
expr_assign_copy (sblock_t *sblock, const expr_t *e, operand_t **op, operand_t *src)
{
statement_t *s;
const expr_t *dst_expr = e->assign.dst;
const expr_t *src_expr = e->assign.src;
type_t *dst_type = get_type (dst_expr);
type_t *src_type = get_type (src_expr);
unsigned count;
const expr_t *count_expr;
operand_t *dst = 0;
operand_t *size = 0;
static const char *opcode_sets[][2] = {
{"move", "movep"},
{"memset", "memsetp"},
};
const unsigned max_count = 1 << 16;
const char **opcode_set = opcode_sets[0];
const char *opcode;
int need_ptr = 0;
st_type_t type = st_move;
operand_t *use = 0;
operand_t *def = 0;
operand_t *kill = 0;
if ((src && src->op_type == op_nil) || src_expr->type == ex_nil) {
// switch to memset because nil is type agnostic 0 and structures
// can be any size
src_expr = new_int_expr (0);
sblock = statement_subexpr (sblock, src_expr, &src);
opcode_set = opcode_sets[1];
if (op) {
*op = nil_operand (dst_type, src_expr);
}
type = st_memset;
if (is_indirect (dst_expr)) {
need_ptr = 1;
}
} else {
if (is_indirect (src_expr)) {
//src_expr = expr_file_line (address_expr (src_expr, 0), e);
src_expr = address_expr (src_expr, 0);
need_ptr = 1;
}
if (!src) {
// This is the very right-hand node of a non-nil assignment chain
// (there may be more chains somewhere within src_expr, but they
// are not part of this chain as they are separated by another
// expression).
sblock = statement_subexpr (sblock, src_expr, &src);
}
// send the source operand back up through the assignment chain
// before modifying src if its address is needed
if (op) {
*op = src;
}
if (is_indirect (dst_expr)) {
if (is_variable (src_expr)) {
// FIXME this probably needs to be more agressive
// shouldn't emit code...
sblock = statement_subexpr (sblock, src_expr, &use);
}
if (options.code.progsversion == PROG_VERSION) {
// FIXME it was probably a mistake extracting the operand
// type from the statement expression in dags. Also, can't
// use address_expr() because src_expr may be a function call
// and unary_expr doesn't like that
src_expr = expr_file_line (
new_address_expr (get_type (src_expr), src_expr, 0),
src_expr);
s = lea_statement (src, 0, src_expr);
sblock_add_statement (sblock, s);
src = s->opc;
} else {
src = operand_address (src, src_expr);
}
need_ptr = 1;
}
}
if (need_ptr) {
// dst_expr and/or src_expr are dereferenced pointers, so need to
// un-dereference dst_expr to get the pointer and switch to movep
// or memsetp instructions.
if (is_variable (dst_expr)) {
// FIXME this probably needs to be more agressive
// shouldn't emit code...
sblock = statement_subexpr (sblock, dst_expr, &def);
//FIXME is this even necessary? if it is, should use copy_operand
sblock = statement_subexpr (sblock, dst_expr, &kill);
}
//dst_expr = expr_file_line (address_expr (dst_expr, 0), e);
dst_expr = address_expr (dst_expr, 0);
need_ptr = 1;
}
sblock = statement_subexpr (sblock, dst_expr, &dst);
if (type_size (dst_type) != type_size (src_type)) {
bug (e, "dst and src sizes differ in expr_assign_copy: %d %d",
type_size (dst_type), type_size (src_type));
}
count = min (type_size (dst_type), type_size (src_type));
if (count < (1 << 16)) {
//count_expr = expr_file_line (new_short_expr (count), e);
count_expr = new_short_expr (count);
} else {
//count_expr = expr_file_line (new_int_expr (count), e);
count_expr = new_int_expr (count);
}
sblock = statement_subexpr (sblock, count_expr, &size);
if (count < max_count && !need_ptr) {
opcode = opcode_set[0];
} else {
opcode = opcode_set[1];
type++; // from st_move/st_memset to st_ptrmove/st_ptrmemset
}
s = new_statement (type, opcode, e);
s->opa = src;
s->opb = size;
s->opc = dst;
s->use = use;
s->def = def;
s->kill = kill;
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
expr_assign (sblock_t *sblock, const expr_t *e, operand_t **op)
{
statement_t *s;
const expr_t *src_expr = e->assign.src;
const expr_t *dst_expr = e->assign.dst;
type_t *dst_type = get_type (dst_expr);
operand_t *src = 0;
operand_t *dst = 0;
operand_t *ofs = 0;
operand_t *target = 0;
pr_ushort_t mode = 0; // assign
const char *opcode = "assign";
st_type_t type = st_assign;
if (src_expr->type == ex_assign) {
sblock = statement_subexpr (sblock, src_expr, &src);
if (is_structural (dst_type) || dst_type->width > 4) {
return expr_assign_copy (sblock, e, op, src);
}
if (is_indirect (dst_expr)) {
goto dereference_dst;
} else {
sblock = statement_subexpr (sblock, dst_expr, &dst);
}
} else {
if (is_structural (dst_type) || dst_type->width > 4) {
return expr_assign_copy (sblock, e, op, src);
}
if (is_indirect (dst_expr)) {
// If both dst_expr and src_expr are indirect, then a staging temp
// is needed, but emitting src_expr first generates that temp
// because src is null. If src_expr is not indirect and is a simple
// variable reference, then just the ref will be generated and thus
// will be assigned to the dereferenced destination. If src_expr
// is not simple, then a temp will be generated, so all good.
sblock = statement_subexpr (sblock, src_expr, &src);
goto dereference_dst;
} else {
// dst_expr is direct and known to be an l-value, so emitting
// its expression will simply generate a reference to that l-value
// which will be used as the default location to store src_expr's
// result
sblock = statement_subexpr (sblock, dst_expr, &dst);
src = dst;
sblock = statement_subexpr (sblock, src_expr, &src);
}
}
if (0) {
dereference_dst:
// dst_expr is a dereferenced pointer, so need to get its addressing
// parameters (base and offset) and switch to store instructions.
sblock = addressing_mode (sblock, dst_expr, &dst, &ofs, &mode, &target);
if (mode != 0) {
opcode = "store";
type = st_ptrassign;
} else {
ofs = 0;
}
}
if (op) {
*op = src;
}
if (src == dst) {
return sblock;
}
if (options.code.progsversion < PROG_VERSION
&& is_entity (dst->type) && ofs && is_field (ofs->type)) {
// need to get a pointer type, entity.field expressions do not provide
// one directly. FIXME it was probably a mistake extracting the operand
// type from the statement expression in dags
//dst_expr = expr_file_line (address_expr (dst_expr, 0), dst_expr);
dst_expr = address_expr (dst_expr, 0);
s = new_statement (st_address, "lea", dst_expr);
s->opa = dst;
s->opb = ofs;
s->opc = temp_operand (&type_ptr, dst_expr);
sblock_add_statement (sblock, s);
dst = s->opc;
ofs = 0;
}
s = new_statement (type, opcode, e);
s->opa = dst;
s->opb = ofs;
s->opc = src;
if (type != st_ptrassign) {
statement_add_def (s, copy_operand (target));
statement_add_kill (s, target);
}
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
vector_call (sblock_t *sblock, const expr_t *earg, const expr_t *param, int ind,
operand_t **op)
{
//FIXME this should be done in the expression tree
const expr_t *a, *v, *n;
int i;
static const char *names[] = {"x", "y", "z"};
for (i = 0; i < 3; i++) {
n = new_name_expr (names[i]);
v = new_float_expr (earg->value->v.vector_val[i]);
a = assign_expr (field_expr (param, n), v);
param = new_param_expr (get_type (earg), ind);
//a->line = earg->line;
//a->file = earg->file;
sblock = statement_single (sblock, a);
}
sblock = statement_subexpr (sblock, param, op);
return sblock;
}
static sblock_t *
expr_call_v6p (sblock_t *sblock, const expr_t *call, operand_t **op)
{
const expr_t *func = call->branch.target;
const expr_t *args = call->branch.args;
const expr_t *param;
operand_t *arguments[2] = {0, 0};
int count = 0;
int ind;
const char *opcode;
const char *pref = "";
statement_t *s;
operand_t *use = 0;
// function arguments are in reverse order
for (auto li = args->list.head; li; li = li->next) {
auto a = li->expr;
if (a->type == ex_args) {
// v6p uses callN and pr_argc
continue;
}
count++;
}
ind = count;
for (auto li = args->list.head; li; li = li->next) {
auto a = li->expr;
if (a->type == ex_args) {
// v6p uses callN and pr_argc
continue;
}
ind--;
param = new_param_expr (get_type (a), ind);
if (count && options.code.progsversion != PROG_ID_VERSION && ind < 2) {
pref = "r";
sblock = statement_subexpr (sblock, param, &arguments[ind]);
if (options.code.vector_calls && a->type == ex_value
&& a->value->lltype == ev_vector)
sblock = vector_call (sblock, a, param, ind, &arguments[ind]);
else
sblock = statement_subexpr (sblock, a, &arguments[ind]);
operand_t *p;
sblock = statement_subexpr (sblock, param, &p);
p->next = use;
use = p;
continue;
}
if (is_struct (get_type (param)) || is_union (get_type (param))) {
const expr_t *mov = assign_expr (param, a);
//mov->line = a->line;
//mov->file = a->file;
sblock = statement_single (sblock, mov);
} else {
operand_t *p = 0;
sblock = statement_subexpr (sblock, param, &p);
if (options.code.vector_calls && a->type == ex_value
&& a->value->lltype == ev_vector) {
sblock = vector_call (sblock, a, param, ind, 0);
} else {
operand_t *arg = p;
arg = p;
sblock = statement_subexpr (sblock, a, &arg);
if (arg != p) {
s = assign_statement (p, arg, a);
sblock_add_statement (sblock, s);
}
}
p->next = use;
use = p;
}
}
opcode = va (0, "%scall%d", pref, count);
s = new_statement (st_func, opcode, call);
sblock = statement_subexpr (sblock, func, &s->opa);
s->opb = arguments[0];
s->opc = arguments[1];
s->use = use;
if (op) {
*op = return_operand (call->branch.ret_type, call);
}
sblock_add_statement (sblock, s);
sblock->next = new_sblock ();
return sblock->next;
}
static sblock_t *
expr_call (sblock_t *sblock, const expr_t *call, operand_t **op)
{
if (options.code.progsversion < PROG_VERSION) {
return expr_call_v6p (sblock, call, op);
}
if (!current_func->arguments) {
current_func->arguments = defspace_new (ds_virtual);
}
defspace_t *arg_space = current_func->arguments;
const expr_t *func = call->branch.target;
const expr_t *args_va_list = 0; // .args (...) parameter
const expr_t *args_params = 0; // first arg in ...
operand_t *use = 0;
operand_t *kill = 0;
int num_params = 0;
defspace_reset (arg_space);
int num_args = list_count (&call->branch.args->list);
const expr_t *args[num_args];
list_scatter_rev (&call->branch.args->list, args);
int arg_num = 0;
for (int i = 0; i < num_args; i++) {
const expr_t *a = args[i];
const char *arg_name = va (0, ".arg%d", arg_num++);
def_t *def = new_def (arg_name, 0, current_func->arguments,
sc_argument);
type_t *arg_type = get_type (a);
int size = type_size (arg_type);
int alignment = arg_type->alignment;
if (alignment < 4) {
alignment = 4;
}
def->offset = defspace_alloc_aligned_highwater (arg_space, size,
alignment);
def->type = arg_type;
def->reg = current_func->temp_reg;
const expr_t *def_expr = expr_file_line (new_def_expr (def), call);
if (a->type == ex_args) {
args_va_list = def_expr;
} else {
if (args_va_list && !args_params) {
args_params = def_expr;
}
if (args_va_list) {
num_params++;
}
const expr_t *assign = assign_expr (def_expr, a);
//expr_file_line (assign, call);
sblock = statement_single (sblock, assign);
}
// The call both uses and kills the arguments: use is obvious, but kill
// is because the callee has direct access to them and might modify
// them
// need two ops for the one def because there's two lists
operand_t *u = def_operand (def, arg_type, call);
operand_t *k = def_operand (def, arg_type, call);
u->next = use;
use = u;
k->next = kill;
kill = k;
}
if (args_va_list) {
const expr_t *assign;
const expr_t *count;
const expr_t *list;
const expr_t *args_count = field_expr (args_va_list,
new_name_expr ("count"));
const expr_t *args_list = field_expr (args_va_list,
new_name_expr ("list"));
//expr_file_line (args_count, call);
//expr_file_line (args_list, call);
count = new_short_expr (num_params);
assign = assign_expr (args_count, count);
//expr_file_line (assign, call);
sblock = statement_single (sblock, assign);
if (args_params) {
list = address_expr (args_params, &type_param);
} else {
list = new_nil_expr ();
}
//expr_file_line (list, call);
assign = assign_expr (args_list, list);
//expr_file_line (assign, call);
sblock = statement_single (sblock, assign);
}
statement_t *s = new_statement (st_func, "call", call);
sblock = statement_subexpr (sblock, func, &s->opa);
if (!op) {
s->opc = short_operand (0, call);
} else {
if (!*op) {
*op = temp_operand (call->branch.ret_type, call);
}
s->opc = *op;
}
s->use = use;
s->kill = kill;
sblock_add_statement (sblock, s);
sblock->next = new_sblock ();
return sblock->next;
}
static sblock_t *
expr_branch (sblock_t *sblock, const expr_t *e, operand_t **op)
{
if (e->branch.type != pr_branch_call) {
internal_error (e, "unexpected branch type in expression: %d",
e->branch.type);
}
return expr_call (sblock, e, op);
}
static sblock_t *
statement_branch (sblock_t *sblock, const expr_t *e)
{
static const char *opcodes[] = {
"ifz",
"ifb",
"ifa",
0, // special handling
"ifnz",
"ifae",
"ifbe",
0, // not used here
};
statement_t *s = 0;
const char *opcode;
if (e->branch.type == pr_branch_call) {
return expr_call (sblock, e, 0);
}
if (e->branch.type == pr_branch_jump) {
if (e->branch.index) {
s = new_statement (st_flow, "jump", e);
sblock = statement_subexpr (sblock, e->branch.target, &s->opa);
sblock = statement_subexpr (sblock, e->branch.index, &s->opb);
} else {
s = new_statement (st_flow, "jump", e);
s->opa = label_operand (e->branch.target);
}
} else {
opcode = opcodes [e->branch.type];
s = new_statement (st_flow, opcode, e);
sblock = statement_subexpr (sblock, e->branch.test, &s->opc);
s->opa = label_operand (e->branch.target);
}
sblock_add_statement (sblock, s);
sblock->next = new_sblock ();
return sblock->next;
}
static operand_t *
find_def_operand (const expr_t *ref)
{
operand_t *op = 0;
if (ref->type == ex_alias) {
return find_def_operand (ref->alias.expr);
} else if (ref->type == ex_address) {
return find_def_operand (ref->address.lvalue);
} else if (ref->type == ex_symbol) {
expr_symbol (0, ref, &op);
} else if (ref->type == ex_def) {
expr_def (0, ref, &op);
} else if (ref->type == ex_expr) {
debug (ref, "unexpected expr type: %s", expr_names[ref->type]);
} else {
internal_error (ref, "unexpected expr type: %s",
expr_names[ref->type]);
}
return op;
}
static sblock_t *
ptr_addressing_mode (sblock_t *sblock, const expr_t *ref,
operand_t **base, operand_t **offset, pr_ushort_t *mode,
operand_t **target)
{
type_t *type = get_type (ref);
if (!is_ptr (type)) {
internal_error (ref, "expected pointer in ref");
}
if (ref->type == ex_address
&& (!ref->address.offset || is_constant (ref->address.offset))
&& ref->address.lvalue->type == ex_alias
&& (!ref->address.lvalue->alias.offset
|| is_constant (ref->address.lvalue->alias.offset))) {
const expr_t *lvalue = ref->address.lvalue;
const expr_t *offs = ref->address.offset;
const expr_t *alias;
if (lvalue->alias.offset) {
if (offs) {
offs = binary_expr ('+', offs, lvalue->alias.offset);
} else {
offs = lvalue->alias.offset;
}
}
type = type->t.fldptr.type;
if (offs) {
const expr_t *lv = lvalue->alias.expr;
type_t *lvtype = get_type (lv);
int o = expr_int (offs);
if (o < 0 || o + type_size (type) > type_size (lvtype)) {
// not a valid offset for the type, which technically should
// be an error, but there may be legitimate reasons for doing
// such pointer shenanigans
goto just_a_pointer;
}
alias = new_offset_alias_expr (type, lv, expr_int (offs));
} else {
alias = new_alias_expr (type, lvalue->alias.expr);
}
//expr_file_line (alias, ref);
return addressing_mode (sblock, alias, base, offset, mode, target);
} else if (ref->type != ex_alias || ref->alias.offset) {
// probably just a pointer
just_a_pointer:
sblock = statement_subexpr (sblock, ref, base);
*offset = short_operand (0, ref);
*mode = 2; // mode C: ptr + constant index
} else if (is_ptr (get_type (ref->alias.expr))) {
// cast of one pointer type to another
return ptr_addressing_mode (sblock, ref->alias.expr, base, offset,
mode, target);
} else {
// alias with no offset
if (!is_integral (get_type (ref->alias.expr))) {
internal_error (ref, "expected integer expr in ref");
}
const expr_t *intptr = ref->alias.expr;
if (intptr->type != ex_expr
|| (intptr->expr.op != '+'
&& intptr->expr.op != '-')) {
// treat ref as simple pointer
sblock = statement_subexpr (sblock, ref, base);
*offset = short_operand (0, ref);
*mode = 2; // mode C: ptr + constant index
} else {
const expr_t *ptr = intptr->expr.e1;
const expr_t *offs = intptr->expr.e2;
int const_offs;
if (target) {
if (ptr->type == ex_alias
&& is_ptr (get_type (ptr->alias.expr))) {
*target = find_def_operand (ptr->alias.expr);
}
}
// move the +/- to the offset
offs = unary_expr (intptr->expr.op, offs);
// make the base a pointer again
ptr = new_alias_expr (ref->alias.type, ptr);
sblock = statement_subexpr (sblock, ptr, base);
if (is_constant (offs)
&& (const_offs = expr_int (offs)) < 32768
&& const_offs >= -32768) {
*mode = 2;
*offset = short_operand (const_offs, ref);
} else {
*mode = 3;
sblock = statement_subexpr (sblock, offs, offset);
}
}
}
return sblock;
}
static sblock_t *
addressing_mode (sblock_t *sblock, const expr_t *ref,
operand_t **base, operand_t **offset, pr_ushort_t *mode,
operand_t **target)
{
if (is_indirect (ref)) {
// ref is known to be either ex_expr or ex_uexpr, with '.' for
// the operator
if (ref->type == ex_expr) {
const expr_t *ent_expr = ref->expr.e1;
const expr_t *fld_expr = ref->expr.e2;
if (!is_entity (get_type (ent_expr))
|| !is_field (get_type (fld_expr))) {
print_expr (ref);
internal_error (ref, "expected entity.field");
}
sblock = statement_subexpr (sblock, ent_expr, base);
sblock = statement_subexpr (sblock, fld_expr, offset);
*mode = 1;//entity.field
} else if (ref->type == ex_uexpr) {
sblock = ptr_addressing_mode (sblock, ref->expr.e1, base, offset,
mode, target);
} else {
internal_error (ref, "unexpected expression type for indirect: %s",
expr_names[ref->type]);
}
} else {
sblock = statement_subexpr (sblock, ref, base);
*offset = short_operand (0, ref);
*mode = 0;
*target = 0;
}
return sblock;
}
static sblock_t *
statement_return (sblock_t *sblock, const expr_t *e)
{
const char *opcode;
statement_t *s;
debug (e, "RETURN");
opcode = "return";
if (!e->retrn.ret_val) {
if (options.code.progsversion == PROG_ID_VERSION) {
auto n = new_expr ();
*n = *e;
n->retrn.ret_val = new_float_expr (0);
e = n;
}
}
s = new_statement (st_func, opcode, e);
if (options.code.progsversion < PROG_VERSION) {
if (e->retrn.ret_val) {
const expr_t *ret_val = e->retrn.ret_val;
type_t *ret_type = get_type (ret_val);
// at_return is used for passing the result of a void_return
// function through void. v6 progs always use .return for the
// return value, so don't need to do anything special: just call
// the function and do a normal void return
if (!e->retrn.at_return) {
s->opa = return_operand (ret_type, e);
}
sblock = statement_subexpr (sblock, ret_val, &s->opa);
}
} else {
if (!e->retrn.at_return && e->retrn.ret_val) {
const expr_t *ret_val = e->retrn.ret_val;
type_t *ret_type = get_type (ret_val);
operand_t *target = 0;
pr_ushort_t ret_crtl = type_size (ret_type) - 1;
pr_ushort_t mode = 0;
sblock = addressing_mode (sblock, ret_val, &s->opa, &s->opb, &mode,
&target);
ret_crtl |= mode << 5;
s->opc = short_operand (ret_crtl, e);
statement_add_use (s, target);
} else {
if (e->retrn.at_return) {
const expr_t *call = e->retrn.ret_val;
if (!call || !is_function_call (call)) {
internal_error (e, "@return with no call");
}
// FIXME hard-coded reg, and assumes 3 is free
#define REG 3
const expr_t *with = new_with_expr (11, REG, new_short_expr (0));
def_t *ret_ptr = new_def (0, 0, 0, sc_local);
operand_t *ret_op = def_operand (ret_ptr, &type_void, e);
ret_ptr->reg = REG;
//expr_file_line (with, e);
sblock = statement_single (sblock, with);
sblock = statement_subexpr (sblock, call, &ret_op);
}
s->opa = short_operand (0, e);
s->opb = short_operand (0, e);
s->opc = short_operand (-1, e); // void return
}
}
sblock_add_statement (sblock, s);
sblock->next = new_sblock ();
sblock = sblock->next;
return sblock;
}
static sblock_t *
statement_adjstk (sblock_t *sblock, const expr_t *e)
{
statement_t *s = new_statement (st_func, "adjstk", e);
s->opa = short_operand (e->adjstk.mode, e);
s->opb = short_operand (e->adjstk.offset, e);
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
statement_with (sblock_t *sblock, const expr_t *e)
{
statement_t *s = new_statement (st_func, "with", e);
s->opa = short_operand (e->with.mode, e);
s->opc = short_operand (e->with.reg, e);
sblock = statement_subexpr (sblock, e->with.with, &s->opb);
sblock_add_statement (sblock, s);
return sblock;
}
static statement_t *
lea_statement (operand_t *pointer, operand_t *offset, const expr_t *e)
{
statement_t *s = new_statement (st_address, "lea", e);
s->opa = pointer;
s->opb = offset;
s->opc = temp_operand (&type_ptr, e);
return s;
}
static statement_t *
movep_statement (operand_t *dst, operand_t *src, const type_t *type,
const expr_t *e)
{
operand_t *dst_addr = operand_address (dst, e);
statement_t *s = new_statement (st_ptrmove, "movep", e);
s->opa = src;
//FIXME large types
s->opb = short_operand (type_size (type), e);
s->opc = dst_addr;
return s;
}
static statement_t *
load_statement (operand_t *ptr, operand_t *offs, operand_t *op, const expr_t *e)
{
statement_t *s = new_statement (st_expr, "load", e);
s->opa = ptr;
s->opb = offs;
s->opc = op;
return s;
}
static sblock_t *
expr_deref (sblock_t *sblock, const expr_t *deref, operand_t **op)
{
const type_t *load_type = deref->expr.type;
const expr_t *ptr_expr = deref->expr.e1;
operand_t *base = 0;
operand_t *offset = 0;
operand_t *target = 0;
pr_ushort_t mode;
statement_t *s;
sblock = addressing_mode (sblock, deref, &base, &offset, &mode, &target);
if (!*op) {
*op = temp_operand (load_type, deref);
}
switch (mode) {
case 0://direct def access
// FIXME should check that offset is 0, but currently,
// addressing_mode always sets offset to 0 for mode 0
s = assign_statement (*op, base, deref);
sblock_add_statement (sblock, s);
statement_add_use (s, target);
return sblock;
case 1://entity.field
case 2://const indexed pointer
case 3://var indexed pointer
break;
default:
internal_error (deref, "unexpected addressing mode: %d", mode);
}
if (low_level_type (load_type) == ev_void) {
s = lea_statement (base, offset, ptr_expr);
sblock_add_statement (sblock, s);
s = movep_statement (*op, s->opc, load_type, deref);
sblock_add_statement (sblock, s);
} else {
s = load_statement (base, offset, *op, deref);
sblock_add_statement (sblock, s);
}
statement_add_use (s, target);
return sblock;
}
static sblock_t *
expr_block (sblock_t *sblock, const expr_t *e, operand_t **op)
{
if (!e->block.result)
internal_error (e, "block sub-expression without result");
sblock = statement_slist (sblock, &e->block.list);
sblock = statement_subexpr (sblock, e->block.result, op);
return sblock;
}
static sblock_t *
expr_alias (sblock_t *sblock, const expr_t *e, operand_t **op)
{
operand_t *aop = 0;
type_t *type;
int offset = 0;
if (e->alias.offset) {
offset = expr_int (e->alias.offset);
}
type = e->alias.type;
sblock = statement_subexpr (sblock, e->alias.expr, &aop);
*op = offset_alias_operand (type, offset, aop, e);
return sblock;
}
static sblock_t *
expr_expr (sblock_t *sblock, const expr_t *e, operand_t **op)
{
const char *opcode;
statement_t *s;
opcode = convert_op (e->expr.op);
if (!opcode)
internal_error (e, "ice ice baby");
if (strcmp (opcode, "ne") == 0 && is_string (get_type (e->expr.e1))) {
opcode = "cmp";
}
if (strcmp (opcode, "dot") == 0) {
if (type_width (get_type (e->expr.e1)) == 2) {
opcode = "cdot";
}
if (type_width (get_type (e->expr.e1)) == 3) {
opcode = "vdot";
}
if (type_width (get_type (e->expr.e1)) == 4) {
opcode = "qdot";
}
}
s = new_statement (st_expr, opcode, e);
sblock = statement_subexpr (sblock, e->expr.e1, &s->opa);
sblock = statement_subexpr (sblock, e->expr.e2, &s->opb);
if (!*op)
*op = temp_operand (e->expr.type, e);
s->opc = *op;
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
expr_cast (sblock_t *sblock, const expr_t *e, operand_t **op)
{
const type_t *src_type;
const type_t *type = e->expr.type;
statement_t *s;
src_type = get_type (e->expr.e1);
if (is_scalar (src_type) && is_scalar (type)) {
operand_t *src = 0;
sblock = statement_subexpr (sblock, e->expr.e1, &src);
*op = temp_operand (e->expr.type, e);
s = new_statement (st_expr, "conv", e);
s->opa = src;
if (options.code.progsversion == PROG_VERSION) {
int from = type_cast_map[src_type->type];
int to = type_cast_map[type->type];
int width = type_width (src_type) - 1;
int conv = TYPE_CAST_CODE (from, to, width);
s->opb = short_operand (conv, e);
}
s->opc = *op;
sblock_add_statement (sblock, s);
} else {
sblock = expr_alias (sblock, e, op);
}
return sblock;
}
static sblock_t *
expr_negate (sblock_t *sblock, const expr_t *e, operand_t **op)
{
expr_t *neg;
expr_t *zero;
zero = new_nil_expr ();
zero->file = e->file;
zero->line = e->line;
zero = (expr_t *) convert_nil (zero, e->expr.type);
neg = new_binary_expr ('-', zero, e->expr.e1);
neg->expr.type = e->expr.type;
neg->file = e->file;
neg->line = e->line;
return statement_subexpr (sblock, neg, op);
}
static sblock_t *
expr_uexpr (sblock_t *sblock, const expr_t *e, operand_t **op)
{
const char *opcode;
statement_t *s;
switch (e->expr.op) {
case '.':
sblock = expr_deref (sblock, e, op);
break;
case 'C':
sblock = expr_cast (sblock, e, op);
break;
case '-':
// progs has no neg instruction!?!
sblock = expr_negate (sblock, e, op);
break;
default:
opcode = convert_op (e->expr.op);
if (!opcode)
internal_error (e, "ice ice baby");
s = new_statement (st_expr, opcode, e);
sblock = statement_subexpr (sblock, e->expr.e1, &s->opa);
if (!*op)
*op = temp_operand (e->expr.type, e);
s->opc = *op;
sblock_add_statement (sblock, s);
}
return sblock;
}
static sblock_t *
expr_horizontal (sblock_t *sblock, const expr_t *e, operand_t **op)
{
const char *opcode = "hops";
statement_t *s;
int hop;
type_t *res_type = e->hop.type;
type_t *vec_type = get_type (e->hop.vec);
switch (e->hop.op) {
case '&':
hop = 0;
break;
case '|':
hop = 1;
break;
case '^':
hop = 2;
break;
case '+':
if (is_integral (vec_type)) {
hop = 3;
} else {
hop = 7;
}
break;
case NAND:
hop = 4;
break;
case NOR:
hop = 5;
break;
case XNOR:
hop = 6;
break;
default:
internal_error (e, "invalid horizontal op");
}
hop |= (type_width (vec_type) - 1) << 3;
hop |= (pr_type_size[vec_type->type] - 1) << 5;
s = new_statement (st_expr, opcode, e);
sblock = statement_subexpr (sblock, e->hop.vec, &s->opa);
s->opb = short_operand (hop, e);
if (!*op) {
*op = temp_operand (res_type, e);
}
s->opc = *op;
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
expr_swizzle (sblock_t *sblock, const expr_t *e, operand_t **op)
{
const char *opcode = "swizzle";
statement_t *s;
int swiz = 0;
type_t *res_type = e->swizzle.type;
for (int i = 0; i < 4; i++) {
swiz |= (e->swizzle.source[i] & 3) << (2 * i);
}
swiz |= (e->swizzle.neg & 0xf) << 8;
swiz |= (e->swizzle.zero & 0xf) << 12;
s = new_statement (st_expr, opcode, e);
sblock = statement_subexpr (sblock, e->swizzle.src, &s->opa);
s->opb = short_operand (swiz, e);
if (!*op) {
*op = temp_operand (res_type, e);
}
s->opc = *op;
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
expr_extend (sblock_t *sblock, const expr_t *e, operand_t **op)
{
type_t *src_type = get_type (e->extend.src);
type_t *res_type = e->extend.type;
int src_width = type_width (src_type);
int res_width = type_width (res_type);
type_t *src_base = base_type (src_type);
type_t *res_base = base_type (res_type);
static int mode[4][4] = {
{-1, 0, 1, 2},
{-1,-1, 3, 4},
{-1,-1,-1, 5},
{-1,-1,-1,-1},
};
int ext = mode[src_width - 1][res_width - 1];
ext |= (e->extend.extend & 3) << 3;
ext |= (pr_type_size[res_base->type] - 1) << 5;
ext |= e->extend.reverse << 6;
if (ext < 0 || res_base != src_base) {
internal_error (e, "invalid type combination for extend %d %d %d",
ext, src_width, res_width);
}
statement_t *s = new_statement (st_expr, "extend", e);
sblock = statement_subexpr (sblock, e->extend.src, &s->opa);
s->opb = short_operand (ext, e);
if (!*op) {
*op = temp_operand (res_type, e);
}
s->opc = *op;
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
expr_def (sblock_t *sblock, const expr_t *e, operand_t **op)
{
*op = def_operand (e->def, e->def->type, e);
return sblock;
}
static sblock_t *
expr_symbol (sblock_t *sblock, const expr_t *e, operand_t **op)
{
symbol_t *sym = e->symbol;
if (sym->sy_type == sy_var) {
*op = def_operand (sym->s.def, sym->type, e);
} else if (sym->sy_type == sy_const) {
*op = value_operand (sym->s.value, e);
} else if (sym->sy_type == sy_func) {
if (!sym->s.func) {
make_function (sym, 0, pr.symtab->space, sc_extern);
}
*op = def_operand (sym->s.func->def, 0, e);
} else {
internal_error (e, "unexpected symbol type: %s for %s",
symtype_str (sym->sy_type), sym->name);
}
return sblock;
}
static sblock_t *
expr_temp (sblock_t *sblock, const expr_t *e, operand_t **op)
{
if (!e->temp.op)
((expr_t *) e)->temp.op = temp_operand (e->temp.type, e);
*op = e->temp.op;
return sblock;
}
static int
statement_copy_elements (sblock_t **sblock, const expr_t *dst, const expr_t *src, int base)
{
int index = 0;
for (auto li = src->vector.list.head; li; li = li->next) {
auto e = li->expr;
if (e->type == ex_vector) {
index += statement_copy_elements (sblock, dst, e, index + base);
} else {
int size = type_size (base_type (get_type (dst)));
type_t *src_type = get_type (e);
const expr_t *dst_ele = new_offset_alias_expr (src_type, dst,
size * (index + base));
index += type_width (src_type);
*sblock = statement_single (*sblock, assign_expr (dst_ele, e));
}
}
return index;
}
static sblock_t *
expr_vector_e (sblock_t *sblock, const expr_t *e, operand_t **op)
{
const expr_t *tmp;
type_t *vec_type = get_type (e);
int file = pr.source_file;
int line = pr.source_line;
pr.source_file = e->file;
pr.source_line = e->line;
tmp = new_temp_def_expr (vec_type);
statement_copy_elements (&sblock, tmp, e, 0);
pr.source_file = file;
pr.source_line = line;
sblock = statement_subexpr (sblock, tmp, op);
return sblock;
}
static sblock_t *
expr_nil (sblock_t *sblock, const expr_t *e, operand_t **op)
{
type_t *nil = e->nil;
const expr_t *size_expr;
size_t nil_size;
operand_t *zero;
operand_t *size;
statement_t *s;
if (!is_structural (nil)) {
*op = value_operand (new_nil_val (nil), e);
return sblock;
}
if (!*op) {
*op = temp_operand (nil, e);
}
nil_size = type_size (nil);
if (nil_size < 0x10000) {
size_expr = new_short_expr (nil_size);
} else {
size_expr = new_int_expr (nil_size);
}
sblock = statement_subexpr (sblock, new_int_expr(0), &zero);
sblock = statement_subexpr (sblock, size_expr, &size);
s = new_statement (st_memset, "memset", e);
s->opa = zero;
s->opb = size;
s->opc = *op;
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
expr_value (sblock_t *sblock, const expr_t *e, operand_t **op)
{
*op = value_operand (e->value, e);
return sblock;
}
static sblock_t *
expr_selector (sblock_t *sblock, const expr_t *e, operand_t **op)
{
return statement_subexpr (sblock, e->selector.sel_ref, op);
}
static sblock_t *
statement_subexpr (sblock_t *sblock, const expr_t *e, operand_t **op)
{
static expr_f sfuncs[ex_count] = {
[ex_block] = expr_block,
[ex_expr] = expr_expr,
[ex_uexpr] = expr_uexpr,
[ex_horizontal] = expr_horizontal,
[ex_swizzle] = expr_swizzle,
[ex_def] = expr_def,
[ex_symbol] = expr_symbol,
[ex_temp] = expr_temp,
[ex_vector] = expr_vector_e,
[ex_nil] = expr_nil,
[ex_value] = expr_value,
[ex_selector] = expr_selector,
[ex_alias] = expr_alias,
[ex_address] = expr_address,
[ex_assign] = expr_assign,
[ex_branch] = expr_branch,
[ex_extend] = expr_extend,
};
if (!e) {
*op = 0;
return sblock;
}
if (e->type >= ex_count)
internal_error (e, "bad sub-expression type: %d", e->type);
if (!sfuncs[e->type])
internal_error (e, "unexpected sub-expression type: %s",
expr_names[e->type]);
sblock = sfuncs[e->type] (sblock, e, op);
return sblock;
}
static sblock_t *
statement_ignore (sblock_t *sblock, const expr_t *e)
{
return sblock;
}
static sblock_t *
statement_state (sblock_t *sblock, const expr_t *e)
{
statement_t *s;
s = new_statement (st_state, "state", e);
sblock = statement_subexpr (sblock, e->state.frame, &s->opa);
sblock = statement_subexpr (sblock, e->state.think, &s->opb);
sblock = statement_subexpr (sblock, e->state.step, &s->opc);
sblock_add_statement (sblock, s);
return sblock;
}
static void
build_bool_block (expr_t *block, expr_t *e)
{
switch (e->type) {
case ex_bool:
build_bool_block (block, (expr_t *) e->boolean.e);
return;
case ex_label:
append_expr (block, e);
return;
case ex_assign:
append_expr (block, e);
return;
case ex_branch:
append_expr (block, e);
return;
case ex_expr:
if (e->expr.op == OR || e->expr.op == AND) {
build_bool_block (block, (expr_t *) e->expr.e1);
build_bool_block (block, (expr_t *) e->expr.e2);
} else {
append_expr (block, e);
}
return;
case ex_uexpr:
break;
case ex_block:
if (!e->block.result) {
for (auto t = e->block.head; t; t = t->next) {
build_bool_block (block, (expr_t *) t->expr);
}
return;
}
break;
default:
;
}
internal_error (e, "bad boolean");
}
static int
is_goto_expr (const expr_t *e)
{
return e && e->type == ex_branch && e->branch.type == pr_branch_jump
&& !e->branch.index;
}
static int
is_if_expr (const expr_t *e)
{
return e && e->type == ex_branch && e->branch.type != pr_branch_jump
&& e->branch.type != pr_branch_call;;
}
static int
is_label_expr (const expr_t *e)
{
return e && e->type == ex_label;
}
static sblock_t *
statement_bool (sblock_t *sblock, const expr_t *e)
{
const expr_t *l;
expr_t *block = new_block_expr (0);
build_bool_block (block, (expr_t *) e);
int num_expr = list_count (&block->block.list);
expr_t *exprs[num_expr + 1];
list_scatter (&block->block.list, (const expr_t **) exprs);
exprs[num_expr] = 0; // mark end of list
expr_t **d = exprs;
expr_t **s = exprs;
while (s[0]) {
if (is_if_expr (s[0]) && is_goto_expr (s[1])) {
l = s[0]->branch.target;
for (auto e = s + 2; is_label_expr (e[0]); e++) {
if (e[0] == l) {
((expr_t *) l)->label.used--;
// invert logic of if
s[0]->branch.type ^= pr_branch_ne;
// use goto's label in if
s[0]->branch.target = s[1]->branch.target;
l = 0;
break;
}
}
*d++ = *s++; // copy if
if (!l) {
s++; // skip over goto
}
} else if (is_goto_expr (*s)) {
l = s[0]->branch.target;
for (auto e = s + 1; is_label_expr (e[0]); e++) {
if (e[0] == l) {
((expr_t *) l)->label.used--;
l = 0;
break;
}
}
if (l) {
*d++ = *s++; // label survived, copy goto
} else {
s++; // skip over goto
}
} else {
*d++ = *s++;
}
}
block->block.list = (ex_list_t) {};
list_gather (&block->block.list, (const expr_t **) exprs, d - exprs);
sblock = statement_slist (sblock, &block->block.list);
return sblock;
}
static sblock_t *
statement_label (sblock_t *sblock, const expr_t *e)
{
if (sblock->statements) {
sblock->next = new_sblock ();
sblock = sblock->next;
}
if (e->label.used) {
((expr_t *) e)->label.dest = sblock;
((expr_t *) e)->label.next = sblock->labels;
sblock->labels = (ex_label_t *) &e->label;
} else {
if (e->label.symbol) {
warning (e, "unused label %s", e->label.symbol->name);
} else {
debug (e, "dropping unused label %s", e->label.name);
}
}
return sblock;
}
static sblock_t *
statement_block (sblock_t *sblock, const expr_t *e)
{
if (sblock->statements) {
sblock->next = new_sblock ();
sblock = sblock->next;
}
sblock = statement_slist (sblock, &e->block.list);
if (e->block.is_call) {
// for a fuction call, the call expresion is in only the result, not
// the actual block
sblock = statement_single (sblock, e->block.result);
}
return sblock;
}
static sblock_t *
statement_expr (sblock_t *sblock, const expr_t *e)
{
if (e->expr.op < 256)
debug (e, "e %c", e->expr.op);
else
debug (e, "e %d", e->expr.op);
if (options.warnings.executable)
warning (e, "Non-executable statement; executing programmer instead.");
return sblock;
}
static sblock_t *
statement_uexpr (sblock_t *sblock, const expr_t *e)
{
debug (e, "e ue %d", e->expr.op);
if (options.warnings.executable)
warning (e, "Non-executable statement; executing programmer instead.");
return sblock;
}
static sblock_t *
statement_memset (sblock_t *sblock, const expr_t *e)
{
const expr_t *dst = e->memset.dst;
const expr_t *val = e->memset.val;
const expr_t *count = e->memset.count;
const char *opcode = "memset";
statement_t *s;
if (is_constant (count)) {
if (is_int (get_type (count))
&& (unsigned) expr_int (count) < 0x10000) {
count = new_short_expr (expr_int (count));
}
if (is_uint (get_type (count)) && expr_int (count) < 0x10000) {
count = new_short_expr (expr_uint (count));
}
}
s = new_statement (st_memset, opcode, e);
sblock = statement_subexpr (sblock, dst, &s->opc);
sblock = statement_subexpr (sblock, count, &s->opb);
sblock = statement_subexpr (sblock, val, &s->opa);
sblock_add_statement (sblock, s);
return sblock;
}
static sblock_t *
statement_assign (sblock_t *sblock, const expr_t *e)
{
return expr_assign (sblock, e, 0);
}
static sblock_t *
statement_nonexec (sblock_t *sblock, const expr_t *e)
{
if (!e->rvalue && options.warnings.executable)
warning (e, "Non-executable statement; executing programmer instead.");
return sblock;
}
static sblock_t *
statement_single (sblock_t *sblock, const expr_t *e)
{
static statement_f sfuncs[ex_count] = {
[ex_error] = statement_ignore,
[ex_state] = statement_state,
[ex_bool] = statement_bool,
[ex_label] = statement_label,
[ex_block] = statement_block,
[ex_expr] = statement_expr,
[ex_uexpr] = statement_uexpr,
[ex_def] = statement_nonexec,
[ex_symbol] = statement_nonexec,
[ex_temp] = statement_nonexec,
[ex_vector] = statement_nonexec,
[ex_nil] = statement_nonexec,
[ex_value] = statement_nonexec,
[ex_memset] = statement_memset,
[ex_assign] = statement_assign,
[ex_branch] = statement_branch,
[ex_return] = statement_return,
[ex_adjstk] = statement_adjstk,
[ex_with] = statement_with,
};
if (e->type >= ex_count || !sfuncs[e->type]) {
internal_error (e, "bad expression type: %s",
e->type < ex_count ? expr_names[e->type] : "?");
}
sblock = sfuncs[e->type] (sblock, e);
return sblock;
}
sblock_t *
statement_slist (sblock_t *sblock, const ex_list_t *slist)
{
for (auto s = slist->head; s; s = s->next) {
sblock = statement_single (sblock, s->expr);
}
return sblock;
}
static void
move_labels (sblock_t *dst, sblock_t *src)
{
ex_label_t *src_labels = src->labels;
if (!src_labels)
return;
src_labels->dest = dst;
while (src_labels->next) {
src_labels = src_labels->next;
src_labels->dest = dst;
}
src_labels->next = dst->labels;
dst->labels = src->labels;
src->labels = 0;
}
static void
move_code (sblock_t *dst, sblock_t *src)
{
if (!src->statements)
return;
*dst->tail = src->statements;
dst->tail = src->tail;
src->statements = 0;
src->tail = &src->statements;
}
static int
merge_blocks (sblock_t **blocks)
{
sblock_t **sb;
sblock_t *sblock;
statement_t *s;
int did_something = 0;
if (!*blocks)
return did_something;
// merge any blocks that can be merged
for (sblock = *blocks; sblock; sblock = sblock->next) {
if (sblock->statements && sblock->next) {
s = (statement_t *) sblock->tail;
// func and flow statements end blocks
if (s->type >= st_func)
continue;
// labels begin blocks
if (sblock->next->labels)
continue;
// blocks can be merged
move_code (sblock, sblock->next);
did_something = 1;
}
}
for (sb = blocks; (*sb)->next;) {
if (!(*sb)->statements) {
// empty non-final block
// move labels from empty block to next block
if ((*sb)->labels)
move_labels ((*sb)->next, (*sb));
sblock = *sb;
*sb = (*sb)->next;
free_sblock (sblock);
continue;
}
sb = &(*sb)->next;
}
// so long as blocks doesn't become null, remove an empty final block
if (sb != blocks) {
if (!(*sb)->statements && !(*sb)->labels) {
// empty final block with no labels
sblock = *sb;
*sb = (*sb)->next;
free_sblock (sblock);
}
}
return did_something;
}
static void
remove_label_from_dest (ex_label_t *label)
{
sblock_t *sblock;
ex_label_t **l;
if (!label || !label->dest)
return;
debug (0, "dropping deceased label %s", label->name);
sblock = label->dest;
label->dest = 0;
for (l = &sblock->labels; *l; l = &(*l)->next) {
if (*l == label) {
*l = label->next;
label->next = 0;
break;
}
}
}
static void
unuse_label (ex_label_t *label)
{
if (label && !--label->used)
remove_label_from_dest (label);
}
static int
thread_jumps (sblock_t *blocks)
{
sblock_t *sblock;
int did_something = 0;
if (!blocks)
return 0;
for (sblock = blocks; sblock; sblock = sblock->next) {
statement_t *s;
ex_label_t **label, *l;
if (!sblock->statements)
continue;
s = (statement_t *) sblock->tail;
if (statement_is_goto (s)) {
label = &s->opa->label;
if (!(*label)->dest && (*label)->symbol) {
error (s->opa->expr, "undefined label `%s'",
(*label)->symbol->name);
(*label)->symbol = 0;
}
} else if (statement_is_cond (s)) {
label = &s->opa->label;
} else {
continue;
}
for (l = *label;
l->dest && l->dest->statements
&& statement_is_goto (l->dest->statements);
l = *statement_get_labelref (l->dest->statements)) {
// empty loop
}
if (l != *label) {
unuse_label (*label);
l->used++;
*label = l;
did_something = 1;
}
if ((statement_is_goto (s) || statement_is_cond (s))
&& (*label)->dest == sblock->next) {
statement_t **p;
unuse_label (*label);
for (p = &sblock->statements; *p != s; p = &(*p)->next)
;
free_statement (s);
*p = 0;
sblock->tail = p;
did_something = 1;
}
}
return did_something;
}
static int
remove_dead_blocks (sblock_t *blocks)
{
sblock_t *sblock;
int did_something;
int did_anything = 0;
int pass = 0;
if (!blocks)
return 0;
do {
debug (0, "dead block pass %d", pass++);
did_something = 0;
blocks->reachable = 1;
for (sblock = blocks; sblock->next; sblock = sblock->next) {
sblock_t *sb = sblock->next;
statement_t *s;
if (sb->labels) {
sb->reachable = 1;
continue;
}
if (!sblock->statements) {
sb->reachable = 1;
continue;
}
s = (statement_t *) sblock->tail;
if (statement_is_cond (s)
&& sb->statements && statement_is_goto (sb->statements)
&& statement_get_target (s) == sb->next) {
debug (0, "merging if/goto %p %p", sblock, sb);
ex_label_t **labelref = statement_get_labelref(s);
unuse_label (*labelref);
*labelref = *statement_get_labelref (sb->statements);
(*labelref)->used++;
invert_conditional (s);
sb->reachable = 0;
for (sb = sb->next; sb; sb = sb->next)
sb->reachable = 1;
break;
} else if (!statement_is_goto (s) && !statement_is_return (s)) {
sb->reachable = 1;
continue;
}
sb->reachable = 0;
}
for (sblock = blocks; sblock; sblock = sblock->next) {
while (sblock->next && !sblock->next->reachable) {
sblock_t *sb = sblock->next;
statement_t *s;
ex_label_t *label = 0;
debug (0, "removing dead block %p", sb);
if (sb->statements) {
s = (statement_t *) sb->tail;
ex_label_t **labelref = statement_get_labelref (s);
label = labelref ? *labelref : 0;
}
unuse_label (label);
did_something = 1;
did_anything = 1;
sblock->next = sb->next;
free_sblock (sb);
}
}
} while (did_something);
return did_anything;
}
static void
super_dealloc_warning (const expr_t *expr, pseudoop_t *op)
{
warning (expr,
"control may reach end of derived -dealloc without invoking %s",
op->name);
}
static void
search_for_super_dealloc (sblock_t *sblock)
{
operand_t *op;
pseudoop_t *super_dealloc = new_pseudoop ("[super dealloc]");
int super_dealloc_found = 0;
super_dealloc->next = current_func->pseudo_ops;
current_func->pseudo_ops = super_dealloc;
super_dealloc->uninitialized = super_dealloc_warning;
while (sblock) {
for (statement_t *st = sblock->statements; st; st = st->next) {
if (statement_is_return (st)) {
op = pseudo_operand (super_dealloc, st->expr);
statement_add_use (st, op);
continue;
}
if (!statement_is_call (st)) {
continue;
}
// effectively checks target
if (st->opa->op_type != op_def
|| strcmp (st->opa->def->name, "obj_msgSend_super") != 0) {
continue;
}
// function arguments are in reverse order, and the selector
// is the second argument (or second last in the list)
const expr_t *arg = 0;
auto arguments = st->expr->branch.args;
for (auto li = arguments->list.head; li; li = li->next) {
if (li->next && !li->next->next) {
arg = li->expr;
}
}
if (arg && is_selector (arg)) {
selector_t *sel = get_selector (arg);
if (sel && strcmp (sel->name, "dealloc") == 0) {
op = pseudo_operand (super_dealloc, st->expr);
statement_add_def (st, op);
super_dealloc_found++;
}
}
}
sblock = sblock->next;
}
// warn only when NOT optimizing because flow analysis will catch the
// missed invokation
if (!super_dealloc_found && !options.code.optimize) {
warning (0, "Derived class -dealloc does not call [super dealloc]");
}
}
static void
check_final_block (sblock_t *sblock)
{
statement_t *s = 0;
if (!sblock)
return;
while (sblock->next)
sblock = sblock->next;
if (sblock->statements) {
s = (statement_t *) sblock->tail;
if (statement_is_goto (s))
return; // the end of function is the end of a loop
if (statement_is_return (s))
return;
}
if (!is_void(current_func->sym->type->t.func.type))
warning (0, "control reaches end of non-void function");
if (s && s->type >= st_func) {
// func and flow end blocks, so we need to add a new block to take the
// return
sblock->next = new_sblock ();
sblock = sblock->next;
}
s = new_statement (st_func, "return", 0);
if (options.code.progsversion == PROG_VERSION) {
s->opa = short_operand (0, 0);
s->opb = short_operand (0, 0);
s->opc = short_operand (-1, 0);
} else {
if (options.traditional
|| options.code.progsversion == PROG_ID_VERSION) {
s->opa = return_operand (&type_void, 0);
}
}
sblock_add_statement (sblock, s);
}
void
dump_dot_sblock (const void *data, const char *fname)
{
print_sblock ((sblock_t *) data, fname);
}
sblock_t *
make_statements (const expr_t *e)
{
sblock_t *sblock = new_sblock ();
int did_something;
int pass = 0;
class_t *class;
if (options.block_dot.expr)
dump_dot ("expr", e, dump_dot_expr);
if (e->type != ex_block) {
statement_single (sblock, e);
} else {
statement_slist (sblock, &e->block.list);
}
if (options.block_dot.initial)
dump_dot ("initial", sblock, dump_dot_sblock);
do {
did_something = thread_jumps (sblock);
if (options.block_dot.thread)
dump_dot (va (0, "thread-%d", pass), sblock, dump_dot_sblock);
did_something |= remove_dead_blocks (sblock);
did_something |= merge_blocks (&sblock);
if (options.block_dot.dead)
dump_dot (va (0, "dead-%d", pass), sblock, dump_dot_sblock);
pass++;
} while (did_something);
check_final_block (sblock);
if (current_class && (class = extract_class (current_class))) {
// If the class is a root class, then it is not possible for there
// to be a call to [super dealloc] so do not check. However, any
// derived class implementeing -dealloc must call [super dealloc]
// (or some other deallocator (FIXME: later))
// FIXME better check for dealloc?
if (class->super_class && !strcmp (current_func->name, "-dealloc")) {
search_for_super_dealloc (sblock);
}
}
if (options.block_dot.final)
dump_dot ("final", sblock, dump_dot_sblock);
return sblock;
}
static void
count_temp (operand_t *op)
{
if (!op)
return;
if (op->op_type == op_temp) {
while (op->tempop.alias)
op = op->tempop.alias;
op->tempop.users++;
}
}
void
statements_count_temps (sblock_t *sblock)
{
statement_t *st;
while (sblock) {
for (st = sblock->statements; st; st = st->next) {
count_temp (st->opa);
count_temp (st->opb);
count_temp (st->opc);
}
sblock = sblock->next;
}
}