shithub: battleship

ref: 04c90470556a5d3a66fb592ff7b6fee1228be95e
dir: /btsd.c/

#include <u.h>
#include <libc.h>
#include <thread.h>
#include <draw.h>
#include <mouse.h>
#include <keyboard.h>
#include <geometry.h>
#include "dat.h"
#include "fns.h"

int debug;
int mainstacksize = 16*KB;

Playerq playerq;


void
pushplayer(Player *p)
{
	qlock(&playerq);
	if(++playerq.nplayers > playerq.cap){
		playerq.cap = playerq.nplayers;
		playerq.players = erealloc(playerq.players, playerq.cap * sizeof p);
	}
	playerq.players[playerq.nplayers-1] = p;
	qunlock(&playerq);
}

Player *
popplayer(void)
{
	Player *p;

	p = nil;
	if(playerq.nplayers > 0){
		qlock(&playerq);
		p = playerq.players[--playerq.nplayers];
		qunlock(&playerq);
	}
	return p;
}

void
netrecvthread(void *arg)
{
	Chanpipe *cp;
	Ioproc *io;
	char buf[256];
	int n;

	cp = arg;
	io = ioproc();

	while((n = ioread(io, cp->fd, buf, sizeof(buf)-1)) > 0){
		buf[n] = 0;
		chanprint(cp->c, "%s", buf);
	}
	chanclose(cp->c);
	if(debug)
		fprint(2, "[%d] %d lost connection\n", threadpid(threadid()), threadid());
}

void
serveproc(void *arg)
{
	NetConnInfo *nci[2];
	Player **m;
	Chanpipe *cp;
	Alt a[3];
	int i, n0;
	char *s;

	threadsetname("serveproc ");

	m = arg;
	s = nil;

	nci[0] = getnetconninfo(nil, m[0]->fd);
	nci[1] = getnetconninfo(nil, m[1]->fd);
	if(nci[0] == nil || nci[1] == nil)
		sysfatal("getnetconninfo: %r");
	threadsetname("serveproc %s ↔ %s", nci[0]->raddr, nci[1]->raddr);
	freenetconninfo(nci[0]);
	freenetconninfo(nci[1]);

	cp = emalloc(2*sizeof(Chanpipe));
	cp[0].c = chancreate(sizeof(char*), 1);
	cp[0].fd = m[0]->fd;
	cp[1].c = chancreate(sizeof(char*), 1);
	cp[1].fd = m[1]->fd;

	a[0].c = cp[0].c; a[0].v = &s; a[0].op = CHANRCV;
	a[1].c = cp[1].c; a[1].v = &s; a[1].op = CHANRCV;
	a[2].op = CHANEND;

	threadcreate(netrecvthread, &cp[0], mainstacksize);
	threadcreate(netrecvthread, &cp[1], mainstacksize);

	n0 = truerand();
	write(m[n0%2]->fd, "wait", 4);
	write(m[(n0+1)%2]->fd, "play", 4);

	while((i = alt(a)) >= 0){
		if(debug)
			fprint(2, "[%d] %d said '%s'\n", threadpid(threadid()), threadid(), s);
		if(a[i].err != nil){
			write(m[i^1]->fd, "won", 3);
			/* TODO free the player */
			pushplayer(m[i^1]);
			goto out;
		}
		if(write(m[i^1]->fd, s, strlen(s)) != strlen(s)){
			write(m[i]->fd, "won", 3);
			/* TODO free the player */
			pushplayer(m[i]);
			goto out;
		}
		free(s);
	}
out:
	if(debug)
		fprint(2, "[%d] serveproc ending\n", threadpid(threadid()));
	chanclose(cp[0].c);
	chanclose(cp[1].c);
	/* TODO make sure this is the last thread to exit */
//	recv(cp[0].done)
//	recv(cp[1].done)
	free(m);
}

void
reaper(void *)
{
	Ioproc *io;
	char buf[8];
	ulong i;
	int n;

	threadsetname("reaper");

	io = ioproc();

	for(;;){
		for(i = 0; i < playerq.nplayers; i++){
			n = read(playerq.players[i]->sfd, buf, sizeof buf);
			if(n < 0 || strncmp(buf, "Closed", 6) == 0){
				qlock(&playerq);
				close(playerq.players[i]->fd);
				close(playerq.players[i]->sfd);
				free(playerq.players[i]);
				memmove(&playerq.players[i], &playerq.players[i+1], (--playerq.nplayers-i)*sizeof(Player*));
				qunlock(&playerq);
			}
		}
		iosleep(io, HZ2MS(3));
	}
}

void
matchmaker(void *)
{
	Player **match;

	threadsetname("matchmaker");

	for(;;){
		if(playerq.nplayers < 2){
			sleep(100);
			continue;
		}

		match = emalloc(2*sizeof(Player*));
		match[0] = popplayer();
		match[1] = popplayer();
		match[1]->o = match[0];
		match[0]->o = match[1];

		proccreate(serveproc, match, mainstacksize);
	}
}

void
listenthread(void *arg)
{
	char *addr, adir[40], ldir[40], aux[128], *s;
	int acfd, lcfd, dfd, sfd;
	Player *p;

	addr = arg;

	acfd = announce(addr, adir);
	if(acfd < 0)
		sysfatal("announce: %r");

	if(debug)
		fprint(2, "listening on %s\n", addr);

	while((lcfd = listen(adir, ldir)) >= 0){
		if((dfd = accept(lcfd, ldir)) >= 0){
			fd2path(dfd, aux, sizeof aux);
			s = strrchr(aux, '/');
			*s = 0;
			snprint(aux, sizeof aux, "%s/status", aux);
			sfd = open(aux, OREAD);
			if(sfd < 0)
				sysfatal("open: %r");

			p = emalloc(sizeof *p);
			p->fd = dfd;
			p->sfd = sfd;
			pushplayer(p);
		}
		close(lcfd);
	}

	threadexitsall("listen: %r");
}

void
usage(void)
{
	fprint(2, "usage: %s [-d] [-a addr]\n", argv0);
	threadexitsall("usage");
}

void
threadmain(int argc, char *argv[])
{
	char *addr;

	addr = "tcp!*!3047";
	ARGBEGIN{
	case 'd':
		debug++;
		break;
	case 'a':
		addr = EARGF(usage());
		break;
	}ARGEND
	if(argc != 0)
		usage();

	threadcreate(listenthread, addr, mainstacksize);
	proccreate(matchmaker, nil, mainstacksize);
	proccreate(reaper, nil, mainstacksize);
	yield();
}