ref: d7740736da8f2ca830d558cdedd89266b9019375
dir: /6/isel.c/
#include <stdlib.h>
#include <stdio.h>
#include <inttypes.h>
#include <stdarg.h>
#include <inttypes.h>
#include <ctype.h>
#include <string.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include "util.h"
#include "parse.h"
#include "mi.h"
#include "asm.h"
#include "../config.h"
/* forward decls */
Loc *selexpr(Isel *s, Node *n);
#define Nfloatregargs 8
#define Nintregargs 6
regid floatargregs[] = {
Rxmm0d, Rxmm1d, Rxmm2d, Rxmm3d,
Rxmm4d, Rxmm5d, Rxmm6d, Rxmm7d,
};
regid intargregs[] = {Rrdi, Rrsi, Rrdx, Rrcx, Rr8, Rr9};
#define Nfloatregrets 2
#define Nintregrets 2
regid fltretregs[] = {Rxmm0d, Rxmm1d};
regid intretregs[] = {Rrax, Rrdx};
/* used to decide which operator is appropriate
* for implementing various conditional operators */
struct {
AsmOp test;
AsmOp jmp;
AsmOp getflag;
} reloptab[Numops] = {
[Olnot] = {Itest, Ijz, Isetz}, /* lnot invalid for floats */
/* signed int */
[Oeq] = {Icmp, Ijz, Isetz},
[One] = {Icmp, Ijnz, Isetnz},
[Ogt] = {Icmp, Ijg, Isetg},
[Oge] = {Icmp, Ijge, Isetge},
[Olt] = {Icmp, Ijl, Isetl},
[Ole] = {Icmp, Ijle, Isetle},
/* unsigned int */
[Oueq] = {Icmp, Ijz, Isetz},
[Oune] = {Icmp, Ijnz, Isetnz},
[Ougt] = {Icmp, Ija, Iseta},
[Ouge] = {Icmp, Ijae, Isetae},
[Oult] = {Icmp, Ijb, Isetb},
[Oule] = {Icmp, Ijbe, Isetbe},
/* float */
[Ofeq] = {Icomis, Ijz, Isetz},
[Ofne] = {Icomis, Ijnz, Isetnz},
[Ofgt] = {Icomis, Ija, Iseta},
[Ofge] = {Icomis, Ijae, Isetae},
[Oflt] = {Icomis, Ijb, Isetb},
[Ofle] = {Icomis, Ijbe, Isetbe},
};
static Mode
tymode(Type *t)
{
/* FIXME: What should the mode for, say, structs be when we have no
* intention of loading /through/ the pointer? For now, we'll just say it's
* the pointer mode, since we expect to address through the pointer */
t = tybase(t);
switch (t->type) {
case Tyflt32: return ModeF; break;
case Tyflt64: return ModeD; break;
default:
if (isstacktype(t))
return ModeQ;
switch (tysize(t)) {
case 1: return ModeB; break;
case 2: return ModeW; break;
case 4: return ModeL; break;
case 8: return ModeQ; break;
}
break;
}
return ModeNone;
}
static Mode tymodepart(Type *t, int is_float, size_t displacement)
{
assert(isstacktype(t));
size_t sz = tysize(t);
if (is_float) {
switch(sz - displacement) {
case 4: return ModeF; break;
default: return ModeD; break;
}
} else {
switch(sz - displacement) {
case 1: return ModeB; break;
case 2: return ModeW; break;
case 4: return ModeL; break;
default: return ModeQ; break;
}
}
}
static Mode
forcefltmode(Mode m)
{
assert(m != ModeNone);
switch (m) {
case ModeQ: return ModeD;
case ModeD: return ModeD;
default: return ModeF;
}
}
static Mode
forceintmode(Mode m)
{
assert(m != ModeNone);
switch (m) {
case ModeD: return ModeQ;
case ModeF: return ModeL;
default: return m;
}
}
static Mode
mode(Node *n)
{
if (n->type == Nexpr)
return tymode(exprtype(n));
else if (n->type == Ndecl)
return tymode(n->decl.type);
else
die("invalid node type");
return ModeNone;
}
static Loc *
varloc(Isel *s, Node *n)
{
ssize_t off;
Loc *l, *rip;
/* we need to try getting it from the stack first, in case we
* forced it to stack for addressing */
if (hthas(s->globls, n)) {
rip = locphysreg(Rrip);
l = locmeml(htget(s->globls, n), rip, NULL, mode(n));
} else if (hthas(s->envoff, n)) {
off = ptoi(htget(s->envoff, n));
l = locmem(off, s->envp, NULL, mode(n));
} else if (hthas(s->stkoff, n)) {
off = ptoi(htget(s->stkoff, n));
l = locmem(-off, locphysreg(Rrbp), NULL, mode(n));
} else if (stacknode(n)) {
assert(0);
} else {
l = htget(s->reglocs, n);
if (!l) {
l = locreg(mode(n));
htput(s->reglocs, n, l);
}
}
return l;
}
static Loc *
loc(Isel *s, Node *n)
{
Node *v;
Ucon *uc;
Loc *l;
if (n->type == Ndecl) {
l = varloc(s, n);
} else {
switch (exprop(n)) {
case Ovar:
l = varloc(s, n);
break;
case Oucon:
uc = finducon(exprtype(n), n->expr.args[0]);
l = loclit(uc->id, mode(n));
break;
case Olit:
v = n->expr.args[0];
switch (v->lit.littype) {
case Lvoid: l = NULL; break;
case Lchr: l = loclit(v->lit.chrval, mode(n)); break;
case Lbool: l = loclit(v->lit.boolval, mode(n)); break;
case Lint: l = loclit(v->lit.intval, mode(n)); break;
default:
die("Literal type %s should be blob", litstr[v->lit.littype]);
}
break;
default:
die("Node %s not leaf in loc()", opstr[exprop(n)]);
break;
}
}
return l;
}
static Insn *
mkinsnv(AsmOp op, va_list ap)
{
Loc *l;
Insn *i;
int n;
static size_t insnid;
n = 0;
i = malloc(sizeof(Insn));
i->op = op;
i->uid = insnid++;
while ((l = va_arg(ap, Loc*)) != NULL)
i->args[n++] = l;
i->nargs = n;
return i;
}
Insn *
mkinsn(int op, ...)
{
va_list ap;
Insn *i;
va_start(ap, op);
i = mkinsnv(op, ap);
va_end(ap);
return i;
}
static void
g(Isel *s, int op, ...)
{
va_list ap;
Insn *i;
va_start(ap, op);
i = mkinsnv(op, ap);
va_end(ap);
if (debugopt['i']) {
printf("GEN[uid=%zd] ", i->uid);
dbgiprintf(stdout, i);
}
lappend(&s->curbb->il, &s->curbb->ni, i);
}
static void
movz(Isel *s, Loc *src, Loc *dst)
{
if (src->mode == dst->mode)
g(s, Imov, src, dst, NULL);
else
g(s, Imovzx, src, dst, NULL);
}
static void
load(Isel *s, Loc *a, Loc *b)
{
Loc *l;
assert(b->type == Locreg);
if (a->type == Locreg)
l = locmem(0, b, NULL, a->mode);
else
l = a;
if (isfloatmode(b->mode))
g(s, Imovs, l, b, NULL);
else
g(s, Imov, l, b, NULL);
}
static void
stor(Isel *s, Loc *a, Loc *b)
{
Loc *l;
assert(a->type == Locreg || a->type == Loclit);
if (b->type == Locreg)
l = locmem(0, b, NULL, b->mode);
else
l = b;
if (isfloatmode(b->mode))
g(s, Imovs, a, l, NULL);
else
g(s, Imov, a, l, NULL);
}
/* ensures that a location is within a reg */
static Loc *
newr(Isel *s, Loc *a)
{
Loc *r;
r = locreg(a->mode);
if (a->type == Locreg) {
if (isfloatmode(a->mode))
g(s, Imovs, a, r, NULL);
else
g(s, Imov, a, r, NULL);
} else {
load(s, a, r);
}
return r;
}
static Loc *
inr(Isel *s, Loc *a)
{
if (a->type == Locreg)
return a;
return newr(s, a);
}
/* ensures that a location is within a reg or an imm */
static Loc *
inri(Isel *s, Loc *a)
{
if (a->type == Locreg || a->type == Loclit)
return a;
else
return newr(s, a);
}
/* If we're testing equality, etc, it's a bit silly
* to generate the test, store it to a bite, expand it
* to the right width, and then test it again. Try to optimize
* for these common cases.
*
* if we're doing the optimization to avoid
* multiple tests, we want to eval the children
* of the first arg, instead of the first arg
* directly */
static void
selcjmp(Isel *s, Node *n, Node **args)
{
Loc *a, *b;
Loc *l1, *l2;
AsmOp cond, jmp;
cond = reloptab[exprop(args[0])].test;
jmp = reloptab[exprop(args[0])].jmp;
/* if we have a cond, we're knocking off the redundant test,
* and want to eval the children */
if (cond) {
a = selexpr(s, args[0]->expr.args[0]);
if (args[0]->expr.nargs == 2)
b = selexpr(s, args[0]->expr.args[1]);
else
b = a;
a = newr(s, a);
} else {
cond = Itest;
jmp = Ijnz;
b = newr(s, selexpr(s, args[0])); /* cond */
a = b;
}
/* the jump targets will always be evaluated the same way */
l1 = loclbl(args[1]); /* if true */
l2 = loclbl(args[2]); /* if false */
g(s, cond, b, a, NULL);
g(s, jmp, l1, NULL);
g(s, Ijmp, l2, NULL);
}
/* Generate variable length jump. There are 3 case
* for this:
*
* 1) Short list: Simple comparison sequence.
* 2) Sparse list: binary search
* 3) Dense list: Jump table
*
* If a value is not in the jump table, then the 0th
* value is used.
*/
static void
selvjmp(Isel *s, Node *n, Node **args)
{
}
static Loc *
binop(Isel *s, AsmOp op, Node *x, Node *y)
{
Loc *a, *b;
a = selexpr(s, x);
b = selexpr(s, y);
a = newr(s, a);
g(s, op, b, a, NULL);
return a;
}
/* We have a few common cases to optimize here:
* Oaddr(expr)
* or:
* Oadd(
* reg,
* reg||const)
*
* or:
* Oadd(
* reg,
* Omul(reg,
* 2 || 4 || 8)))
*/
static int
ismergablemul(Node *n, int *r)
{
int v;
if (exprop(n) != Omul)
return 0;
if (exprop(n->expr.args[1]) != Olit)
return 0;
if (n->expr.args[1]->expr.args[0]->type != Nlit)
return 0;
if (n->expr.args[1]->expr.args[0]->lit.littype != Lint)
return 0;
v = n->expr.args[1]->expr.args[0]->lit.intval;
if (v != 2 && v != 4 && v != 8)
return 0;
*r = v;
return 1;
}
static Loc *
memloc(Isel *s, Node *e, Mode m)
{
Node **args;
Loc *l, *b, *o; /* location, base, offset */
int scale;
scale = 1;
l = NULL;
args = e->expr.args;
if (exprop(e) == Oadd) {
b = selexpr(s, args[0]);
if (ismergablemul(args[1], &scale))
o = selexpr(s, args[1]->expr.args[0]);
else
o = selexpr(s, args[1]);
if (b->type != Locreg)
b = inr(s, b);
if (o->type == Loclit) {
l = locmem(scale*o->lit, b, NULL, m);
} else {
b = inr(s, b);
o = inr(s, o);
l = locmems(0, b, o, scale, m);
}
} else {
l = selexpr(s, e);
l = inr(s, l);
l = locmem(0, l, NULL, m);
}
assert(l != NULL);
return l;
}
static const
Mode szmodes[] = {
[8] = ModeQ,
[4] = ModeL,
[2] = ModeW,
[1] = ModeB
};
static void
blit(Isel *s, Loc *to, Loc *from, size_t dstoff, size_t srcoff, size_t sz, size_t align)
{
size_t i, modesz;
Loc *sp, *dp; /* pointers to src, dst */
Loc *tmp, *src, *dst; /* source memory, dst memory */
assert(szmodes[align] != ModeNone); /* make sure we have a valid alignment */
if (!sz)
return;
sp = inr(s, from);
dp = inr(s, to);
i = 0;
if (align == 0)
align = 8;
for (modesz = align; szmodes[modesz] != ModeNone; modesz /= 2) {
tmp = locreg(szmodes[modesz]);
while (i + modesz <= sz) {
src = locmem(i + srcoff, sp, NULL, szmodes[modesz]);
dst = locmem(i + dstoff, dp, NULL, szmodes[modesz]);
g(s, Imov, src, tmp, NULL);
g(s, Imov, tmp, dst, NULL);
i += modesz;
}
}
}
static void
clear(Isel *s, Loc *val, size_t sz, size_t align)
{
Loc *dp, *zero, *dst; /* source memory, dst memory */
size_t modesz, i;
i = 0;
dp = inr(s, val);
if (align == 0)
align = 8;
for (modesz = align; szmodes[modesz] != ModeNone; modesz /= 2) {
zero = locreg(szmodes[modesz]);
g(s, Ixor, zero, zero, NULL);
while (i + modesz <= sz) {
dst = locmem(i, dp, NULL, szmodes[modesz]);
g(s, Imov, zero, dst, NULL);
i += modesz;
}
}
}
static void
call(Isel *s, Node *n)
{
AsmOp op;
Node *fn;
Loc *f, *e;
if (exprop(n) == Ocall) {
op = Icall;
fn = n->expr.args[0];
assert(tybase(exprtype(fn))->type == Tycode);
f = locmeml(htget(s->globls, fn), NULL, NULL, mode(fn));
} else {
op = Icallind;
f = selexpr(s, n->expr.args[0]);
e = selexpr(s, n->expr.args[1]);
g(s, Imov, e, locphysreg(Rrax), NULL);
}
g(s, op, f, NULL);
}
static void
placearg(Isel *s, Node *argn, Loc *argloc, PassIn p, Loc *rsp, int vararg, size_t *nfloats, size_t *nints, size_t *argoff)
{
/*
placearg may be called when argn is stored at argloc, but it may also
be called when argloc is a small piece of argn, as in the case when
small structs are being passed. In those circumstances, p is PassInSSE
or PassInInt, and argn is irrelevant. Therefore, argn should not be
relied on when p is PassInSSE or PassInInt.
*/
Loc *src, *dst;
size_t a;
if (p == PassInNoPref) {
if (stacknode(argn)) {
p = PassInMemory;
} else if (!vararg && isfloatmode(argloc->mode) && *nfloats < Nfloatregargs) {
p = PassInSSE;
} else if (!vararg && isintmode(argloc->mode) && *nints < Nintregargs) {
p = PassInInt;
} else {
p = PassInMemory;
}
}
switch (p) {
case PassInMemory:
if (stacknode(argn)) {
src = locreg(ModeQ);
g(s, Ilea, argloc, src, NULL);
a = tyalign(exprtype(argn));
blit(s, rsp, src, *argoff, 0, size(argn), a);
*argoff += size(argn);
} else {
dst = locmem(*argoff, rsp, NULL, argloc->mode);
argloc = inri(s, argloc);
stor(s, argloc, dst);
*argoff += size(argn);
}
break;
case PassInSSE:
dst = coreg(floatargregs[*nfloats], forcefltmode(argloc->mode));
argloc = inri(s, argloc);
if (isfloatmode(argloc->mode)) {
g(s, Imovs, argloc, dst, NULL);
} else {
g(s, Imov, argloc, dst, NULL);
}
(*nfloats)++;
break;
case PassInInt:
dst = coreg(intargregs[*nints], forceintmode(argloc->mode));
argloc = inri(s, argloc);
g(s, Imov, argloc, dst, NULL);
(*nints)++;
break;
case PassInNoPref: /* impossible */
die("cannot determine how to pass arg");
break;
}
}
static int
sufficientregs(ArgType a, size_t nfloats, size_t nints)
{
static const struct {
int ireg;
int freg;
} needed[] = {
[ArgAggrI] = {1, 0},
[ArgAggrFI] = {1, 1},
[ArgAggrIF] = {1, 1},
[ArgAggrII] = {2, 0},
[ArgAggrFF] = {0, 2},
};
return (needed[a].freg + nfloats <= Nfloatregargs) && (needed[a].ireg + nints <= Nintregargs);
}
static Loc *
plus8(Isel *s, Loc *base)
{
Loc *forcedreg = locreg(ModeQ);
if (base->type == Loclbl || (base->type == Locmeml && !base->mem.base)) {
forcedreg = loclitl(base->lbl);
} else {
g(s, Ilea, base, forcedreg, NULL);
}
return locmem(8, forcedreg, NULL, ModeQ);
}
static void
gencall(Isel *s, Node *n)
{
Loc *arg; /* values we reduced */
size_t argsz, argoff, nargs, falseargs, vasplit;
size_t nfloats, nints;
Loc *retloc1, *retloc2, *rsp; /* hard-coded registers */
Loc *ret;
size_t ri, rf;
Loc *stkbump; /* calculated stack offset */
Type *t, *fn;
Node **args;
Node *retnode;
ArgType rettype;
size_t i;
int vararg;
rsp = locphysreg(Rrsp);
t = exprtype(n);
ri = 0;
rf = 0;
retloc1 = NULL;
retloc2 = NULL;
rettype = classify(t);
switch (rettype) {
case ArgVoid:
case ArgBig:
break;
case ArgReg:
retloc1 = coreg((istyfloat(t)) ? Rxmm0d : Rrax, mode(n));
break;
case ArgAggrI:
retloc1 = coreg(intretregs[ri++], tymodepart(t, 0, 0));
break;
case ArgAggrF:
retloc1 = coreg(fltretregs[rf++], tymodepart(t, 1, 0));
break;
case ArgAggrII:
retloc1 = coreg(intretregs[ri++], tymodepart(t, 0, 0));
retloc2 = coreg(intretregs[ri++], tymodepart(t, 0, 8));
break;
case ArgAggrIF:
retloc1 = coreg(intretregs[ri++], tymodepart(t, 0, 0));
retloc2 = coreg(fltretregs[rf++], tymodepart(t, 1, 8));
break;
case ArgAggrFI:
retloc1 = coreg(fltretregs[rf++], tymodepart(t, 1, 0));
retloc2 = coreg(intretregs[ri++], tymodepart(t, 0, 8));
break;
case ArgAggrFF:
retloc1 = coreg(fltretregs[rf++], tymodepart(t, 1, 0));
retloc2 = coreg(fltretregs[rf++], tymodepart(t, 1, 8));
break;
}
fn = tybase(exprtype(n->expr.args[0]));
/* calculate the number of args we expect to see, adjust
* for a hidden return argument. */
vasplit = countargs(fn);
argsz = 0;
/*
* { the function itself, [optional environment], [optional return information], real arg 1, ... }
*/
falseargs = 1;
if (exprop(n) == Ocallind) {
falseargs++;
}
if (rettype != ArgVoid) {
retnode = n->expr.args[falseargs];
if (rettype != ArgBig) {
falseargs++;
}
}
args = &n->expr.args[falseargs];
nargs = n->expr.nargs - falseargs;
/* Have to calculate the amount to bump the stack
* pointer by in one pass first, otherwise if we push
* one at a time, we evaluate the args in reverse order.
* Not good.
*
* Skip the first operand, since it's the function itself
*
* Strictly speaking, we might waste a little space here,
* since some of these args might actually get passed in
* registers.
*/
for (i = 0; i < nargs; i++) {
argsz = align(argsz, min(size(args[i]), Ptrsz));
argsz += size(args[i]);
argsz = align(argsz, 8);
}
argsz = align(argsz, 16);
stkbump = loclit(argsz, ModeQ);
if (argsz)
g(s, Isub, stkbump, rsp, NULL);
/* Now, we can evaluate the arguments */
argoff = 0;
nfloats = 0;
nints = 0;
vararg = 0;
for (i = 0; i < nargs; i++) {
arg = selexpr(s, args[i]);
t = exprtype(args[i]);
argoff = alignto(argoff, t);
ArgType a = ArgBig;
if (i >= vasplit)
vararg = 1;
else
argoff = align(argoff, 8);
if (!vararg) {
a = classify(t);
}
if (!sufficientregs(a, nfloats, nints)) {
a = ArgBig;
}
switch(a) {
case ArgVoid:
break;
case ArgReg:
case ArgBig:
/* placearg can figure this out */
placearg(s, args[i], arg, PassInNoPref, rsp, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrI:
placearg(s, args[i], arg, PassInInt, rsp, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrF:
placearg(s, args[i], arg, PassInSSE, rsp, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrII:
placearg(s, args[i], arg , PassInInt, rsp, vararg, &nfloats, &nints, &argoff);
placearg(s, args[i], plus8(s, arg), PassInInt, rsp, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrIF:
placearg(s, args[i], arg , PassInInt, rsp, vararg, &nfloats, &nints, &argoff);
placearg(s, args[i], plus8(s, arg), PassInSSE, rsp, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrFI:
placearg(s, args[i], arg , PassInSSE, rsp, vararg, &nfloats, &nints, &argoff);
placearg(s, args[i], plus8(s, arg), PassInInt, rsp, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrFF:
placearg(s, args[i], arg , PassInSSE, rsp, vararg, &nfloats, &nints, &argoff);
placearg(s, args[i], plus8(s, arg), PassInSSE, rsp, vararg, &nfloats, &nints, &argoff);
break;
}
}
call(s, n);
if (argsz)
g(s, Iadd, stkbump, rsp, NULL);
switch (rettype) {
case ArgVoid:
case ArgBig:
/*
* No need to do anything. The return location, if any, was taken care of
* as the hidden argument.
*/
break;
case ArgReg:
/* retnode is the actual thing we're storing in */
ret = varloc(s, retnode);
if (isfloatmode(retloc1->mode))
g(s, Imovs, retloc1, ret, NULL);
else
g(s, Imov, retloc1, ret, NULL);
break;
case ArgAggrI:
g(s, Imov, retloc1, locmem(0, inri(s, selexpr(s, retnode)), NULL, ModeQ), NULL);
break;
case ArgAggrF:
g(s, Imovs, retloc1, locmem(0, inri(s, selexpr(s, retnode)), NULL, ModeD), NULL);
break;
case ArgAggrII:
g(s, Imov, retloc1, locmem(0, inri(s, selexpr(s, retnode)), NULL, ModeQ), NULL);
g(s, Imov, retloc2, locmem(8, inri(s, selexpr(s, retnode)), NULL, ModeQ), NULL);
break;
case ArgAggrIF:
g(s, Imov, retloc1, locmem(0, inri(s, selexpr(s, retnode)), NULL, ModeQ), NULL);
g(s, Imovs, retloc2, locmem(8, inri(s, selexpr(s, retnode)), NULL, ModeD), NULL);
break;
case ArgAggrFI:
g(s, Imovs, retloc1, locmem(0, inri(s, selexpr(s, retnode)), NULL, ModeD), NULL);
g(s, Imov, retloc2, locmem(8, inri(s, selexpr(s, retnode)), NULL, ModeQ), NULL);
break;
case ArgAggrFF:
g(s, Imovs, retloc1, locmem(0, inri(s, selexpr(s, retnode)), NULL, ModeD), NULL);
g(s, Imovs, retloc2, locmem(8, inri(s, selexpr(s, retnode)), NULL, ModeD), NULL);
break;
}
}
static Loc*
rolop(Isel *s, Node *n, Node *l, Node *r)
{
int64_t lv, rv;
Type *ty;
ty = tybase(exprtype(n));
if (!istyunsigned(ty))
return NULL;
if (exprop(l) != Obsl && exprop(l) != Obsr)
return NULL;
if (exprop(r) != Obsl && exprop(r) != Obsr)
return NULL;
if (exprop(r) == exprop(l))
return NULL;
if (exprop(l->expr.args[0]) != Ovar)
return NULL;
if (exprop(r->expr.args[0]) != Ovar)
return NULL;
if (l->expr.args[0]->expr.did != r->expr.args[0]->expr.did)
return NULL;
if (exprop(l->expr.args[1]) != Olit)
return NULL;
if (l->expr.args[1]->expr.args[0]->type != Nlit)
return NULL;
if (l->expr.args[1]->expr.args[0]->lit.littype != Lint)
return NULL;
lv = l->expr.args[1]->expr.args[0]->lit.intval;
if (exprop(r->expr.args[1]) != Olit)
return NULL;
if (r->expr.args[1]->expr.args[0]->type != Nlit)
return NULL;
if (r->expr.args[1]->expr.args[0]->lit.littype != Lint)
return NULL;
rv = r->expr.args[1]->expr.args[0]->lit.intval;
if (lv + rv != 8*size(n))
return NULL;
if (exprop(l) == Obsl)
return binop(s, Irol, l->expr.args[0], l->expr.args[1]);
else
return binop(s, Irol, r->expr.args[0], r->expr.args[1]);
}
Loc *
selexpr(Isel *s, Node *n)
{
Loc *a, *b, *c, *d, *r;
Loc *edx, *cl; /* x86 wants some hard-coded regs */
Node **args;
size_t al;
Op op;
args = n->expr.args;
edx = locphysreg(Redx);
cl = locphysreg(Rcl);
r = NULL;
switch (exprop(n)) {
case Oadd: r = binop(s, Iadd, args[0], args[1]); break;
case Osub: r = binop(s, Isub, args[0], args[1]); break;
case Oband: r = binop(s, Iand, args[0], args[1]); break;
case Obxor: r = binop(s, Ixor, args[0], args[1]); break;
case Obor:
r = rolop(s, n, args[0], args[1]);
if (!r)
r = binop(s, Ior, args[0], args[1]);
break;
case Omul:
if (size(args[0]) == 1) {
a = selexpr(s, args[0]);
b = inr(s, selexpr(s, args[1]));
c = locphysreg(Ral);
r = locreg(a->mode);
g(s, Imov, a, c, NULL);
g(s, Iimul_r, b, NULL);
g(s, Imov, c, r, NULL);
} else {
r = binop(s, Iimul, args[0], args[1]);
}
break;
case Odiv:
case Omod:
/* these get clobbered by the div insn */
a = selexpr(s, args[0]);
b = selexpr(s, args[1]);
b = newr(s, b);
c = coreg(Reax, mode(n));
r = locreg(a->mode);
g(s, Imov, a, c, NULL);
if (istysigned(exprtype(args[0]))) {
switch (r->mode) {
case ModeB: g(s, Imovsx, c, coreg(Rrax, ModeW), NULL); break;
case ModeW: g(s, Icwd, NULL); break;
case ModeL: g(s, Icdq, NULL); break;
case ModeQ: g(s, Icqo, NULL); break;
default: die("invalid mode in division"); break;
}
g(s, Iidiv, b, NULL);
} else {
if (r->mode == ModeB)
g(s, Ixor, locphysreg(Rah), locphysreg(Rah), NULL);
else
g(s, Ixor, edx, edx, NULL);
g(s, Idiv, b, NULL);
}
if (exprop(n) == Odiv)
d = coreg(Reax, mode(n));
else if (r->mode != ModeB)
d = coreg(Redx, mode(n));
else
d = locphysreg(Rah);
g(s, Imov, d, r, NULL);
break;
case Oneg:
r = selexpr(s, args[0]);
r = newr(s, r);
g(s, Ineg, r, NULL);
break;
/* fp expressions */
case Ofadd: r = binop(s, Iadds, args[0], args[1]); break;
case Ofsub: r = binop(s, Isubs, args[0], args[1]); break;
case Ofmul: r = binop(s, Imuls, args[0], args[1]); break;
case Ofdiv: r = binop(s, Idivs, args[0], args[1]); break;
case Ofneg:
r = selexpr(s, args[0]);
r = newr(s, r);
a = NULL;
b = NULL;
if (mode(args[0]) == ModeF) {
a = locreg(ModeF);
b = loclit(1LL << (31), ModeF);
g(s, Imovs, r, a, NULL);
} else if (mode(args[0]) == ModeD) {
a = locreg(ModeQ);
b = loclit(1LL << 63, ModeQ);
g(s, Imov, r, a, NULL);
}
g(s, Ixor, b, a, NULL);
g(s, Imov, a, r, NULL);
break;
case Obsl:
case Obsr:
a = newr(s, selexpr(s, args[0]));
b = selexpr(s, args[1]);
if (b->type == Loclit) {
d = b;
} else {
c = coreg(Rcl, b->mode);
g(s, Imov, b, c, NULL);
d = cl;
}
if (exprop(n) == Obsr) {
if (istysigned(n->expr.type))
g(s, Isar, d, a, NULL);
else
g(s, Ishr, d, a, NULL);
} else {
g(s, Ishl, d, a, NULL);
}
r = a;
break;
case Obnot:
r = selexpr(s, args[0]);
r = newr(s, r);
g(s, Inot, r, NULL);
break;
case Oderef:
r = memloc(s, args[0], mode(n));
break;
case Oaddr:
a = selexpr(s, args[0]);
if (a->type == Loclbl || (a->type == Locmeml && !a->mem.base)) {
r = loclitl(a->lbl);
} else {
r = locreg(ModeQ);
g(s, Ilea, a, r, NULL);
}
break;
case Olnot:
a = newr(s, selexpr(s, args[0]));
b = locreg(ModeB);
r = locreg(mode(n));
/* lnot only valid for integer-like values */
g(s, reloptab[exprop(n)].test, a, a, NULL);
g(s, reloptab[exprop(n)].getflag, b, NULL);
movz(s, b, r);
break;
case Oeq: case One: case Ogt: case Oge: case Olt: case Ole:
case Ofeq: case Ofne: case Ofgt: case Ofge: case Oflt: case Ofle:
case Oueq: case Oune: case Ougt: case Ouge: case Oult: case Oule:
a = selexpr(s, args[0]);
b = selexpr(s, args[1]);
a = newr(s, a);
c = locreg(ModeB);
r = locreg(mode(n));
g(s, reloptab[exprop(n)].test, b, a, NULL);
g(s, reloptab[exprop(n)].getflag, c, NULL);
movz(s, c, r);
return r;
case Oasn: /* relabel */
die("Unimplemented op %s", opstr[exprop(n)]);
break;
case Oset:
op = exprop(args[0]);
assert(op == Ovar || op == Oderef || op == Ogap);
assert(!stacknode(args[0]));
if (op == Ogap)
break;
b = selexpr(s, args[1]);
if (exprop(args[0]) == Oderef)
a = memloc(s, args[0]->expr.args[0], mode(n));
else
a = selexpr(s, args[0]);
b = inri(s, b);
if (isfloatmode(b->mode))
g(s, Imovs, b, a, NULL);
else
g(s, Imov, b, a, NULL);
r = b;
break;
case Ocall:
case Ocallind:
gencall(s, n);
break;
case Oret:
a = locstrlbl(s->cfg->end->lbls[0]);
g(s, Ijmp, a, NULL);
break;
case Ojmp:
g(s, Ijmp, loclbl(args[0]), NULL);
break;
case Ocjmp:
selcjmp(s, n, args);
break;
case Ovjmp:
selvjmp(s, n, args);
break;
case Oucon:
assert(isenum(tybase(exprtype(n))));
/* fallthrough */
case Olit:
r = loc(s, n);
break;
case Ovar:
if (isconstfn(n)) {
r = locreg(ModeQ);
a = loc(s, n);
g(s, Ilea, a, r, NULL);
} else {
r = loc(s, n);
}
break;
case Ogap:
break;
case Oblit:
a = selexpr(s, args[0]);
r = selexpr(s, args[1]);
al = alignto(1, args[0]->expr.type->sub[0]);
blit(s, a, r, 0, 0, args[2]->expr.args[0]->lit.intval, al);
break;
case Oclear:
a = selexpr(s, args[0]);
clear(s, a, args[1]->expr.args[0]->lit.intval, 0);
break;
/* cast operators that actually modify the values */
case Otrunc:
a = selexpr(s, args[0]);
a = inr(s, a);
r = locreg(mode(n));
g(s, Imov, a, r, NULL);
break;
case Ozwiden:
a = selexpr(s, args[0]);
a = inr(s, a);
r = locreg(mode(n));
movz(s, a, r);
break;
case Oswiden:
a = selexpr(s, args[0]);
a = inr(s, a);
r = locreg(mode(n));
g(s, Imovsx, a, r, NULL);
break;
case Oint2flt:
a = selexpr(s, args[0]);
a = inr(s ,a);
r = locreg(mode(n));
g(s, Icvttsi2sd, a, r, NULL);
break;
case Oflt2int:
a = selexpr(s, args[0]);
a = inr(s ,a);
r = locreg(mode(n));
g(s, Icvttsd2si, a, r, NULL);
break;
case Oflt2flt:
a = selexpr(s, args[0]);
r = locreg(mode(n));
if (a->mode == ModeD)
g(s, Icvttsd2ss, a, r, NULL);
else
g(s, Icvttss2sd, a, r, NULL);
break;
case Odead:
case Oundef:
case Odef:
/* nothing */
break;
/* These operators should never show up in the reduced trees,
* since they should have been replaced with more primitive
* expressions by now */
case Obad: case Oauto: case Opreinc: case Opostinc: case Opredec:
case Otern: case Opostdec: case Olor: case Oland: case Oaddeq:
case Osubeq: case Omuleq: case Odiveq: case Omodeq: case Oboreq:
case Obandeq: case Obxoreq: case Obsleq: case Obsreq: case Omemb:
case Oslbase: case Osllen: case Ocast: case Outag: case Oudata:
case Otup: case Oarr: case Ostruct:
case Oslice: case Oidx: case Osize: case Otupget: case Otupmemb:
case Obreak: case Ocontinue:
case Numops:
dumpn(n, stdout);
die("Should not see %s in isel", opstr[exprop(n)]);
break;
}
return r;
}
static void
isel(Isel *s, Node *n)
{
switch (n->type) {
case Nexpr:
selexpr(s, n);
break;
case Ndecl:
break;
default:
die("Bad node type in isel()");
break;
}
}
/* %rax is for int returns, %xmm0d is for floating returns */
Reg
savedregs[] = {
Rr12, Rr13, Rr14, Rr15,
Rnone
};
static void
movearg(Isel *s, Loc *dst, PassIn p, Mode m, size_t *nfloats, size_t *nints, size_t *argoff)
{
Loc *a;
assert(m != ModeNone);
switch(p) {
case PassInInt:
a = coreg(intargregs[*nints], forceintmode(m));
g(s, Imov, a, dst, NULL);
(*nints)++;
break;
case PassInSSE:
a = coreg(floatargregs[*nfloats], forcefltmode(m));
g(s, Imovs, a, dst, NULL);
(*nfloats)++;
break;
default: /* no need to move if on stack */
break;
}
}
static void
retrievearg(Isel *s, Node *argn, int vararg, size_t *nfloats, size_t *nints, size_t *argoff)
{
Loc *l;
if (stacknode(argn)) {
htput(s->stkoff, argn, itop(-(*argoff + 2*Ptrsz)));
*argoff += size(argn);
} else if (!vararg && isfloatmode(mode(argn)) && *nfloats < Nfloatregargs) {
l = loc(s, argn);
movearg(s, l, PassInSSE, forcefltmode(mode(argn)), nfloats, nints, argoff);
htput(s->reglocs, argn, l);
} else if (!vararg && isintmode(mode(argn)) && *nints < Nintregargs) {
l = loc(s, argn);
movearg(s, l, PassInInt, forceintmode(mode(argn)), nfloats, nints, argoff);
htput(s->reglocs, argn, l);
} else if (tybase(decltype(argn))->type != Tyvoid) {
/* varargs go on the stack */
htput(s->stkoff, argn, itop(-(*argoff + 2*Ptrsz)));
*argoff += size(argn);
}
}
void
addarglocs(Isel *s, Func *fn)
{
size_t i, nints, nfloats, nargs;
size_t argoff;
int vararg;
Node *arg;
Type *t;
argoff = 0;
nfloats = 0;
nints = 0;
vararg = 0;
nargs = countargs(fn->type);
for (i = 0; i < fn->nargs; i++) {
arg = fn->args[i];
t = decltype(arg);
argoff = alignto(argoff, t);
ArgType a = ArgBig;
Loc *l = NULL;
if (i >= nargs)
vararg = 1;
else
argoff = align(argoff, 8);
if (!vararg) {
a = classify(t);
}
if (!sufficientregs(a, nfloats, nints)) {
a = ArgBig;
}
switch(a) {
case ArgVoid:
break;
case ArgReg:
case ArgBig:
/* retrievearg can figure this out */
retrievearg(s, arg, vararg, &nfloats, &nints, &argoff);
break;
case ArgAggrI:
l = loc(s, arg);
movearg(s, l, PassInInt, ModeQ, &nfloats, &nints, &argoff);
break;
case ArgAggrF:
l = loc(s, arg);
movearg(s, l, PassInSSE, ModeD, &nfloats, &nints, &argoff);
break;
case ArgAggrII:
l = loc(s, arg);
movearg(s, l , PassInInt, ModeQ, &nfloats, &nints, &argoff);
movearg(s, plus8(s, l), PassInInt, ModeQ, &nfloats, &nints, &argoff);
break;
case ArgAggrIF:
l = loc(s, arg);
movearg(s, l , PassInInt, ModeQ, &nfloats, &nints, &argoff);
movearg(s, plus8(s, l), PassInSSE, ModeD, &nfloats, &nints, &argoff);
break;
case ArgAggrFI:
l = loc(s, arg);
movearg(s, l , PassInSSE, ModeD, &nfloats, &nints, &argoff);
movearg(s, plus8(s, l), PassInInt, ModeQ, &nfloats, &nints, &argoff);
break;
case ArgAggrFF:
l = loc(s, arg);
movearg(s, l , PassInSSE, ModeD, &nfloats, &nints, &argoff);
movearg(s, plus8(s, l), PassInSSE, ModeD, &nfloats, &nints, &argoff);
break;
}
}
}
static void
prologue(Isel *s, Func *fn, size_t sz)
{
Loc *rsp;
Loc *rbp;
Loc *stksz;
Loc *phys;
size_t i;
rsp = locphysreg(Rrsp);
rbp = locphysreg(Rrbp);
stksz = loclit(sz, ModeQ);
/* enter function */
g(s, Ipush, rbp, NULL);
g(s, Imov, rsp, rbp, NULL);
g(s, Isub, stksz, rsp, NULL);
/* save registers */
for (i = 0; savedregs[i] != Rnone; i++) {
phys = locphysreg(savedregs[i]);
s->calleesave[i] = locreg(phys->mode);
if (isfloatmode(phys->mode)) {
g(s, Imovs, phys, s->calleesave[i], NULL);
} else {
g(s, Imov, phys, s->calleesave[i], NULL);
}
}
if (s->envp)
g(s, Imov, locphysreg(Rrax), s->envp, NULL);
addarglocs(s, fn);
s->nsaved = i;
s->stksz = stksz; /* need to update if we spill */
}
static void
epilogue(Isel *s)
{
Loc *rsp, *rbp;
Loc *ret;
size_t i;
size_t ri = 0, rf = 0;
rsp = locphysreg(Rrsp);
rbp = locphysreg(Rrbp);
switch (s->rettype) {
case ArgVoid:
break;
case ArgReg:
/* s->ret is a value, and will be returned that way */
ret = loc(s, s->ret);
if (istyfloat(exprtype(s->ret)))
g(s, Imovs, ret, coreg(Rxmm0d, ret->mode), NULL);
else
g(s, Imov, ret, coreg(Rax, ret->mode), NULL);
break;
case ArgBig:
/* s->ret is an address, and will be returned that way */
ret = loc(s, s->ret);
g(s, Imov, ret, coreg(Rax, ret->mode), NULL);
break;
case ArgAggrI:
/* s->ret is an address, and will be returned as values */
ret = loc(s, s->ret);
load(s, locmem(0, ret, NULL, ModeQ), coreg(intretregs[ri++], ModeQ));
break;
case ArgAggrF:
ret = loc(s, s->ret);
load(s, locmem(0, ret, NULL, ModeD), coreg(fltretregs[rf++], ModeD));
break;
case ArgAggrII:
ret = loc(s, s->ret);
load(s, locmem(0, ret, NULL, ModeQ), coreg(intretregs[ri++], ModeQ));
load(s, locmem(8, ret, NULL, ModeQ), coreg(intretregs[ri++], ModeQ));
break;
case ArgAggrIF:
ret = loc(s, s->ret);
load(s, locmem(0, ret, NULL, ModeQ), coreg(intretregs[ri++], ModeQ));
load(s, locmem(8, ret, NULL, ModeD), coreg(fltretregs[rf++], ModeD));
break;
case ArgAggrFI:
ret = loc(s, s->ret);
load(s, locmem(0, ret, NULL, ModeD), coreg(fltretregs[rf++], ModeD));
load(s, locmem(8, ret, NULL, ModeQ), coreg(intretregs[ri++], ModeQ));
break;
case ArgAggrFF:
ret = loc(s, s->ret);
load(s, locmem(0, ret, NULL, ModeD), coreg(fltretregs[rf++], ModeD));
load(s, locmem(8, ret, NULL, ModeD), coreg(fltretregs[rf++], ModeD));
break;
}
/* restore registers */
for (i = 0; savedregs[i] != Rnone; i++)
if (isfloatmode(s->calleesave[i]->mode))
g(s, Imovs, s->calleesave[i], locphysreg(savedregs[i]), NULL);
else
g(s, Imov, s->calleesave[i], locphysreg(savedregs[i]), NULL);
/* leave function */
g(s, Imov, rbp, rsp, NULL);
g(s, Ipop, rbp, NULL);
g(s, Iret, NULL);
}
static Asmbb *
mkasmbb(Bb *bb)
{
Asmbb *as;
if (!bb)
return NULL;
as = zalloc(sizeof(Asmbb));
as->id = bb->id;
as->pred = bsdup(bb->pred);
as->succ = bsdup(bb->succ);
as->lbls = memdup(bb->lbls, bb->nlbls*sizeof(char*));
as->nlbls = bb->nlbls;
return as;
}
static void
handlesmallstructargs(Isel *is, Func *fn)
{
/*
* Perform a last-minute adjustment to fn->stksz to handle small structs
* that will be passed in registers. We do this inside selfunc so that
* generics will be specialized.
*/
size_t vasplit = countargs(fn->type);
size_t i = 0;
Type *t;
Node *arg;
for (i = 0; i < fn->nargs; i++) {
arg = fn->args[i];
t = decltype(arg);
int vararg = 0;
ArgType a = ArgBig;
if (i >= vasplit)
vararg = 1;
if (!vararg) {
a = classify(t);
}
switch(a) {
case ArgVoid:
case ArgReg:
case ArgBig:
/* No need for any extra space for this arg */
break;
case ArgAggrI:
case ArgAggrF:
fn->stksz += 8;
fn->stksz = align(fn->stksz, min(8, Ptrsz));
htput(fn->stkoff, fn->args[i], itop(fn->stksz));
break;
case ArgAggrII:
case ArgAggrIF:
case ArgAggrFI:
case ArgAggrFF:
fn->stksz += 16;
fn->stksz = align(fn->stksz, min(16, Ptrsz));
htput(fn->stkoff, fn->args[i], itop(fn->stksz));
break;
}
}
}
void
selfunc(Isel *is, Func *fn, Htab *globls, Htab *strtab)
{
int fileid, lastline;
Node *n;
Bb *bb;
size_t i, j;
char buf[128];
char *path;
for (i = 0; i < fn->cfg->nbb; i++)
lappend(&is->bb, &is->nbb, mkasmbb(fn->cfg->bb[i]));
is->curbb = is->bb[0];
fileid = fn->loc.file;
if (fileid >= 0 && fn->loc.line > 0) {
path = file.files[fileid];
bprintf(buf, sizeof buf, "%s/%s:%d", is->cwd, path, fn->loc.line);
g(is, Icomment, locstrlbl(buf), NULL);
bprintf(buf, sizeof buf, "%zd %d", fileid + 1, fn->loc.line);
g(is, Iloc, locstrlbl(buf), NULL);
}
handlesmallstructargs(is, fn);
prologue(is, fn, fn->stksz);
lastline = -1;
for (j = 0; j < fn->cfg->nbb - 1; j++) {
is->curbb = is->bb[j];
if (!is->bb[j])
continue;
bb = fn->cfg->bb[j];
for (i = 0; i < bb->nnl; i++) {
/* put in a comment that says where this line comes from */
n = bb->nl[i];
fileid = n->loc.file;
if (n->loc.file >= 0 && n->loc.line != -1 && n->loc.line != lastline) {
lastline = n->loc.line;
path = file.files[fileid];
bprintf(buf, sizeof buf, "%s/%s:%d", is->cwd, path, n->loc.line);
g(is, Icomment, locstrlbl(buf), NULL);
bprintf(buf, sizeof buf, "%zd %d", fileid + 1, n->loc.line);
g(is, Iloc, locstrlbl(buf), NULL);
}
isel(is, fn->cfg->bb[j]->nl[i]);
}
}
is->curbb = is->bb[is->nbb - 1];
epilogue(is);
peep(is);
regalloc(is);
is->stksz->lit = align(is->stksz->lit, 16);
}