#include	<u.h>#include	<libc.h> #define	nelem(x)	(sizeof(x)/sizeof((x)[0])) typedef	struct	Rs	Rs; enum{	BSIZE	= 24*1024,	/* multiple of SECTOR */	SLEEP	= 100,	SECTOR	= 2048,}; int	TARGET	= -1;		/* scsi target of rom writer */int	MEMSIZE	= 8*1024;	/* in kilobytes */int	NPROC	= 3;		/* number of read-ahead processes */int	NBUF;			/* (MEMSIZE*1024)/BSIZE/NPROC */ struct	Rs{	uchar*	buf;	int	nin;		/* whole buffers read */	int	nout;		/* whole buffers consumed */	int	done;		/* reader is done */	int	last;		/* partial bytes of last buffer */}; int	scsifc;int	scsifd;int	filefd;int	error;Rs	rs[50]; void	reader(char*, int);void	checkerror(void);void	puts(char*);void	putn(int);void	scsi(uchar*, int, uchar*, int); voidmain(int argc, char *argv[]){	char *ifile, *ofile;	int i, first;	long ob, t, total;	uchar cmd[10], cmdwt[10], cmdwr[10];	char scsiname[50];	Dir dir;	Rs *r; 	ifile = "cd-rom";	ofile = 0;	ARGBEGIN {	default:		fprint(2, "unknown option: %c\n", ARGC());		break;	case 't':		TARGET = atoi(ARGF());		break;	case 'm':		/* in meg if < 100, in kb if >= 100 */		MEMSIZE = atoi(ARGF());		if(MEMSIZE < 100)			MEMSIZE *= 1024;		break;	case 'n':		NPROC = atoi(ARGF());		break;	case 'o':		ofile = ARGF();		break;	} ARGEND	if(argc > 0)		ifile = argv[0];	if(NPROC >= nelem(rs)) {		fprint(2, "too many processes\n");		exits("-n");	}	NBUF = (MEMSIZE*1024)/BSIZE/NPROC;	for(r=rs; r<rs+NPROC; r++) {		r->buf = malloc(NBUF*BSIZE);		if(r->buf == 0) {			fprint(2, "malloc failed\n");			exits("-m");		}		r->nin = 0;		r->nout = 0;		r->done = 0;	} 	if(dirstat(ifile, &dir) < 0) {		fprint(2, "cant stat %s %r\n", ifile);		exits("stat");	}  	if(TARGET >= 0) {		sprint(scsiname, "#S/%d/cmd", TARGET);		scsifc = open(scsiname, ORDWR);		sprint(scsiname, "#S/%d/data", TARGET);		scsifd = open(scsiname, ORDWR);		if(scsifc < 0 || scsifd < 0) {			fprint(2, "cant open scsi chan %r\n");			exits("open");		} 		/*		 * reserve track		 */		memset(cmd, 0, sizeof(cmd));		cmd[0] = 0xe4;		t = (dir.length + SECTOR - 1) / SECTOR;		cmd[5] = t>>24;		cmd[6] = t>>16;		cmd[7] = t>>8;		cmd[8] = t>>0;		scsi(cmd, 10, cmd, 0); 		/*		 * write track		 */		memset(cmdwt, 0, sizeof(cmdwt));		cmdwt[0] = 0xe6;		cmdwt[5] = 1;		cmdwt[6] = 1;		cmdwt[7] = (BSIZE/SECTOR)>>8;		cmdwt[8] = (BSIZE/SECTOR)>>0; 		/*		 * write		 */		memset(cmdwr, 0, sizeof(cmdwr));		cmdwr[0] = 0x2a;		cmdwr[7] = (BSIZE/SECTOR)>>8;		cmdwr[8] = (BSIZE/SECTOR)>>0;	} else	if(ofile) {		filefd = create(ofile, OWRITE, 0666);		if(filefd < 0) {			fprint(2, "cant create %s %r\n", ofile);			exits("create");		}	} else		filefd = 1; 	/*	 * start all the reading processes	 */	error = 0;	for(i=0; i<NPROC; i++)		if(rfork(RFPROC|RFNOWAIT|RFFDG|RFMEM) == 0)			reader(ifile, i); 	/*	 * wait until we have full buffers	 */	for(;;) {		checkerror();		for(r=rs; r<rs+NPROC; r++)			if(!r->done && r->nin<NBUF)				goto wfull;		break;	wfull:		sleep(SLEEP);	} 	/*	 * copy buffers to output	 */	t = time(0);	total = 0;	r = rs;	ob = 0;	first = 1;	for(;;) {		checkerror();		if(r->nin - r->nout > 0) {			if(TARGET >= 0) {				if(first) {					scsi(cmdwt, 10, r->buf+ob, BSIZE);					first = 0;				} else {					scsi(cmdwr, 10, r->buf+ob, BSIZE);				}			} else {				i = write(filefd, r->buf+ob, BSIZE);				if(i != BSIZE) {					fprint(2, "write %d %d\n", r->last, i);					exits("write");				}			}			total += BSIZE;			r->nout++; 			r++;			if(r >= rs+NPROC) {				r = rs;				ob += BSIZE;				if(ob >= NBUF*BSIZE)					ob = 0;			}			continue;		}		if(r->done) {			if(r->last) {				if(TARGET >= 0) {					i = (r->last + SECTOR - 1) / SECTOR;					if(first) {						cmdwt[7] = i>>8;						cmdwt[8] = i>>0;						scsi(cmdwt, 10, r->buf+ob, i*SECTOR);					} else {						cmdwr[7] = i>>8;						cmdwr[8] = i>>0;						scsi(cmdwr, 10, r->buf+ob, i*SECTOR);					}				} else {					i = write(filefd, r->buf+ob, r->last);					if(i != r->last) {						fprint(2, "write %d %d\n", r->last, i);						exits("write");					}				}			}			total += r->last;			break;		}		sleep(SLEEP);		continue;	} 	if(TARGET >= 0) {		/*		 * flush cache		 */		memset(cmd, 0, sizeof(cmd));		cmd[0] = 0x35;		scsi(cmd, 10, cmd, 0);	} 	t = time(0)-t;	if(t <= 0)		t = 1;	fprint(2, "rate = %ld (%ld/%ld)\n", total/t, total, t); 	if(TARGET >= 0) {		/*		 * fixate		 */		memset(cmd, 0, sizeof(cmd));		cmd[0] = 0xe9;		cmd[8] = 1;		scsi(cmd, 10, cmd, 0);	} 	fprint(2, "done\n"); out:	error = 1;	exits(0);} voidcheckerror(void){	if(error) {		puts("error set: exiting\n");		exits(0);	}} voidputs(char *s){	write(2, s, strlen(s));} voidputn(int n){	char x[10]; 	strcpy(x, "x: ");	x[0] = n+'0';	puts(x);} voidreader(char *file, int pn){	Rs *r;	int f, n;	long of, ob; 	f = open(file, OREAD);	if(f < 0) {		putn(pn);		puts("cant open ");		puts(file);		puts("\n");		error = 1;		exits(0);	}	r = rs+pn;	of = pn*BSIZE;	ob = 0;	for(;;) {		checkerror();		if(r->nin - r->nout >= NBUF) {			sleep(SLEEP);			continue;		}		seek(f, of, 0);		n = read(f, r->buf+ob, BSIZE);		if(n == BSIZE) {			r->nin++;			of += NPROC*BSIZE;			ob += BSIZE;			if(ob >= NBUF*BSIZE)				ob = 0;			continue;		}		if(n < 0) {			putn(pn);			puts("io error reading\n");			error = 1;			exits(0);		}		r->last = n;		r->done = 1;		exits(0);	}} voidscsi(uchar *cmd, int ccount, uchar *data, int dcount){	uchar resp[6], sense[255];	long status; 	if(write(scsifc, cmd, ccount) != ccount) {		fprint(2, "scsi write ctl %r\n");		exits("wctl");	}	if(write(scsifd, data, dcount) != dcount) {		fprint(2, "scsi write data %r\n");		exits("wdata");	}	if(read(scsifc, resp, 4) != 4) {		fprint(2, "scsi read ctl %r\n");		exits("rctl");	}	status = (resp[0]<<24) | (resp[1]<<16) | (resp[2]<<8) | (resp[3]<<0);	if(status == 0x6000)		return;	if(status != 0x6002) {		fprint(2, "bad status %.4x\n", status);		exits("status");	} 	/*	 * request sense	 */	memset(resp, 0, sizeof(resp));	resp[0] = 0x03;	resp[4] = sizeof(sense); 	if(write(scsifc, resp, sizeof(resp)) != sizeof(resp)) {		fprint(2, "scsi write ctl1 %r\n");		exits("wctl");	}	if(read(scsifd, sense, sizeof(sense)) <= 0) {		fprint(2, "scsi read data %r\n");		exits("rdata");	}	if(read(scsifc, resp, 4) != 4) {		fprint(2, "scsi read ctl %r\n");		exits("rctl");	}	fprint(2, "scsi reqsense: key #%2.2ux code #%2.2ux #%2.2ux\n",		sense[2], sense[12], sense[13]);	if((sense[2] & 0x0F) == 0x00 || (sense[2] & 0x0F) == 0x06)		return;	fprint(2, "exiting\n");	exits("scsi");}