ref: a411870ee4640241e3c494367d922847da84f972
dir: /utils/8l/obj.c/
#define EXTERN #include "l.h" #include <ar.h> #ifndef DEFAULT #define DEFAULT '9' #endif char *noname = "<none>"; char symname[] = SYMDEF; char thechar = '8'; char *thestring = "386"; char** libdir; int nlibdir = 0; static int maxlibdir = 0; /* * -H0 -T0x40004C -D0x10000000 is garbage unix * -H1 -T0xd0 -R4 is unix coff * -H2 -T4128 -R4096 is plan9 format * -H3 -Tx -Rx is MS-DOS .COM * -H4 -Tx -Rx is fake MS-DOS .EXE * -H5 -T0x80100020 -R4096 is ELF */ void usage(void) { diag("usage: %s [-options] objects", argv0); errorexit(); } static int isobjfile(char *f) { int n, v; Biobuf *b; char buf1[5], buf2[SARMAG]; b = Bopen(f, OREAD); if(b == nil) return 0; n = Bread(b, buf1, 5); if(n == 5 && (buf1[2] == 1 && buf1[3] == '<' || buf1[3] == 1 && buf1[4] == '<')) v = 1; /* good enough for our purposes */ else{ Bseek(b, 0, 0); n = Bread(b, buf2, SARMAG); v = n == SARMAG && strncmp(buf2, ARMAG, SARMAG) == 0; } Bterm(b); return v; } void main(int argc, char *argv[]) { int i, c; char *a; char name[LIBNAMELEN]; Binit(&bso, 1, OWRITE); cout = -1; listinit(); memset(debug, 0, sizeof(debug)); nerrors = 0; outfile = "8.out"; HEADTYPE = -1; INITTEXT = -1; INITTEXTP = -1; INITDAT = -1; INITRND = -1; INITENTRY = 0; ARGBEGIN { default: c = ARGC(); if(c >= 0 && c < sizeof(debug)) debug[c]++; break; case 'o': /* output to (next arg) */ outfile = ARGF(); break; case 'E': a = ARGF(); if(a) INITENTRY = a; break; case 'H': a = ARGF(); if(a) HEADTYPE = atolwhex(a); break; case 'L': addlibpath(EARGF(usage())); break; case 'T': a = ARGF(); if(a) INITTEXT = atolwhex(a); break; case 'P': a = ARGF(); if(a) INITTEXTP = atolwhex(a); break; case 'D': a = ARGF(); if(a) INITDAT = atolwhex(a); break; case 'R': a = ARGF(); if(a) INITRND = atolwhex(a); break; case 'x': /* produce export table */ doexp = 1; if(argv[1] != nil && argv[1][0] != '-' && !isobjfile(argv[1])){ a = ARGF(); if(strcmp(a, "*") == 0) allexport = 1; else readundefs(a, SEXPORT); } break; case 'u': /* produce dynamically loadable module */ dlm = 1; debug['l']++; if(argv[1] != nil && argv[1][0] != '-' && !isobjfile(argv[1])) readundefs(ARGF(), SIMPORT); break; } ARGEND USED(argc); if(*argv == 0) usage(); if(!debug['9'] && !debug['U'] && !debug['B']) debug[DEFAULT] = 1; a = getenv("ccroot"); if(a != nil && *a != '\0') { if(!fileexists(a)) { diag("nonexistent $ccroot: %s", a); errorexit(); } }else a = ""; snprint(name, sizeof(name), "%s/%s/lib", a, thestring); addlibpath(name); if(HEADTYPE == -1) { if(debug['U']) HEADTYPE = 1; if(debug['B']) HEADTYPE = 2; if(debug['9']) HEADTYPE = 2; } switch(HEADTYPE) { default: diag("unknown -H option"); errorexit(); case 0: /* this is garbage */ HEADR = 20L+56L; if(INITTEXT == -1) INITTEXT = 0x40004CL; if(INITDAT == -1) INITDAT = 0x10000000L; if(INITRND == -1) INITRND = 0; break; case 1: /* is unix coff */ HEADR = 0xd0L; if(INITTEXT == -1) INITTEXT = 0xd0; if(INITDAT == -1) INITDAT = 0x400000; if(INITRND == -1) INITRND = 0; break; case 2: /* plan 9 */ HEADR = 32L; if(INITTEXT == -1) INITTEXT = 4096+32; if(INITDAT == -1) INITDAT = 0; if(INITRND == -1) INITRND = 4096; break; case 3: /* MS-DOS .COM */ HEADR = 0; if(INITTEXT == -1) INITTEXT = 0x0100; if(INITDAT == -1) INITDAT = 0; if(INITRND == -1) INITRND = 4; break; case 4: /* fake MS-DOS .EXE */ HEADR = 0x200; if(INITTEXT == -1) INITTEXT = 0x0100; if(INITDAT == -1) INITDAT = 0; if(INITRND == -1) INITRND = 4; HEADR += (INITTEXT & 0xFFFF); if(debug['v']) Bprint(&bso, "HEADR = 0x%ld\n", HEADR); break; case 5: /* elf executable */ HEADR = rnd(Ehdr32sz+3*Phdr32sz, 16); if(INITTEXT == -1) INITTEXT = 0x80100020L; if(INITDAT == -1) INITDAT = 0; if(INITRND == -1) INITRND = 4096; break; } if (INITTEXTP == -1) INITTEXTP = INITTEXT; if(INITDAT != 0 && INITRND != 0) print("warning: -D0x%lux is ignored because of -R0x%lux\n", INITDAT, INITRND); if(debug['v']) Bprint(&bso, "HEADER = -H0x%ld -T0x%lux -D0x%lux -R0x%lux\n", HEADTYPE, INITTEXT, INITDAT, INITRND); Bflush(&bso); for(i=1; optab[i].as; i++) if(i != optab[i].as) { diag("phase error in optab: %d", i); errorexit(); } for(i=0; i<Ymax; i++) ycover[i*Ymax + i] = 1; ycover[Yi0*Ymax + Yi8] = 1; ycover[Yi1*Ymax + Yi8] = 1; ycover[Yi0*Ymax + Yi32] = 1; ycover[Yi1*Ymax + Yi32] = 1; ycover[Yi8*Ymax + Yi32] = 1; ycover[Yal*Ymax + Yrb] = 1; ycover[Ycl*Ymax + Yrb] = 1; ycover[Yax*Ymax + Yrb] = 1; ycover[Ycx*Ymax + Yrb] = 1; ycover[Yrx*Ymax + Yrb] = 1; ycover[Yax*Ymax + Yrx] = 1; ycover[Ycx*Ymax + Yrx] = 1; ycover[Yax*Ymax + Yrl] = 1; ycover[Ycx*Ymax + Yrl] = 1; ycover[Yrx*Ymax + Yrl] = 1; ycover[Yf0*Ymax + Yrf] = 1; ycover[Yal*Ymax + Ymb] = 1; ycover[Ycl*Ymax + Ymb] = 1; ycover[Yax*Ymax + Ymb] = 1; ycover[Ycx*Ymax + Ymb] = 1; ycover[Yrx*Ymax + Ymb] = 1; ycover[Yrb*Ymax + Ymb] = 1; ycover[Ym*Ymax + Ymb] = 1; ycover[Yax*Ymax + Yml] = 1; ycover[Ycx*Ymax + Yml] = 1; ycover[Yrx*Ymax + Yml] = 1; ycover[Yrl*Ymax + Yml] = 1; ycover[Ym*Ymax + Yml] = 1; for(i=0; i<D_NONE; i++) { reg[i] = -1; if(i >= D_AL && i <= D_BH) reg[i] = (i-D_AL) & 7; if(i >= D_AX && i <= D_DI) reg[i] = (i-D_AX) & 7; if(i >= D_F0 && i <= D_F0+7) reg[i] = (i-D_F0) & 7; } zprg.link = P; zprg.pcond = P; zprg.back = 2; zprg.as = AGOK; zprg.from.type = D_NONE; zprg.from.index = D_NONE; zprg.from.scale = 1; zprg.to = zprg.from; pcstr = "%.6lux "; nuxiinit(); histgen = 0; textp = P; datap = P; edatap = P; pc = 0; dtype = 4; cout = create(outfile, 1, 0775); if(cout < 0) { diag("cannot create %s: %r", outfile); errorexit(); } version = 0; cbp = buf.cbuf; cbc = sizeof(buf.cbuf); firstp = prg(); lastp = firstp; if(INITENTRY == 0) { INITENTRY = "_main"; if(debug['p']) INITENTRY = "_mainp"; if(!debug['l']) lookup(INITENTRY, 0)->type = SXREF; } else if(!(*INITENTRY >= '0' && *INITENTRY <= '9')) lookup(INITENTRY, 0)->type = SXREF; while(*argv) objfile(*argv++); if(!debug['l']) loadlib(); firstp = firstp->link; if(firstp == P) errorexit(); if(doexp || dlm){ EXPTAB = "_exporttab"; zerosig(EXPTAB); zerosig("etext"); zerosig("edata"); zerosig("end"); if(dlm){ import(); HEADTYPE = 2; INITTEXT = INITDAT = 0; INITRND = 8; INITENTRY = EXPTAB; } export(); } patch(); follow(); dodata(); dostkoff(); if(debug['p']) if(debug['1']) doprof1(); else doprof2(); span(); doinit(); asmb(); undef(); if(debug['v']) { Bprint(&bso, "%5.2f cpu time\n", cputime()); Bprint(&bso, "%ld symbols\n", nsymbol); Bprint(&bso, "%ld memory used\n", thunk); Bprint(&bso, "%d sizeof adr\n", sizeof(Adr)); Bprint(&bso, "%d sizeof prog\n", sizeof(Prog)); } Bflush(&bso); errorexit(); } void addlibpath(char *arg) { char **p; if(nlibdir >= maxlibdir) { if(maxlibdir == 0) maxlibdir = 8; else maxlibdir *= 2; p = malloc(maxlibdir*sizeof(*p)); if(p == nil) { diag("out of memory"); errorexit(); } memmove(p, libdir, nlibdir*sizeof(*p)); free(libdir); libdir = p; } libdir[nlibdir++] = strdup(arg); } char* findlib(char *file) { int i; char name[LIBNAMELEN]; for(i = 0; i < nlibdir; i++) { snprint(name, sizeof(name), "%s/%s", libdir[i], file); if(fileexists(name)) return libdir[i]; } return nil; } void loadlib(void) { int i; long h; Sym *s; loop: xrefresolv = 0; for(i=0; i<libraryp; i++) { if(debug['v']) Bprint(&bso, "%5.2f autolib: %s (from %s)\n", cputime(), library[i], libraryobj[i]); objfile(library[i]); } if(xrefresolv) for(h=0; h<nelem(hash); h++) for(s = hash[h]; s != S; s = s->link) if(s->type == SXREF) goto loop; } void errorexit(void) { if(nerrors) { if(cout >= 0) remove(outfile); exits("error"); } exits(0); } void objfile(char *file) { long off, esym, cnt, l; int f, work; Sym *s; char magbuf[SARMAG]; char name[LIBNAMELEN], pname[LIBNAMELEN]; struct ar_hdr arhdr; char *e, *start, *stop; if(debug['v']) Bprint(&bso, "%5.2f ldobj: %s\n", cputime(), file); Bflush(&bso); if(file[0] == '-' && file[1] == 'l') { snprint(pname, sizeof(pname), "lib%s.a", file+2); e = findlib(pname); if(e == nil) { diag("cannot find library: %s", file); errorexit(); } snprint(name, sizeof(name), "%s/%s", e, pname); file = name; } f = open(file, 0); if(f < 0) { diag("cannot open %s: %r", file); errorexit(); } l = read(f, magbuf, SARMAG); if(l != SARMAG || strncmp(magbuf, ARMAG, SARMAG)){ /* load it as a regular file */ l = seek(f, 0L, 2); seek(f, 0L, 0); ldobj(f, l, file); close(f); return; } l = read(f, &arhdr, SAR_HDR); if(l != SAR_HDR) { diag("%s: short read on archive file symbol header", file); goto out; } if(strncmp(arhdr.name, symname, strlen(symname))) { diag("%s: first entry not symbol header", file); goto out; } esym = SARMAG + SAR_HDR + atolwhex(arhdr.size); off = SARMAG + SAR_HDR; /* * just bang the whole symbol file into memory */ seek(f, off, 0); cnt = esym - off; start = malloc(cnt + 10); cnt = read(f, start, cnt); if(cnt <= 0){ close(f); return; } stop = &start[cnt]; memset(stop, 0, 10); work = 1; while(work) { if(debug['v']) Bprint(&bso, "%5.2f library pass: %s\n", cputime(), file); Bflush(&bso); work = 0; for(e = start; e < stop; e = strchr(e+5, 0) + 1) { s = lookup(e+5, 0); if(s->type != SXREF) continue; sprint(pname, "%s(%s)", file, s->name); if(debug['v']) Bprint(&bso, "%5.2f library: %s\n", cputime(), pname); Bflush(&bso); l = e[1] & 0xff; l |= (e[2] & 0xff) << 8; l |= (e[3] & 0xff) << 16; l |= (e[4] & 0xff) << 24; seek(f, l, 0); /* need readn to read the dumps (at least) */ l = readn(f, &arhdr, SAR_HDR); if(l != SAR_HDR) goto bad; if(strncmp(arhdr.fmag, ARFMAG, sizeof(arhdr.fmag))) goto bad; l = atolwhex(arhdr.size); ldobj(f, l, pname); if(s->type == SXREF) { diag("%s: failed to load: %s", file, s->name); errorexit(); } work = 1; xrefresolv = 1; } } return; bad: diag("%s: bad or out of date archive", file); out: close(f); } int zaddr(uchar *p, Adr *a, Sym *h[]) { int c, t, i; int l; Sym *s; Auto *u; t = p[0]; c = 1; if(t & T_INDEX) { a->index = p[c]; a->scale = p[c+1]; c += 2; } else { a->index = D_NONE; a->scale = 0; } a->offset = 0; if(t & T_OFFSET) { a->offset = p[c] | (p[c+1]<<8) | (p[c+2]<<16) | (p[c+3]<<24); c += 4; } a->sym = S; if(t & T_SYM) { a->sym = h[p[c]]; c++; } a->type = D_NONE; if(t & T_FCONST) { a->ieee.l = p[c] | (p[c+1]<<8) | (p[c+2]<<16) | (p[c+3]<<24); a->ieee.h = p[c+4] | (p[c+5]<<8) | (p[c+6]<<16) | (p[c+7]<<24); c += 8; a->type = D_FCONST; } else if(t & T_SCONST) { for(i=0; i<NSNAME; i++) a->scon[i] = p[c+i]; c += NSNAME; a->type = D_SCONST; } if(t & T_TYPE) { a->type = p[c]; c++; } s = a->sym; if(s == S) return c; t = a->type; if(t != D_AUTO && t != D_PARAM) return c; l = a->offset; for(u=curauto; u; u=u->link) { if(u->asym == s) if(u->type == t) { if(u->aoffset > l) u->aoffset = l; return c; } } while(nhunk < sizeof(Auto)) gethunk(); u = (Auto*)hunk; nhunk -= sizeof(Auto); hunk += sizeof(Auto); u->link = curauto; curauto = u; u->asym = s; u->aoffset = l; u->type = t; return c; } void addlib(char *obj) { char fn1[LIBNAMELEN], fn2[LIBNAMELEN], comp[LIBNAMELEN], *p, *name; int i, search; if(histfrogp <= 0) return; name = fn1; search = 0; if(histfrog[0]->name[1] == '/') { sprint(name, ""); i = 1; } else if(histfrog[0]->name[1] == '.') { sprint(name, "."); i = 0; } else { sprint(name, ""); i = 0; search = 1; } for(; i<histfrogp; i++) { snprint(comp, sizeof comp, histfrog[i]->name+1); for(;;) { p = strstr(comp, "$O"); if(p == 0) break; memmove(p+1, p+2, strlen(p+2)+1); p[0] = thechar; } for(;;) { p = strstr(comp, "$M"); if(p == 0) break; if(strlen(comp)+strlen(thestring)-2+1 >= sizeof comp) { diag("library component too long"); return; } memmove(p+strlen(thestring), p+2, strlen(p+2)+1); memmove(p, thestring, strlen(thestring)); } if(strlen(fn1) + strlen(comp) + 3 >= sizeof(fn1)) { diag("library component too long"); return; } if(i > 0 || !search) strcat(fn1, "/"); strcat(fn1, comp); } cleanname(name); if(search){ p = findlib(name); if(p != nil){ snprint(fn2, sizeof(fn2), "%s/%s", p, name); name = fn2; } } for(i=0; i<libraryp; i++) if(strcmp(name, library[i]) == 0) return; if(libraryp == nelem(library)){ diag("too many autolibs; skipping %s", name); return; } p = malloc(strlen(name) + 1); strcpy(p, name); library[libraryp] = p; p = malloc(strlen(obj) + 1); strcpy(p, obj); libraryobj[libraryp] = p; libraryp++; } void addhist(long line, int type) { Auto *u; Sym *s; int i, j, k; u = malloc(sizeof(Auto)); s = malloc(sizeof(Sym)); s->name = malloc(2*(histfrogp+1) + 1); u->asym = s; u->type = type; u->aoffset = line; u->link = curhist; curhist = u; j = 1; for(i=0; i<histfrogp; i++) { k = histfrog[i]->value; s->name[j+0] = k>>8; s->name[j+1] = k; j += 2; } } void histtoauto(void) { Auto *l; while(l = curhist) { curhist = l->link; l->link = curauto; curauto = l; } } void collapsefrog(Sym *s) { int i; /* * bad encoding of path components only allows * MAXHIST components. if there is an overflow, * first try to collapse xxx/.. */ for(i=1; i<histfrogp; i++) if(strcmp(histfrog[i]->name+1, "..") == 0) { memmove(histfrog+i-1, histfrog+i+1, (histfrogp-i-1)*sizeof(histfrog[0])); histfrogp--; goto out; } /* * next try to collapse . */ for(i=0; i<histfrogp; i++) if(strcmp(histfrog[i]->name+1, ".") == 0) { memmove(histfrog+i, histfrog+i+1, (histfrogp-i-1)*sizeof(histfrog[0])); goto out; } /* * last chance, just truncate from front */ memmove(histfrog+0, histfrog+1, (histfrogp-1)*sizeof(histfrog[0])); out: histfrog[histfrogp-1] = s; } void nopout(Prog *p) { p->as = ANOP; p->from.type = D_NONE; p->to.type = D_NONE; } uchar* readsome(int f, uchar *buf, uchar *good, uchar *stop, int max) { int n; n = stop - good; memmove(buf, good, stop - good); stop = buf + n; n = MAXIO - n; if(n > max) n = max; n = read(f, stop, n); if(n <= 0) return 0; return stop + n; } void ldobj(int f, long c, char *pn) { long ipc; Prog *p, *t; uchar *bloc, *bsize, *stop; int v, o, r, skip; Sym *h[NSYM], *s, *di; ulong sig; static int files; static char **filen; char **nfilen; if((files&15) == 0){ nfilen = malloc((files+16)*sizeof(char*)); memmove(nfilen, filen, files*sizeof(char*)); free(filen); filen = nfilen; } filen[files++] = strdup(pn); bsize = buf.xbuf; bloc = buf.xbuf; di = S; newloop: memset(h, 0, sizeof(h)); version++; histfrogp = 0; ipc = pc; skip = 0; loop: if(c <= 0) goto eof; r = bsize - bloc; if(r < 100 && r < c) { /* enough for largest prog */ bsize = readsome(f, buf.xbuf, bloc, bsize, c); if(bsize == 0) goto eof; bloc = buf.xbuf; goto loop; } o = bloc[0] | (bloc[1] << 8); if(o <= AXXX || o >= ALAST) { if(o < 0) goto eof; diag("%s: opcode out of range %d", pn, o); print(" probably not a .8 file\n"); errorexit(); } if(o == ANAME || o == ASIGNAME) { sig = 0; if(o == ASIGNAME) { sig = bloc[2] | (bloc[3]<<8) | (bloc[4]<<16) | (bloc[5]<<24); bloc += 4; c -= 4; } stop = memchr(&bloc[4], 0, bsize-&bloc[4]); if(stop == 0){ bsize = readsome(f, buf.xbuf, bloc, bsize, c); if(bsize == 0) goto eof; bloc = buf.xbuf; stop = memchr(&bloc[4], 0, bsize-&bloc[4]); if(stop == 0){ fprint(2, "%s: name too long\n", pn); errorexit(); } } v = bloc[2]; /* type */ o = bloc[3]; /* sym */ bloc += 4; c -= 4; r = 0; if(v == D_STATIC) r = version; s = lookup((char*)bloc, r); c -= &stop[1] - bloc; bloc = stop + 1; if(debug['S'] && r == 0) sig = 1729; if(sig != 0){ if(s->sig != 0 && s->sig != sig) diag("incompatible type signatures %lux(%s) and %lux(%s) for %s", s->sig, filen[s->file], sig, pn, s->name); s->sig = sig; s->file = files-1; } if(debug['W']) print(" ANAME %s\n", s->name); h[o] = s; if((v == D_EXTERN || v == D_STATIC) && s->type == 0) s->type = SXREF; if(v == D_FILE) { if(s->type != SFILE) { histgen++; s->type = SFILE; s->value = histgen; } if(histfrogp < MAXHIST) { histfrog[histfrogp] = s; histfrogp++; } else collapsefrog(s); } goto loop; } while(nhunk < sizeof(Prog)) gethunk(); p = (Prog*)hunk; nhunk -= sizeof(Prog); hunk += sizeof(Prog); p->as = o; p->line = bloc[2] | (bloc[3] << 8) | (bloc[4] << 16) | (bloc[5] << 24); p->back = 2; r = zaddr(bloc+6, &p->from, h) + 6; r += zaddr(bloc+r, &p->to, h); bloc += r; c -= r; if(debug['W']) print("%P\n", p); switch(p->as) { case AHISTORY: if(p->to.offset == -1) { addlib(pn); histfrogp = 0; goto loop; } addhist(p->line, D_FILE); /* 'z' */ if(p->to.offset) addhist(p->to.offset, D_FILE1); /* 'Z' */ histfrogp = 0; goto loop; case AEND: histtoauto(); if(curtext != P) curtext->to.autom = curauto; curauto = 0; curtext = P; if(c) goto newloop; return; case AGLOBL: s = p->from.sym; if(s->type == 0 || s->type == SXREF) { s->type = SBSS; s->value = 0; } if(s->type != SBSS) { diag("%s: redefinition: %s in %s", pn, s->name, TNAME); s->type = SBSS; s->value = 0; } if(p->to.offset > s->value) s->value = p->to.offset; goto loop; case ADYNT: if(p->to.sym == S) { diag("DYNT without a sym\n%P", p); break; } di = p->to.sym; p->from.scale = 4; if(di->type == SXREF) { if(debug['z']) Bprint(&bso, "%P set to %d\n", p, dtype); di->type = SCONST; di->value = dtype; dtype += 4; } if(p->from.sym == S) break; p->from.offset = di->value; p->from.sym->type = SDATA; if(curtext == P) { diag("DYNT not in text: %P", p); break; } p->to.sym = curtext->from.sym; p->to.type = D_ADDR; p->to.index = D_EXTERN; goto data; case AINIT: if(p->from.sym == S) { diag("INIT without a sym\n%P", p); break; } if(di == S) { diag("INIT without previous DYNT\n%P", p); break; } p->from.offset = di->value; p->from.sym->type = SDATA; goto data; case ADATA: data: if(edatap == P) datap = p; else edatap->link = p; edatap = p; p->link = P; goto loop; case AGOK: diag("%s: GOK opcode in %s", pn, TNAME); pc++; goto loop; case ATEXT: if(curtext != P) { histtoauto(); curtext->to.autom = curauto; curauto = 0; } skip = 0; curtext = p; s = p->from.sym; if(s == S) { diag("%s: no TEXT symbol: %P", pn, p); errorexit(); } if(s->type != 0 && s->type != SXREF) { if(p->from.scale & DUPOK) { skip = 1; goto casdef; } diag("%s: redefinition: %s\n%P", pn, s->name, p); } s->type = STEXT; s->value = pc; lastp->link = p; lastp = p; p->pc = pc; pc++; if(textp == P) { textp = p; etextp = p; goto loop; } etextp->pcond = p; etextp = p; goto loop; case AFMOVF: case AFADDF: case AFSUBF: case AFSUBRF: case AFMULF: case AFDIVF: case AFDIVRF: case AFCOMF: case AFCOMFP: if(skip) goto casdef; if(p->from.type == D_FCONST) { /* size sb 9 max */ sprint(literal, "$%lux", ieeedtof(&p->from.ieee)); s = lookup(literal, 0); if(s->type == 0) { s->type = SBSS; s->value = 4; t = prg(); t->as = ADATA; t->line = p->line; t->from.type = D_EXTERN; t->from.sym = s; t->from.scale = 4; t->to = p->from; if(edatap == P) datap = t; else edatap->link = t; edatap = t; t->link = P; } p->from.type = D_EXTERN; p->from.sym = s; p->from.offset = 0; } goto casdef; case AFMOVD: case AFADDD: case AFSUBD: case AFSUBRD: case AFMULD: case AFDIVD: case AFDIVRD: case AFCOMD: case AFCOMDP: if(skip) goto casdef; if(p->from.type == D_FCONST) { /* size sb 18 max */ sprint(literal, "$%lux.%lux", p->from.ieee.l, p->from.ieee.h); s = lookup(literal, 0); if(s->type == 0) { s->type = SBSS; s->value = 8; t = prg(); t->as = ADATA; t->line = p->line; t->from.type = D_EXTERN; t->from.sym = s; t->from.scale = 8; t->to = p->from; if(edatap == P) datap = t; else edatap->link = t; edatap = t; t->link = P; } p->from.type = D_EXTERN; p->from.sym = s; p->from.offset = 0; } goto casdef; casdef: default: if(skip) nopout(p); if(p->to.type == D_BRANCH) p->to.offset += ipc; lastp->link = p; lastp = p; p->pc = pc; pc++; goto loop; } goto loop; eof: diag("truncated object file: %s", pn); } Sym* lookup(char *symb, int v) { Sym *s; char *p; long h; int l, c; h = v; for(p=symb; c = *p; p++) h = h+h+h + c; l = (p - symb) + 1; h &= 0xffffff; h %= NHASH; for(s = hash[h]; s != S; s = s->link) if(s->version == v) if(memcmp(s->name, symb, l) == 0) return s; while(nhunk < sizeof(Sym)) gethunk(); s = (Sym*)hunk; nhunk -= sizeof(Sym); hunk += sizeof(Sym); s->name = malloc(l + 1); memmove(s->name, symb, l); s->link = hash[h]; s->type = 0; s->version = v; s->value = 0; s->sig = 0; hash[h] = s; nsymbol++; return s; } Prog* prg(void) { Prog *p; while(nhunk < sizeof(Prog)) gethunk(); p = (Prog*)hunk; nhunk -= sizeof(Prog); hunk += sizeof(Prog); *p = zprg; return p; } Prog* copyp(Prog *q) { Prog *p; p = prg(); *p = *q; return p; } Prog* appendp(Prog *q) { Prog *p; p = prg(); p->link = q->link; q->link = p; p->line = q->line; return p; } void gethunk(void) { char *h; long nh; nh = NHUNK; if(thunk >= 5L*NHUNK) { nh = 5L*NHUNK; if(thunk >= 25L*NHUNK) nh = 25L*NHUNK; } h = mysbrk(nh); if(h == (char*)-1) { diag("out of memory"); errorexit(); } hunk = h; nhunk = nh; thunk += nh; } void doprof1(void) { Sym *s; long n; Prog *p, *q; if(debug['v']) Bprint(&bso, "%5.2f profile 1\n", cputime()); Bflush(&bso); s = lookup("__mcount", 0); n = 1; for(p = firstp->link; p != P; p = p->link) { if(p->as == ATEXT) { q = prg(); q->line = p->line; q->link = datap; datap = q; q->as = ADATA; q->from.type = D_EXTERN; q->from.offset = n*4; q->from.sym = s; q->from.scale = 4; q->to = p->from; q->to.type = D_CONST; q = prg(); q->line = p->line; q->pc = p->pc; q->link = p->link; p->link = q; p = q; p->as = AADDL; p->from.type = D_CONST; p->from.offset = 1; p->to.type = D_EXTERN; p->to.sym = s; p->to.offset = n*4 + 4; n += 2; continue; } } q = prg(); q->line = 0; q->link = datap; datap = q; q->as = ADATA; q->from.type = D_EXTERN; q->from.sym = s; q->from.scale = 4; q->to.type = D_CONST; q->to.offset = n; s->type = SBSS; s->value = n*4; } void doprof2(void) { Sym *s2, *s4; Prog *p, *q, *q2, *ps2, *ps4; if(debug['v']) Bprint(&bso, "%5.2f profile 2\n", cputime()); Bflush(&bso); if(debug['e']){ s2 = lookup("_tracein", 0); s4 = lookup("_traceout", 0); }else{ s2 = lookup("_profin", 0); s4 = lookup("_profout", 0); } if(s2->type != STEXT || s4->type != STEXT) { if(debug['e']) diag("_tracein/_traceout not defined %d %d", s2->type, s4->type); else diag("_profin/_profout not defined"); return; } ps2 = P; ps4 = P; for(p = firstp; p != P; p = p->link) { if(p->as == ATEXT) { if(p->from.sym == s2) { p->from.scale = 1; ps2 = p; } if(p->from.sym == s4) { p->from.scale = 1; ps4 = p; } } } for(p = firstp; p != P; p = p->link) { if(p->as == ATEXT) { curtext = p; if(p->from.scale & NOPROF) { /* dont profile */ for(;;) { q = p->link; if(q == P) break; if(q->as == ATEXT) break; p = q; } continue; } /* * JMPL profin */ q = prg(); q->line = p->line; q->pc = p->pc; q->link = p->link; if(debug['e']){ /* embedded tracing */ q2 = prg(); p->link = q2; q2->link = q; q2->line = p->line; q2->pc = p->pc; q2->as = AJMP; q2->to.type = D_BRANCH; q2->to.sym = p->to.sym; q2->pcond = q->link; }else p->link = q; p = q; p->as = ACALL; p->to.type = D_BRANCH; p->pcond = ps2; p->to.sym = s2; continue; } if(p->as == ARET) { /* * RET (default) */ if(debug['e']){ /* embedded tracing */ q = prg(); q->line = p->line; q->pc = p->pc; q->link = p->link; p->link = q; p = q; } /* * RET */ q = prg(); q->as = ARET; q->from = p->from; q->to = p->to; q->link = p->link; p->link = q; /* * JAL profout */ p->as = ACALL; p->from = zprg.from; p->to = zprg.to; p->to.type = D_BRANCH; p->pcond = ps4; p->to.sym = s4; p = q; continue; } } } void nuxiinit(void) { int i, c; for(i=0; i<4; i++) { c = find1(0x04030201L, i+1); if(i < 2) inuxi2[i] = c; if(i < 1) inuxi1[i] = c; inuxi4[i] = c; fnuxi4[i] = c; fnuxi8[i] = c; fnuxi8[i+4] = c+4; } if(debug['v']) { Bprint(&bso, "inuxi = "); for(i=0; i<1; i++) Bprint(&bso, "%d", inuxi1[i]); Bprint(&bso, " "); for(i=0; i<2; i++) Bprint(&bso, "%d", inuxi2[i]); Bprint(&bso, " "); for(i=0; i<4; i++) Bprint(&bso, "%d", inuxi4[i]); Bprint(&bso, "\nfnuxi = "); for(i=0; i<4; i++) Bprint(&bso, "%d", fnuxi4[i]); Bprint(&bso, " "); for(i=0; i<8; i++) Bprint(&bso, "%d", fnuxi8[i]); Bprint(&bso, "\n"); } Bflush(&bso); } int find1(long l, int c) { char *p; int i; p = (char*)&l; for(i=0; i<4; i++) if(*p++ == c) return i; return 0; } int find2(long l, int c) { short *p; int i; p = (short*)&l; for(i=0; i<4; i+=2) { if(((*p >> 8) & 0xff) == c) return i; if((*p++ & 0xff) == c) return i+1; } return 0; } long ieeedtof(Ieee *e) { int exp; long v; if(e->h == 0) return 0; exp = (e->h>>20) & ((1L<<11)-1L); exp -= (1L<<10) - 2L; v = (e->h & 0xfffffL) << 3; v |= (e->l >> 29) & 0x7L; if((e->l >> 28) & 1) { v++; if(v & 0x800000L) { v = (v & 0x7fffffL) >> 1; exp++; } } if(exp <= -126 || exp >= 130) diag("double fp to single fp overflow"); v |= ((exp + 126) & 0xffL) << 23; v |= e->h & 0x80000000L; return v; } double ieeedtod(Ieee *ieeep) { Ieee e; double fr; int exp; if(ieeep->h & (1L<<31)) { e.h = ieeep->h & ~(1L<<31); e.l = ieeep->l; return -ieeedtod(&e); } if(ieeep->l == 0 && ieeep->h == 0) return 0; fr = ieeep->l & ((1L<<16)-1L); fr /= 1L<<16; fr += (ieeep->l>>16) & ((1L<<16)-1L); fr /= 1L<<16; fr += (ieeep->h & (1L<<20)-1L) | (1L<<20); fr /= 1L<<21; exp = (ieeep->h>>20) & ((1L<<11)-1L); exp -= (1L<<10) - 2L; return ldexp(fr, exp); } void undefsym(Sym *s) { int n; n = imports; if(s->value != 0) diag("value != 0 on SXREF"); if(n >= 1<<Rindex) diag("import index %d out of range", n); s->value = n<<Roffset; s->type = SUNDEF; imports++; } void zerosig(char *sp) { Sym *s; s = lookup(sp, 0); s->sig = 0; } void readundefs(char *f, int t) { int i, n; Sym *s; Biobuf *b; char *l, buf[256], *fields[64]; if(f == nil) return; b = Bopen(f, OREAD); if(b == nil){ diag("could not open %s: %r", f); errorexit(); } while((l = Brdline(b, '\n')) != nil){ n = Blinelen(b); if(n >= sizeof(buf)){ diag("%s: line too long", f); errorexit(); } memmove(buf, l, n); buf[n-1] = '\0'; n = getfields(buf, fields, nelem(fields), 1, " \t\r\n"); if(n == nelem(fields)){ diag("%s: bad format", f); errorexit(); } for(i = 0; i < n; i++){ s = lookup(fields[i], 0); s->type = SXREF; s->subtype = t; if(t == SIMPORT) nimports++; else nexports++; } } Bterm(b); }