harvey/sys/src/9/port/devregress.c

#include "u.h"
#include "../port/lib.h"
#include "mem.h"
#include "dat.h"
#include "fns.h"
#include "../port/error.h"

#include "ureg.h"

enum {
	Qdir = 0,
	Qctl = 1,
	Qmalloc,
	Qtsleep,
	Qmax,
};

static Dirtab regressdir[Qmax] = {
	".",		{ Qdir, 0, QTDIR },	0,	0555,
	"regressctl",	{ Qctl, 0 },	0,	0666,
	"malloc",	{ Qmalloc, 0 },	0,	0666,
	"tsleep",	{ Qtsleep, 0 },	0,	0666,
};

int verbose = 0;

static Chan*
regressattach(char* spec)
{
	return devattach('Z', spec);
}

Walkqid*
regresswalk(Chan* c, Chan *nc, char** name, int nname)
{
	return devwalk(c, nc, name, nname, regressdir, Qmax, devgen);
}

static int32_t
regressstat(Chan* c, uint8_t* dp, int32_t n)
{
	return devstat(c, dp, n, regressdir, Qmax, devgen);
}

static Chan*
regressopen(Chan* c, int omode)
{
	return devopen(c, omode, regressdir, Qmax, devgen);
}

static void
regressclose(Chan*unused)
{
}

static int32_t
regressread(Chan *c, void *a, int32_t n, int64_t offset)
{
	switch((uint32_t)c->qid.path){

	case Qdir:
		return devdirread(c, a, n, regressdir, Qmax, devgen);

	case Qmalloc:
	case Qtsleep:
		break;

	default:
		error(Eperm);
		break;
	}

	return n;
}

static int32_t
regresswrite(Chan *c, void *a, int32_t n, int64_t offset)
{
	Proc *up = externup();
	char *p;
	unsigned long amt;

	switch((uint32_t)c->qid.path){

	case Qmalloc:
		p = a;
		amt = strtoull(p, 0, 0);
		if (verbose)
			print("Malloc %d\n", amt);
		p = malloc(amt);
		if (verbose)
			print("Got %p\n", p);
		free(p);
		if (verbose)
			print("Freed %p\n", p);
		return n;

	case Qtsleep:
		p = a;
		amt = strtoull(p, 0, 0);
		if (verbose)
			print("tsleep %d\n", amt);
		tsleep(&up->sleep, return0, 0, amt);
		if (verbose)
			print("done tsleep\n");
		return n;

	case Qctl:
		p = a;
		if (*p == 'v'){
			if (verbose)
				verbose--;
		} else if (*p == 'V')
			verbose++;
		else
			error("Only v or V");
		return n;

	default:
		error(Eperm);
		break;
	}
	return 0;
}

Dev regressdevtab = {
	'Z',
	"regress",

	devreset,
	devinit,
	devshutdown,
	regressattach,
	regresswalk,
	regressstat,
	regressopen,
	devcreate,
	regressclose,
	regressread,
	devbread,
	regresswrite,
	devbwrite,
	devremove,
	devwstat,
};