implement Itest; include "sys.m";	sys: Sys;include "string.m";	str: String;include "draw.m";include "daytime.m";	daytime: Daytime;include "bufio.m";	bufio: Bufio;	Iobuf: import bufio;include "readdir.m";	readdir: Readdir;include "arg.m";include "itslib.m";	S_INFO, S_WARN, S_ERROR, S_FATAL, S_STIME, S_ETIME: import Itslib;include "env.m";	env: Env;include "sh.m"; SUMFILE: con "summary";MSGFILE: con "msgs";README: con "README"; configfile := "";cflag := -1;verbosity := 3;repcount := 1;recroot := "";display_stderr := 0;display_stdout := 0;now := 0; stdout: ref Sys->FD;stderr: ref Sys->FD;context: ref Draw->Context; Test: adt {	spec: string;	fullspec: string;	cmd: Command;	recdir: string;	stdout: string;	stderr: string;	nruns: int;	nwarns: int;	nerrors: int;	nfatals: int;	failed: int;};  Itest: module{	init: fn(ctxt: ref Draw->Context, argv: list of string);};   init(ctxt: ref Draw->Context, args: list of string){	sys = load Sys Sys->PATH;	stdout = sys->fildes(1);	stderr = sys->fildes(2);	context = ctxt;	arg := load Arg Arg->PATH;	if(arg == nil)		nomod(Arg->PATH);	daytime = load Daytime Daytime->PATH;	if(daytime == nil)		nomod(Daytime->PATH);	str = load String String->PATH;	bufio = load Bufio Bufio->PATH;	if(bufio == nil)		nomod(Bufio->PATH);	if(str == nil)		nomod(String->PATH);	readdir = load Readdir Readdir->PATH;	if(readdir == nil)		nomod(Readdir->PATH);	env = load Env Env->PATH;	if(env == nil)		nomod(Env->PATH);	arg->init(args);	while((o := arg->opt()) != 0)		case o {		'c' =>	cflag = toint("c", arg->arg(), 0, 9);		'e' =>	display_stderr++;		'o' =>	display_stdout++;		'r' =>		repcount = toint("r", arg->arg(), 0, -1);		'v' =>	verbosity = toint("v", arg->arg(), 0, 9);		'C' =>	configfile = arg->arg();		'R' =>	recroot = arg->arg();		* =>		usage();		}	args = arg->argv();	arg = nil;	testlist : array of ref Test;	if (args != nil)		testlist = arg_tests(args);	else if (configfile != "")		testlist = config_tests(configfile);	if (testlist == nil)		fatal("No tests to run");	sys->pctl(Sys->FORKENV, nil);	if (env->setenv(Itslib->ENV_VERBOSITY, string verbosity))		fatal("Failed to set environment variable " + Itslib->ENV_VERBOSITY);	if (repcount) 		reps := string repcount;	else		reps = "infinite";	if (len testlist == 1) ts := "";	else ts = "s";	if (repcount == 1) rs := "";	else rs = "s";	mreport(0, S_INFO, 2, sys->sprint("Starting tests - %s run%s of %d test%s", reps, rs, len testlist, ts));	run := big 1; 	if (recroot != nil) 		recn := highest(recroot) + 1;	while (repcount == 0 || run <= big repcount) {		mreport(1, S_INFO, 3, sys->sprint("Starting run %bd", run));		for (i:=0; i<len testlist; i++) {			t := testlist[i];			if (recroot != nil) {				t.recdir = sys->sprint("%s/%d", recroot, recn++);				mreport(2, S_INFO, 3, sys->sprint("Recording in %s", t.recdir));				rfd := sys->create(t.recdir, Sys->OREAD, Sys->DMDIR | 8r770);				if (rfd == nil)					fatal(sys->sprint("Failed to create directory %s: %r\n", t.recdir));				rfd = nil;			}			runtest(t);		}		mreport(1, S_INFO, 3, sys->sprint("Finished run %bd", run));		run++;	}	mreport(0, S_INFO, 2, "Finished tests");} usage(){	sys->fprint(stderr, "Usage itest [-eo] [-c cflag] [-r count] [-v vlevel] [-C cfile] [-R recroot] [testdir ...]\n");	raise "fail: usage";} fatal(s: string){	sys->fprint(stderr, "%s\n", s);	raise "fail: error";} nomod(mod: string){	sys->fprint(stderr, "Failed to load %s\n", mod);	raise "fail: module";} toint(opt, s: string, min, max: int): int{	if (len s == 0 || str->take(s, "[0-9]+-") != s)		fatal(sys->sprint("no value specified for option %s", opt));	v := int s;	if (v < min)		fatal(sys->sprint("option %s value is less than minimum of %d: %d", opt, v, min));	if (max != -1 && v > max)		fatal(sys->sprint("option %s value is greater than maximum of %d: %d", opt, v, max));	return v;} arg_tests(args: list of string): array of ref Test{	al := len args;	ta := array[al] of ref Test;	for (i:=0; i<al; i++) {		tspec := hd args;		args = tl args;		ta[i] = ref Test(tspec, "", nil, "", "", "", 0, 0, 0, 0, 0);		tcheck(ta[i]);	}	return ta;} config_tests(cf: string): array of ref Test{	cl := linelist(cf);	if (cl == nil)		fatal("No tests in config file");	al := len cl;	ta := array[al] of ref Test;	for (i:=0; i<al; i++) {		tspec := hd cl;		cl = tl cl;		ta[i] = ref Test(tspec, "", nil, "", "", "", 0, 0, 0, 0, 0);		tcheck(ta[i]);	}	return ta; } highest(path: string): int{	(da, nd) := readdir->init(path, Readdir->NAME);	high := 0;	for (i:=0; i<nd; i++) {		n := int da[i].name;		if (n > high)			high = n;		}	return high;} tcheck(t: ref Test): int{	td := t.spec;	if (!checkdir(td)) {		fatal(sys->sprint("Failed to find test %s\n", td));		return 0;	}	tf1 := t.spec + "/t.sh";	tf2 := t.spec + "/t.dis";	if (checkexec(tf1)) {		t.fullspec = tf1;		return 1;	}	if (checkexec(tf2)) {		t.fullspec = tf2;		return 1;	}	fatal(sys->sprint("Could not find executable files %s or %s\n", tf1, tf2));	return 0;} checkdir(d: string): int{	(ok, dir) := sys->stat(d);	if (ok != 0 || ! dir.qid.qtype & Sys->QTDIR)		return 0;	return 1;} checkexec(d: string): int{	(ok, dir) := sys->stat(d);	if (ok != 0 || ! dir.mode & 8r100)		return 0;	return 1;}  set_cflag(f: int){	wfile("/dev/jit", string f, 0); } runtest(t: ref Test){	if (t.failed)		return; 	if (cflag != -1) {		mreport(0, S_INFO, 7, sys->sprint("Setting cflag to %d", cflag));		set_cflag(cflag);	}	readme := t.spec + "/" + README;	mreport(2, S_INFO, 3, sys->sprint("Starting test %s cflag=%s", t.spec, rfile("/dev/jit")));	if (verbosity > 8)		display_file(readme);	sync := chan of int;	spawn monitor(t, sync);	<-sync;} monitor(t: ref Test, sync: chan of int){	pid := sys->pctl(Sys->FORKFD|Sys->FORKNS|Sys->FORKENV|Sys->NEWPGRP, nil);	pa := array[2] of ref Sys->FD;	if (sys->pipe(pa))		fatal("Failed to set up pipe");	if (env->setenv(Itslib->ENV_MFD, string pa[0].fd))		fatal("Failed to set environment variable " + Itslib->ENV_MFD);	mlfd: ref Sys->FD;	if (t.recdir != nil) {		mfile := t.recdir+"/"+MSGFILE;		mlfd = sys->create(mfile, Sys->OWRITE, 8r660);		if (mlfd == nil)			fatal(sys->sprint("Failed to create %s: %r'\n", mfile));		t.stdout = t.recdir+"/stdout";		t.stderr = t.recdir+"/stderr";	} else {		t.stdout = "/tmp/itest.stdout";		t.stderr = "/tmp/itest.stderr";	}	cf := int rfile("/dev/jit");	stime := sys->millisec();	swhen := daytime->now();	etime := -1;	rsync := chan of int;	spawn runit(t.fullspec, t.stdout, t.stderr, t.spec, pa[0], rsync);	<-rsync;	pa[0] = nil;	(nwarns, nerrors, nfatals) := (0, 0, 0);	while (1) {		mbuf := array[Sys->ATOMICIO] of byte;		n := sys->read(pa[1], mbuf, len mbuf);		if (n <= 0) break;		msg := string mbuf[:n];		sev := int msg[0:1];		verb := int msg[1:2];		body := msg[2:];		if (sev == S_STIME)			stime = int body;		else if (sev == S_ETIME)			etime = int body;		else {			if (sev == S_WARN) {				nwarns++;				t.nwarns++;			}			else if (sev == S_ERROR) {				nerrors++;				t.nerrors++;			}			else if (sev == S_FATAL) {				nfatals++;				t.nfatals++;			}			mreport(3, sev, verb, sys->sprint("%s: %s", severs(sev), body));		}		if (mlfd != nil)			sys->fprint(mlfd, "%d:%s", now, msg);	}	if (etime < 0) {		etime = sys->millisec();		if (mlfd != nil)			sys->fprint(mlfd, "%d:%s", now, sys->sprint("%d0%d\n", S_ETIME, etime));	}	elapsed := etime-stime;	errsum := sys->sprint("WRN:%d ERR:%d FTL:%d", nwarns, nerrors, nfatals);	mreport(2, S_INFO, 3, sys->sprint("Finished test %s after %dms - %s", t.spec, elapsed, errsum));	if (t.recdir != "") {		wfile(t.recdir+"/"+SUMFILE, sys->sprint("%d %d %d %s\n", swhen, elapsed, cf, t.fullspec), 1);	}	if (display_stdout) {		mreport(2, 0, 0, "Stdout from test:");		display_file(t.stdout);	}	if (display_stderr) {		mreport(2, 0, 0, "Stderr from test:");		display_file(t.stderr);	}	sync <-= pid;} runit(fullspec, sofile, sefile, tpath: string, mfd: ref Sys->FD, sync: chan of int){	pid := sys->pctl(Sys->NEWFD|Sys->FORKNS, mfd.fd::nil);	o, e: ref Sys->FD;	o = sys->create(sofile, Sys->OWRITE, 8r660);	if (o == nil)		treport(mfd, S_ERROR, 0, "Failed to open stdout: %r\n");	else		sys->dup(o.fd, 1);	o = nil;	e = sys->create(sefile, Sys->OWRITE, 8r660);	if (e == nil)		treport(mfd, S_ERROR, 0, "Failed to open stderr: %r\n");	else		sys->dup(e.fd, 2);	e = nil;	sync <-= pid;	args := list of {fullspec};	if (fullspec[len fullspec-1] == 's')		cmd := load Command fullspec;	else {		cmd = load Command "/dis/sh.dis";		args = fullspec :: args;	}	if (cmd == nil) {		treport(mfd, S_FATAL, 0, sys->sprint("Failed to load Command from %s", "/dis/sh.dis"));		return;	}	if (sys->chdir(tpath))		treport(mfd, S_FATAL, 0, "Failed to cd to " + tpath);	{		cmd->init(context, args);	} exception ex {		"*" =>		treport(mfd, S_FATAL, 0, sys->sprint("Exception %s in test %s", ex, fullspec));	}			} severs(sevs: int): string{	SEVMAP :=  array[] of {"INF", "WRN", "ERR", "FTL"};	if (sevs >= len SEVMAP)		sstr := "UNK";	else		sstr = SEVMAP[sevs];	return sstr;}  rfile(file: string): string{	fd := sys->open(file, Sys->OREAD);	if (fd == nil) return nil;	buf := array[Sys->ATOMICIO] of byte;	n := sys->read(fd, buf, len buf);	return string buf[:n];}  wfile(file: string, text: string, create: int): int{	if (create)		fd := sys->create(file, Sys->OWRITE, 8r660);	else 		fd = sys->open(file, Sys->OWRITE);	if (fd == nil) {		sys->fprint(stderr, "Failed to open %s: %r\n", file);		return 0;	}	a := array of byte text;	al := len a;	if (sys->write(fd, a, al) != al) {		sys->fprint(stderr, "Failed to write to %s: %r\n", file);		return 0;	}	fd = nil;	return 1;} linelist(file: string): list of string{	bf := bufio->open(file, Bufio->OREAD);	if (bf == nil)		return nil;	cl : list of string;	while ((line := bf.gets('\n')) != nil) {		if (line[len line -1] == '\n')			line = line[:len line - 1];		cl = line :: cl;	}	bf = nil;	return cl;} display_file(file: string){	bf := bufio->open(file, Bufio->OREAD);	if (bf == nil)		return;	while ((line := bf.gets('\n')) != nil) {		sys->print("                    %s", line);	}} mreport(indent: int, sev: int, verb: int, msg: string){	now = daytime->now();	tm := daytime->local(now);	time := sys->sprint("%4d%02d%02d %02d:%02d:%02d", tm.year+1900, tm.mon-1, tm.mday, tm.hour, tm.min, tm.sec);	pad := "---"[:indent];	term := "";	if (len msg && msg[len msg-1] != '\n')		term = "\n";	if (sev || verb <= verbosity)		sys->print("%s %s%s%s", time, pad, msg, term);}  treport(mfd: ref Sys->FD, sev: int, verb: int, msg: string){	sys->fprint(mfd, "%d%d%s\n", sev, verb, msg);}