/* * nm.c -- drive nm */#include <lib9.h>#include <ar.h>#include <bio.h>#include <mach.h> enum{	CHUNK	=	256	/* must be power of 2 */}; char	*errs;			/* exit status */char	*filename;		/* current file */char	symname[]="__.SYMDEF";	/* table of contents file name */int	multifile;		/* processing multiple files */int	aflag;int	gflag;int	hflag;int	nflag;int	sflag;int	uflag;int	Tflag; Sym	**fnames;		/* file path translation table */Sym	**symptr;int	nsym;Biobuf	bout;char*	argv0; void	error(char*, ...);void	execsyms(int);void	psym(Sym*, void*);void	printsyms(Sym**, long);void	doar(Biobuf*);void	dofile(Biobuf*);void	zenter(Sym*); voidusage(void){	fprint(2, "usage: nm [-aghnsTu] file ...\n");	exits("usage");} voidmain(int argc, char *argv[]){	int i;	Biobuf	*bin; 	Binit(&bout, 1, OWRITE);	argv0 = argv[0];	ARGBEGIN {	default:	usage();	case 'a':	aflag = 1; break;	case 'g':	gflag = 1; break;	case 'h':	hflag = 1; break;	case 'n':	nflag = 1; break;	case 's':	sflag = 1; break;	case 'u':	uflag = 1; break;	case 'T':	Tflag = 1; break;	} ARGEND	if (argc == 0)		usage();	if (argc > 1)		multifile++;	for(i=0; i<argc; i++){		filename = argv[i];		bin = Bopen(filename, OREAD);		if(bin == 0){			error("cannot open %s", filename);			continue;		}		if (isar(bin))			doar(bin);		else{			Bseek(bin, 0, 0);			dofile(bin);		}		Bterm(bin);	}	exits(errs);} /* * read an archive file, * processing the symbols for each intermediate file in it. */voiddoar(Biobuf *bp){	vlong offset;	int size, obj;	char membername[SARNAME]; 	multifile = 1;	for (offset = Boffset(bp);;offset += size) {		size = nextar(bp, offset, membername);		if (size < 0) {			error("phase error on ar header %lld", offset);			return;		}		if (size == 0)			return;		if (strcmp(membername, symname) == 0)			continue;		obj = objtype(bp, 0);		if (obj < 0) {			error("inconsistent file %s in %s",					membername, filename);			return;		}		if (!readar(bp, obj, offset+size, 1)) {			error("invalid symbol reference in file %s",					membername);			return;		}		filename = membername;		nsym=0;		objtraverse(psym, 0);		printsyms(symptr, nsym);	}} /* * process symbols in a file */voiddofile(Biobuf *bp){	int obj; 	obj = objtype(bp, 0);	if (obj < 0)		execsyms(Bfildes(bp));	else	if (readobj(bp, obj)) {		nsym = 0;		objtraverse(psym, 0);		printsyms(symptr, nsym);	}} /* * comparison routine for sorting the symbol table *	this screws up on 'z' records when aflag == 1 */intcmp(const void *vs, const void *vt){	Sym **s, **t; 	s = (Sym**)vs;	t = (Sym**)vt;	if(nflag)		if((*s)->value < (*t)->value)			return -1;		else			return (*s)->value > (*t)->value;	return strcmp((*s)->name, (*t)->name);}/* * enter a symbol in the table of filename elements */voidzenter(Sym *s){	static int maxf = 0; 	if (s->value >= maxf) {		maxf = (s->value+CHUNK-1) &~ (CHUNK-1);		fnames = realloc(fnames, maxf*sizeof(*fnames));		if(fnames == 0) {			error("out of memory", argv0);			exits("memory");		}	}	fnames[s->value] = s;} /* * get the symbol table from an executable file, if it has one */voidexecsyms(int fd){	Fhdr f;	Sym *s;	long n; 	seek(fd, 0, 0);	if (crackhdr(fd, &f) == 0) {		error("Can't read header for %s", filename);		return;	}	if (syminit(fd, &f) < 0)		return;	s = symbase(&n);	nsym = 0;	while(n--)		psym(s++, 0); 	printsyms(symptr, nsym);} voidpsym(Sym *s, void* p){	USED(p);	switch(s->type) {	case 'T':	case 'L':	case 'D':	case 'B':		if (uflag)			return;		if (!aflag && ((s->name[0] == '.' || s->name[0] == '$')))			return;		break;	case 'b':	case 'd':	case 'l':	case 't':		if (uflag || gflag)			return;		if (!aflag && ((s->name[0] == '.' || s->name[0] == '$')))			return;		break;	case 'U':		if (gflag)			return;		break;	case 'Z':		if (!aflag)			return;		break;	case 'm':	case 'f':	/* we only see a 'z' when the following is true*/		if(!aflag || uflag || gflag)			return;		if (strcmp(s->name, ".frame"))			zenter(s);		break;	case 'a':	case 'p':	case 'z':	default:		if(!aflag || uflag || gflag)			return;		break;	}	symptr = realloc(symptr, (nsym+1)*sizeof(Sym*));	if (symptr == 0) {		error("out of memory");		exits("memory");	}	symptr[nsym++] = s;} voidprintsyms(Sym **symptr, long nsym){	Sym *s;	char *cp;	char path[512]; 	if(!sflag)		qsort(symptr, nsym, sizeof(*symptr), cmp);	while (nsym-- > 0) {		s = *symptr++;		if (multifile && !hflag)			Bprint(&bout, "%s:", filename);		if (s->type == 'z') {			fileelem(fnames, (uchar *) s->name, path, 512);			cp = path;		} else			cp = s->name;		if (Tflag)			Bprint(&bout, "%8ux ", s->sig);		if (s->value || s->type == 'a' || s->type == 'p')			Bprint(&bout, "%8llux %c %s\n", s->value, s->type, cp);		else			Bprint(&bout, "         %c %s\n", s->type, cp);	}} voiderror(char *fmt, ...){	char buf[4096], *s;	va_list arg; 	s = buf;	s += sprint(s, "%s: ", argv0);	va_start(arg, fmt);	s = vseprint(s, buf + sizeof(buf) / sizeof(*buf), fmt, arg);	va_end(arg);	*s++ = '\n';	write(2, buf, s - buf);	errs = "errors";}