ref: 12e226ac6798412271020d42645f69138715c519
dir: /utils/c2l/out.c/
#include "cc.h" #define INCREMENT 8 #define DEVNULL "/dev/null" static int indent = 0; static int fd = -1; static int nf = 0; static int mylineno = 1; typedef struct Com{ int lno; char *s; Node *n; int tba; struct Com *nxt; } Com; Com *hdc, *curc; typedef struct File{ char *s; char *f; char *m; int b; int loc; int in; int tg; Node *n; Com *c; struct File *nxt; } File; typedef struct Create{ char *s; struct Create *nxt; } Create; File *fs; Create *cs; static void genmsg(void); static int isloc(void); static void addcreate(char *s) { Create *c; if(strcmp(s, DEVNULL) == 0) return; c = (Create*)malloc(sizeof(Create)); c->s = malloc(strlen(s)+1); strcpy(c->s, s); c->nxt = cs; cs = c; } static int created(char *s) { Create *c; for(c = cs; c != nil; c = c->nxt) if(strcmp(s, c->s) == 0) return 1; return 0; } int dolog(void) { if(justcode) return 0; return domod || !doinc || inmain; } static char* curf(void) { File *f; for(f = fs; f != nil; f = f->nxt) if(f->f != nil) return f->s; return nil; } static char* curm(void) { File *f; for(f = fs; f != nil; f = f->nxt) if(f->f != nil) return f->m; return nil; } void setmod(Sym *s) { if(domod && s->mod == nil && ism() && !(doloc && !isloc())) s->mod = curm(); } char * outmod(char *buf, int up) { char *s, *t; s = curf(); if(s == nil) return ""; t = strchr(s, '.'); if(t != nil) *t = '\0'; strcpy(buf, s); if(t != nil) *t = '.'; if(up == 1 || (up < 0 && ism())) buf[0] = toupper(buf[0]); return buf; } int ism(void) { return !isb(); } int isb(void) { File *f; for(f = fs; f != nil; f = f->nxt) if(f->f != nil) return f->b; return 0; } static int isloc(void) { File *f; for(f = fs; f != nil; f = f->nxt) if(f->f != nil) return f->loc; return 0; } static File* pushf(void) { static File zfile; File *f; f = (File*)malloc(sizeof(File)); *f = zfile; f->s = nil; f->f = nil; f->m = nil; f->nxt = fs; fs = f; return f; } static void popf(void) { File *f; f = fs; fs = fs->nxt; if(f->s != nil) free(f->s); free(f); } static void setf(File *f, char *s) { int n; char *t; if(s != nil){ t = strrchr(s, '/'); f->loc = t == nil; if(t != nil) s = t+1; n = strlen(s); f->s = malloc(n+1); strcpy(f->s, s); s = f->s; if(n > 2 && s[n-2] == '.'){ f->m = malloc(n-1); strncpy(f->m, s, n-2); if(s[n-1] == 'h') s[n-1] = 'm'; else if(s[n-1] == 'c'){ s[n-1] = 'b'; f->b = 1; } else s = nil; } else s = nil; if(s == nil){ free(f->s); if(f->m != nil) free(f->m); f->s = nil; f->m = nil; } } f->f = f->s; if(f->s != nil && nf > 0){ if(doinc || doloc && !f->loc) f->f = DEVNULL; else if(!domod) f->f = nil; } } void outpush0(char *s, Node *n) { File *f; f = pushf(); setf(f, s); if(f->f != nil){ nf++; f->tg = taggen; taggen = 0; f->n = n; f->c = hdc; hdc = nil; } } void outpop0(int lno) { File *f; USED(lno); f = fs; if(f->f != nil){ nf--; taggen = f->tg; f->n->left = (void*)hdc; hdc = f->c; } popf(); } void outpush2(char *s, Node *n) { File *f; f = pushf(); setf(f, s); if(f->f != nil){ if(fd >= 0){ newsec(0); close(fd); close(1); fd = -1; } if(created(f->f)) f->f = DEVNULL; /* don't overwrite original if included again */ fd = create(f->f, OWRITE, 0664); if(fd >= 0) addcreate(f->f); mydup(fd, 1); nf++; f->tg = taggen; taggen = 0; f->c = hdc; if(n != Z) hdc = (void*)n->left; else hdc = nil; f->in = indent; indent = 0; genmsg(); pgen(f->b); } } void outpop2(int lno) { File *f, *g; f = fs; if(f->f != nil){ if(fd >= 0){ newsec(0); output(lno, 1); epgen(f->b); close(fd); close(1); fd = -1; } for(g = fs->nxt; g != nil; g = g->nxt){ if(g->f != nil){ fd = open(g->f, OWRITE); seek(fd, 0, 2); mydup(fd, 1); break; } } nf--; taggen = f->tg; hdc = f->c; indent = f->in; } popf(); } static void xprint(char *s) { if(nerrors == 0) print(s); } static int tot = 0; static void doindent(int d) { int i; for(i = 0; i < d/8; i++) xprint("\t"); for(i = 0; i < d%8; i++) xprint(" "); } void incind(void) { indent += INCREMENT; } void decind(void) { indent -= INCREMENT; } int zeroind(void) { int i = indent; indent = 0; return i; } void restoreind(int i) { indent = i; } void newline0(void) { xprint("\n"); tot = 0; mylineno++; } void newline(void) { if(!outcom(1)){ xprint("\n"); mylineno++; } tot = 0; } static void lprint(char *s) { if(tot == 0) { doindent(indent); tot += indent; } xprint(s); tot += strlen(s); } void prline(char *s) { xprint(s); xprint("\n"); mylineno++; } void prdelim(char *s) { if(*s == '%'){ if(*++s == '=') lprint("%%="); else lprint("%%"); return; } lprint(s); } void prkeywd(char *kw) { lprint(kw); } void prid(char *s) { lprint(s); } static void priddol(char *s, int dol) { char *t; char buf[128]; if(dol){ t = strchr(s, '$'); if(t != nil) *t = '_'; lprint(s); if(t != nil){ strcpy(buf, s); while(slookup(buf)->type != T){ strcat(buf, "x"); lprint("x"); } *t = '$'; } } else lprint(s); } void prsym(Sym *s, int mod) { char buf[128]; int c; if(mod && s->mod && strcmp(s->mod, curm()) != 0 && (!s->limbo || s->class == CEXTERN)){ c = isconsym(s); if(c >= 0){ if(c){ s->mod[0] = toupper(s->mod[0]); lprint(s->mod); s->mod[0] = tolower(s->mod[0]); } else lprint(s->mod); lprint("->"); usemod(s, !c); } } if(s->lname) prid(s->lname); else{ priddol(s->name, s->class == CSTATIC); if(s->lkw){ strcpy(buf, s->name); for(;;){ strcat(buf, "x"); lprint("x"); s = slookup(buf); if(s->type == T) break; } } } } int arrow(Sym *s) { if(s->mod && strcmp(s->mod, curm()) != 0) return isconsym(s) >= 0; return 0; } void prsym0(Sym *s) { int c; if(s->mod && strcmp(s->mod, curm()) != 0){ c = isconsym(s); if(c >= 0) usemod(s, !c); } } static int isprintable(int c) { if(c >= 0x20 && c <= 0x7e) return 1; return c == '\0' || c == '\n' || c == '\t' || c == '\b' || c == '\r' || c == '\f' || c == '\a' || c == '\v'; } static int hex(int c) { if(c < 10) return c+'0'; return c+'a'-10; } void prchar0(vlong x, int quote) { int c, e, i = 0; static char buf[16]; if(quote) buf[i++] = '\''; c = x; if(c < 0 || c > 255 || !isprintable(c)){ if(c&0xffff0000) diag(Z, "character too big"); buf[i++] = '\\'; buf[i++] = 'u'; buf[i++] = hex((c>>12)&0xf); buf[i++] = hex((c>>8)&0xf); buf[i++] = hex((c>>4)&0xf); buf[i++] = hex((c>>0)&0xf); } else{ e = 0; switch(c){ case '\n': e = 'n'; break; case '\t': e = 't'; break; case '\b': e = 'b'; break; case '\r': e = 'r'; break; case '\f': e = 'f'; break; case '\a': e = 'a'; break; case '\v': e = 'v'; break; case '"': if(!quote) e = '"'; break; case '\'': if(quote) e = '\''; break; case '\\': e = '\\'; break; case '%': buf[i++] = c; break; case 0: e = '0'; if(strings) prcom("nul byte in string ?", Z); break; } if(e != 0){ buf[i++] = '\\'; c = e; } buf[i++] = c; } if(quote) buf[i++] = '\''; buf[i] = '\0'; lprint(buf); } void prchar(vlong x) { prchar0(x, 1); } void prstr(char *s) { uchar *t; Rune r; t = (uchar*)s; lprint("\""); while(*t != 0){ if(*t & 0x80){ t += chartorune(&r, (char*)t); prchar0(r, 0); } else prchar0(*t++, 0); } lprint("\""); } void prlstr(Rune *s) { lprint("\""); while(*s != 0) prchar0(*s++, 0); lprint("\""); } void prreal(double x, char *s, int b) { static char buf[128]; if(b != KDEC) diag(Z, "not base 10 in prreal"); if(s != nil) lprint(s); else{ sprint(buf, "%f", x); lprint(buf); } } void prnum(vlong x, int b, Type *t) { static char buf[128]; int w; vlong m; w = 4; if(t != T) w = ewidth[t->etype]; m = MASK(8*w); if(b == KHEX) sprint(buf, "16r%llux", x&m); else if(b == KOCT) sprint(buf, "8r%lluo", x&m); else sprint(buf, "%lld", x); lprint(buf); } char *cb; int cn, csz; static void outcom0(Com *c) { Node *n; char *s, *t, *u; s = c->s; n = c->n; if(comm && c->tba){ t = strchr(s, '\n'); *t = '\0'; fprint(2, "%s:%d: %s", curf(), mylineno, s); *t = '\n'; if(n != Z){ mydup(2, 1); expgen(n); mydup(fd, 1); } fprint(2, "\n"); } while(*s != '\0'){ t = strchr(s, '\n'); *t = '\0'; if(tot != 0) prdelim("\t"); prdelim("# "); while((u = strchr(s, '%')) != nil){ /* do not let print interpret % ! */ *u = 0; lprint(s); *u = '%'; lprint("%%"); s = u+1; } lprint(s); if(n == Z) newline0(); *t = '\n'; s = t+1; } if(n != Z){ expgen(n); newline0(); } } int outcom(int f) { int lno, nl; Com *c; nl = 0; lno = pline+f; c = hdc; while(c != nil && c->lno < lno){ /* print("outcom: %d < %d (f=%d)\n", c->lno, lno, f); */ nl = 1; outcom0(c); hdc = hdc->nxt; free(c->s); free(c); c = hdc; } return nl; } void startcom(int lno) { Com *c, **ac; c = (Com *)malloc(sizeof(Com)); c->lno = lno; c->s = nil; c->n = Z; c->tba = 0; c->nxt = nil; for(ac = &hdc; *ac != nil && (*ac)->lno <= lno; ac = &(*ac)->nxt) ; c->nxt = *ac; curc = *ac = c; } void addcom(int rr) { int i, nb; char *ncb; char s[UTFmax]; Rune r[1]; if(rr >= Runeself){ r[0] = rr; nb = runetochar(s, r); } else{ nb = 1; s[0] = rr; } if(cn+nb-1 >= csz){ csz += 32; ncb = malloc(csz); memcpy(ncb, cb, cn); free(cb); cb = ncb; } for(i = 0; i < nb; i++) cb[cn++] = s[i]; } void endcom(void) { char *s; addcom('\n'); addcom('\0'); s = malloc(strlen(cb)+1); strcpy(s, cb); curc->s = s; /* print("com %d %s\n", curc->lno, s); */ cn = 0; } void linit() { csz = 32; cb = malloc(csz); sysinit(); } static void genmsg(void) { prline("#"); prline("# initially generated by c2l"); prline("#"); prline(""); } void prcom(char *s, Node *n) { Com *c; startcom(pline); c = curc; sprint(cb, "TBA %s", s); cn = strlen(cb); c->n = n; c->tba = 1; endcom(); } void output(long lno, int com) { /* print("output(%ld)\n", lno); */ pline = lno; if(com) outcom(0); } int exists(char *f) { int fd; fd = open(f, OREAD); close(fd); return fd >= 0; }