#include <u.h>
#include <libc.h>
#include <thread.h>

enum {
	Ndflt	= 1<<21,
	Mdflt	= 1,
	Odflt	= 10,
};

N	= Ndflt;
M	= Mdflt;
O	= Odflt;

Channel *c, *endc;

void
sendthread(void*)
{
	ulong i;

	for(i = 0; i < N; i++)
		sendul(c, i);
	threadexits("");
}

void
receivethread(void*)
{
	int i;

	for(i = 0; i <N*M; i++)
		recvul(c);
	sendul(endc, 1);
	threadexits("");
}

void
usage(void)
{
	fprint(2, "usage: chantest [-O nbuf] [-M nproc]\n");
	exits("usage");
}

void
threadmain(int argc, char **argv)
{
	int i;
	uvlong t;

	ARGBEGIN{
	default:
		usage();
	case 'M':
		M = atoi(EARGF(usage()));
		break;
	case 'O':
		O = atoi(EARGF(usage()));
		break;
	}ARGEND
	if(argc > 0)
		usage();

	t = -nsec();
	c = chancreate(sizeof(ulong), O);
	endc = chancreate(sizeof(ulong), 0);

	proccreate(receivethread, nil, 4*1024);
	for(i = 0; i < M; i++)
		proccreate(sendthread, (void*)(uintptr)i, 4*1024);

	recvul(endc);
	t += nsec();

	print("%lld\n", t);

	threadexits("");
}