ref: 732d6bdc22ff2b7647093953d89be464828db998
dir: /src/cmd/cc/cc1/code.c/
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <scc/scc.h>
#include "cc1.h"
static void emitbin(int, void *),
emitcast(int, void *),
emitsym(int, void *),
emitexp(int, void *),
emitsymid(int, void *),
emittext(int, void *),
emitfun(int, void *),
emitdcl(int, void *),
emitinit(int, void *),
emittype(int, void *),
emitbuilt(int, void *);
char *optxt[] = {
[OADD] = "+",
[OSUB] = "-",
[OMUL] = "*",
[OINC] = ":i",
[ODEC] = ":d",
[OPTR] = "@",
[OMOD] = "%",
[ODIV] = "/",
[OSHL] = "l",
[OSHR] = "r",
[OLT] = "<",
[OGT] = ">",
[OGE] = "]",
[OLE] = "[",
[OEQ] = "=",
[ONE] = "!",
[OBAND] = "&",
[OBXOR] = "^",
[OBOR] = "|",
[OASSIGN] = ":",
[OA_MUL] = ":*",
[OA_DIV] = ":/",
[OA_MOD] = ":%",
[OA_ADD] = ":+",
[OA_SUB] = ":-",
[OA_SHL] = ":l",
[OA_SHR] = ":r",
[OA_AND] = ":&",
[OA_XOR] = ":^",
[OA_OR] = ":|",
[OADDR] = "'",
[OSNEG] = "_",
[ONEG] = "n",
[OCPL] = "~",
[OAND] = "a",
[OOR] = "o",
[OASK] = "?",
[OCOMMA] = ",",
[OLABEL] = "L%d\n",
[ODEFAULT] = "\tf\tL%d\n",
[OBSWITCH] = "\ts",
[OESWITCH] = "\tt\tL%d\n",
[OCASE] = "\tv\tL%d",
[OJUMP] = "\tj\tL%d\n",
[OBRANCH] = "\ty\tL%d",
[OEFUN] = "}\n",
[OELOOP] = "\tb\n",
[OBLOOP] = "\te\n",
[ORET] = "\th",
[OPAR] = "p",
[OCALL] = "c",
[OCALLE] = "z",
[OFIELD] = "."
};
void (*opcode[])(int, void *) = {
[OADD] = emitbin,
[OSUB] = emitbin,
[OMUL] = emitbin,
[OINC] = emitbin,
[ODEC] = emitbin,
[OPTR] = emitbin,
[OMOD] = emitbin,
[ODIV] = emitbin,
[OSHL] = emitbin,
[OSHR] = emitbin,
[OLT] = emitbin,
[OGT] = emitbin,
[OGE] = emitbin,
[OLE] = emitbin,
[OEQ] = emitbin,
[ONE] = emitbin,
[OBAND] = emitbin,
[OBXOR] = emitbin,
[OBOR] = emitbin,
[OASSIGN] = emitbin,
[OA_MUL] = emitbin,
[OA_DIV] = emitbin,
[OA_MOD] = emitbin,
[OA_ADD] = emitbin,
[OA_SUB] = emitbin,
[OA_SHL] = emitbin,
[OA_SHR] = emitbin,
[OA_AND] = emitbin,
[OA_XOR] = emitbin,
[OA_OR] = emitbin,
[OADDR] = emitbin,
[OSNEG] = emitbin,
[ONEG] = emitbin,
[OCPL] = emitbin,
[OAND] = emitbin,
[OOR] = emitbin,
[OCOMMA] = emitbin,
[OCAST] = emitcast,
[OSYM] = emitsym,
[OASK] = emitbin,
[OCOLON] = emitbin,
[OFIELD]= emitbin,
[OEXPR] = emitexp,
[OLABEL] = emitsymid,
[ODEFAULT] = emitsymid,
[OCASE] = emitsymid,
[OJUMP] = emitsymid,
[OBRANCH] = emitsymid,
[OEFUN] = emittext,
[OELOOP] = emittext,
[OBLOOP] = emittext,
[OFUN] = emitfun,
[ORET] = emittext,
[ODECL] = emitdcl,
[OBSWITCH] = emittext,
[OESWITCH] = emitsymid,
[OPAR] = emitbin,
[OCALL] = emitbin,
[OCALLE] = emitbin,
[OINIT] = emitinit,
[OBUILTIN] = emitbuilt,
[OTYP] = emittype,
};
static FILE *outfp;
void
icode(void)
{
outfp = stdout;
}
void
freetree(Node *np)
{
if (!np)
return;
freetree(np->left);
freetree(np->right);
free(np);
}
static void
emitnode(Node *np)
{
if (np)
(*opcode[np->op])(np->op, np);
}
Node *
prtree(char *s, Node *np)
{
FILE *tmp = outfp;
outfp = stderr;
fprintf(outfp, "DBG prtree %s", s);
emitnode(np);
putc('\n', outfp);
outfp = tmp;
return np;
}
void
emit(int op, void *arg)
{
extern int failure;
if (failure || onlycpp || onlyheader)
return;
(*opcode[op])(op, arg);
}
static void
emitvar(Symbol *sym)
{
int c;
short flags = sym->flags;
if (flags & SLOCAL)
c = 'T';
else if (flags & SPRIVATE)
c = 'Y';
else if (flags & SGLOBAL)
c = 'G';
else if (flags & SREGISTER)
c = 'R';
else if (flags & SFIELD)
c = 'M';
else if (flags & SEXTERN)
c = 'X';
else
c = 'A';
fprintf(outfp, "%c%u", c, sym->id);
}
static void
emitconst(Node *np)
{
Symbol *sym = np->sym;
Type *tp = np->type;
TUINT u;
switch (tp->op) {
case PTR:
case INT:
case ENUM:
u = (tp->prop & TSIGNED) ? (TUINT) sym->u.i : sym->u.u;
fprintf(outfp,
"#%c%llX",
np->type->letter,
(long long) u & ones(tp->size));
break;
default:
abort();
}
}
static void
emitsym(int op, void *arg)
{
Node *np = arg;
if ((np->sym->flags & SINITLST) == 0) {
/*
* When we have a compound literal we are going
* to call to emitnode for every element of it,
* and it means that we will have two '\t'
* for the first element
*/
putc('\t', outfp);
}
(np->flags & NCONST) ? emitconst(np) : emitvar(np->sym);
}
static void
emitletter(Type *tp)
{
int letter;
letter = (tp->prop&TELLIPSIS) ? 'E' : tp->letter;
putc(letter, outfp);
switch (tp->op) {
case ARY:
case STRUCT:
case UNION:
fprintf(outfp, "%u", tp->id);
}
}
static void
emittype(int op, void *arg)
{
TINT n;
Symbol **sp;
char *tag;
Type *tp = arg;
if (!(tp->prop & TDEFINED))
return;
switch (tp->op) {
case ARY:
emitletter(tp);
putc('\t', outfp);
emitletter(tp->type);
fprintf(outfp,
"\t#%c%llX\n",
sizettype->letter, (long long) tp->n.elem);
return;
case UNION:
case STRUCT:
emitletter(tp);
tag = tp->tag->name;
fprintf(outfp,
"\t\"%s\t#%c%lX\t#%c%X\n",
(tag) ? tag : "",
sizettype->letter,
tp->size,
sizettype->letter,
tp->align);
n = tp->n.elem;
for (sp = tp->p.fields; n-- > 0; ++sp)
emit(ODECL, *sp);
break;
case PTR:
case FTN:
case ENUM:
return;
default:
abort();
}
}
static void
emitstring(Symbol *sym, Type *tp)
{
char *bp, *s, *lim;
int n;
bp = sym->u.s;
lim = &sym->u.s[tp->n.elem];
while (bp < lim) {
s = bp;
while (bp < lim && isprint(*bp))
++bp;
if ((n = bp - s) > 1)
fprintf(outfp, "\t#\"%.*s\n", n, s);
else
bp = s;
if (bp == lim)
break;
do {
fprintf(outfp,
"\t#%c%02X\n",
chartype->letter, (*bp++) & 0xFF);
} while (bp < lim && !isprint(*bp));
}
}
static void
emitdesig(Node *np, Type *tp)
{
Symbol *sym;
size_t n; /* TODO: This should be SIZET */
Node *aux;
Type *p;
if (!np) {
sym = NULL;
} else {
if (!np->sym)
goto emit_expression;
sym = np->sym;
if (sym->flags & SSTRING) {
emitstring(sym, tp);
return;
}
if ((sym->flags & SINITLST) == 0)
goto emit_expression;
}
switch (tp->op) {
case PTR:
case INT:
case ENUM:
if (sym)
aux = *sym->u.init;
else
aux = simplify(convert(constnode(zero), tp, 0));
emitexp(OEXPR, aux);
break;
case UNION:
n = tp->n.elem-1;
aux = (sym) ? sym->u.init[0] : NULL;
emitdesig(aux, aux->type);
break;
case STRUCT:
case ARY:
for (n = 0; n < tp->n.elem; ++n) {
aux = (sym) ? sym->u.init[n] : NULL;
p = (tp->op == ARY) ? tp->type : tp->p.fields[n]->type;
emitdesig(aux, p);
}
break;
default:
abort();
}
if (sym) {
free(sym->u.init);
sym->u.init = NULL;
}
freetree(np);
return;
emit_expression:
emitexp(OEXPR, np);
}
static void
emitinit(int op, void *arg)
{
Node *np = arg;
fputs("\t(\n", outfp);
emitdesig(np, np->type);
fputs(")\n", outfp);
}
static void
emitdcl(int op, void *arg)
{
Symbol *sym = arg;
if (sym->flags & SEMITTED)
return;
emitvar(sym);
putc('\t', outfp);
if (sym->type->op == FTN) {
emitletter(sym->type->type);
putc('\t', outfp);
}
emitletter(sym->type);
fprintf(outfp, "\t\"%s", (sym->name) ? sym->name : "");
if (sym->flags & SFIELD)
fprintf(outfp, "\t#%c%llX", sizettype->letter, sym->u.i);
sym->flags |= SEMITTED;
if ((sym->flags & SHASINIT) == 0)
putc('\n', outfp);
}
static void
emitcast(int op, void *arg)
{
Node *np = arg, *lp = np->left;
emitnode(lp);
if (np->type != voidtype)
fprintf(outfp, "\tg%c", np->type->letter);
}
static void
emitbin(int op, void *arg)
{
Node *np = arg;
char *s;
emitnode(np->left);
emitnode(np->right);
/* do not print in OCOLON case */
if ((s = optxt[op]) != NULL) {
fprintf(outfp, "\t%s", s);
emitletter(np->type);
}
}
static void
emitbuilt(int op, void *arg)
{
Node *np = arg;
emitnode(np->left);
emitnode(np->right);
fprintf(outfp, "\t\"%s\tm", np->sym->name);
emitletter(np->type);
}
static void
emitexp(int op, void *arg)
{
Node *np = arg;
emitnode(np);
putc('\n', outfp);
freetree(np);
}
static void
emitfun(int op, void *arg)
{
Symbol *sym = arg, **sp;
emitdcl(op, arg);
fputs("{\n", outfp);
for (sp = sym->u.pars; sp && *sp; ++sp)
emit(ODECL, *sp);
fputs("\\\n", outfp);
}
static void
emittext(int op, void *arg)
{
fputs(optxt[op], outfp);
}
static void
emitsymid(int op, void *arg)
{
Symbol *sym = arg;
fprintf(outfp, optxt[op], sym->id);
}
Node *
node(int op, Type *tp, Node *lp, Node *rp)
{
Node *np;
np = xmalloc(sizeof(*np));
np->op = op;
np->type = tp;
np->sym = NULL;
np->flags = 0;
np->left = lp;
np->right = rp;
if (lp)
np->flags |= lp->flags & NEFFECT;
if (rp)
np->flags |= rp->flags & NEFFECT;
return np;
}
Node *
varnode(Symbol *sym)
{
Node *np;
Type *tp = sym->type;
np = node(OSYM, sym->type, NULL, NULL);
np->type = sym->type;
np->flags = (tp->op != FTN && tp->op != ARY) ? NLVAL : 0;
np->sym = sym;
return np;
}
Node *
constnode(Symbol *sym)
{
Node *np;
np = node(OSYM, sym->type, NULL, NULL);
np->type = sym->type;
np->flags = NCONST;
np->sym = sym;
return np;
}
Node *
sizeofnode(Type *tp)
{
Symbol *sym;
sym = newsym(NS_IDEN, NULL);
sym->type = sizettype;
sym->u.i = tp->size;
return constnode(sym);
}