#include	"auxi.h" #define RND(x, y)	((((x)+(y)-1)/(y))*(y)) char *cmd;int sflag, dflag; int ifd, ofd;Fhdr ihdr; long	HEADR, INITTEXT, INITDAT, INITRND, INITENTRY;long textsize, datsize, bsssize; int cout;int thumb; static void	get_file(char *);static void	put_file(char *);static void	usage(char *);static long	strxtol(char *);static void	readsyms(void); char	*fail	= "error"; voidmain(int argc, char	*argv[]){	char *a, *ifile, *ofile; 	cmd = argv[0]; 	INITTEXT = -1;	INITDAT = -1;	INITRND = -1;	INITENTRY = -1; 	ARGBEGIN {	/*	 * Options without args	 */	case 's':		sflag = 1;		break;	/*	 * Options with args	 */	case 'T':		a = ARGF();		if(a)			INITTEXT = strxtol(a);		break;	case 'D':		a = ARGF();		if(a)			INITDAT = strxtol(a);		break;	case 'R':		a = ARGF();		if(a)			INITRND = strxtol(a);		break;	case 'E':		a = ARGF();		if(a)			INITENTRY = strxtol(a);		break;	case 'd':		dflag |= strxtol(ARGF());		break;	default:		usage("Invalid option");	} ARGEND 	if (argc != 2)		usage("Wrong number of arguments"); 	ifile = argv[0];	ofile = argv[1]; 	get_file(ifile);	put_file(ofile);	exits(0);} char usagemsg[] ="usage: %s  options infile outfile\n\t options (for outfile): -H[1234] -s -T<text> -D<data> -R<rnd> -E<entry>\n"; static voidusage(char *msg){	fprint(2, "***Error: %s\n", msg);	fprint(2, usagemsg, cmd);	exits("usage");} static longstrxtol(char *s){	char *es;	int base = 0;	long r; 	if (*s == '0')		if (*++s == 'x'){			base = 16;			s++;		}		else			base = 8;	r = strtol(s, &es, base);	if (*es)		usage("bad number");	return(r);} static voidget_file(char *ifile){	int h; 	ifd = open(ifile, OREAD);	if (ifd < 0) {		fprint(2, "5coff: open %s: %r\n", ifile);		exits("open");	}	h = crackhdr(ifd, &ihdr);	if (!h || dflag){		fprint(2, "Crackhdr: %d, type: %d, name: %s\n", h, ihdr.type, ihdr.name);		fprint(2, "txt %llux, ent %llux, txtsz %lux, dataddr %llux\n",			ihdr.txtaddr, ihdr.entry, ihdr.txtsz, ihdr.dataddr);	}	if (!h)		usage("File type not recognized");	machbytype(ihdr.type);	if (dflag)		fprint(2, "name: <%s> pgsize:%ux\n", mach->name, mach->pgsize); 	HEADR = 22+28+3*48;	if(INITTEXT == -1)		INITTEXT = ihdr.txtaddr;	else		ihdr.txtaddr = INITTEXT;	if(INITDAT == -1)		INITDAT = ihdr.dataddr;	else		ihdr.dataddr = INITDAT;	if(INITENTRY == -1)		INITENTRY = ihdr.entry;	else		ihdr.entry = INITENTRY;	textsize = ihdr.txtsz;	datsize = ihdr.datsz;	bsssize = ihdr.bsssz;	if(INITRND > 0)		ihdr.dataddr = INITDAT = RND(INITTEXT+textsize, INITRND);	if(0){		INITTEXT = INITENTRY;		INITDAT = RND(INITTEXT+textsize, 4);	}	if(0){		print("H=%lux T=%lux D=%lux t=%lux d=%lux b=%lux e=%lux\n", HEADR, INITTEXT, INITDAT, textsize, datsize, bsssize, INITENTRY);		print("%llux %llux %llux %lux %lux %lux\n", ihdr.txtaddr, ihdr.dataddr, ihdr.entry, ihdr.txtsz, ihdr.datsz, ihdr.bsssz);	} 	readsyms();} #define WB	128#define WSAFE	(WB-4)char Wbuf[WB];char *wp = Wbuf; voidcflush(void){	if(wp > Wbuf)		write(ofd, Wbuf, wp-Wbuf);	wp = Wbuf;} voidlput(long l){	wp[0] = l>>24;	wp[1] = l>>16;	wp[2] = l>>8;	wp[3] = l;	wp += 4;	if(wp >= Wbuf+WSAFE)		cflush();} voidcput(int l){	wp[0] = l;	wp += 1;	if(wp >= Wbuf+WSAFE)		cflush();} voidhputl(int l){	wp[1] = l>>8;	wp[0] = l;	wp += 2;	if(wp >= Wbuf+WSAFE)		cflush();} voidlputl(long l){	wp[3] = l>>24;	wp[2] = l>>16;	wp[1] = l>>8;	wp[0] = l;	wp += 4;	if(wp >= Wbuf+WSAFE)		cflush();} static voidcopyseg(long sz){	char	buf[1024]; 	cflush();	while (sz > 0){		long n;		long r; 		n = sz;		if (n > sizeof buf)			n = sizeof buf;		sz -= n; 		if ((r = read(ifd, buf, n)) != n){			fprint(2, "%ld = read(...%ld) at %ld\n", r, n, (long)seek(ifd, 0, 1));			perror("Premature eof");			exits(fail);		}		if ((r = write(ofd, buf, n)) != n){			fprint(2, "%ld = write(...%ld)\n", r, n);			perror("Write error!");			exits(fail);		}	}	cflush();} static voidput_file(char *ofile){	ofd = create(ofile, OWRITE, 0666);	if (ofd < 0) {		fprint(2, "5coff: create %s: %r\n", ofile);		exits("create");	}	cout = ofd; 	/* TBS lput for Plan9 header before ? */ 	seek(ifd, ihdr.txtoff, 0);	seek(ofd, HEADR, 0);	copyseg(ihdr.txtsz); 	seek(ifd, ihdr.datoff, 0);	seek(ofd, HEADR+textsize, 0);	copyseg(ihdr.datsz); 	seek(ofd, HEADR+textsize+datsize, 0);	coffsym();	cflush();	cofflc();	cflush();		seek(ofd, 0, 0);	coffhdr();	cflush(); 	close(ifd);	close(ofd);} longentryvalue(void){	return INITENTRY;} voiddiag(char *s, ...){	fprint(2, "%s\n", s);	exits("error");} static voidreadsyms(void){	int i;	long n;	Sym *s; 	if(sflag)		return;	n = syminit(ifd, &ihdr);	beginsym();	for(i = 0; i < n; i++){		s = getsym(i);		newsym(i, s->name, s->value, s->type);	}	endsym();}