ref: 2d2fabb2bb4472d840f896d086e1c1233e1302e1
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;
Channel *playerq;
void
freeplayer(Player *p)
{
close(p->sfd);
close(p->fd);
free(p);
}
int
isconnected(Player *p)
{
char buf[8];
int n;
n = pread(p->sfd, buf, sizeof buf, 0);
if(n < 0 || strncmp(buf, "Close", 5) == 0)
return 0;
return 1;
}
void
netrecvthread(void *arg)
{
Chanpipe *cp;
Ioproc *io;
char buf[256], *e;
int n, tot;
cp = arg;
io = ioproc();
tot = 0;
while((n = ioread(io, cp->fd, buf+tot, sizeof(buf)-1-tot)) > 0){
tot += n;
buf[tot] = 0;
while((e = strchr(buf, '\n')) != nil){
*e++ = 0;
chanprint(cp->c, "%s", buf);
tot -= e-buf;
memmove(buf, e, tot);
}
if(tot >= sizeof(buf)-1)
tot = 0;
}
if(debug)
fprint(2, "[%d] lost connection\n", getpid());
closeioproc(io);
chanclose(cp->c);
threadexits(nil);
}
void
battleproc(void *arg)
{
NetConnInfo *nci[2];
Match *m;
Player *p, *op;
Chanpipe cp[2];
Alt a[3];
int i;
uint n0;
char *s;
Point2 cell;
char *coords[5];
int j, orient;
m = arg;
s = nil;
nci[0] = getnetconninfo(nil, m->pl[0]->fd);
nci[1] = getnetconninfo(nil, m->pl[1]->fd);
if(nci[0] == nil || nci[1] == nil)
sysfatal("getnetconninfo: %r");
threadsetname("battleproc %s ↔ %s", nci[0]->raddr, nci[1]->raddr);
freenetconninfo(nci[0]);
freenetconninfo(nci[1]);
cp[0].c = chancreate(sizeof(char*), 1);
cp[0].fd = m->pl[0]->fd;
cp[1].c = chancreate(sizeof(char*), 1);
cp[1].fd = m->pl[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;
threadsetgrp(cp[0].fd);
threadcreate(netrecvthread, &cp[0], mainstacksize);
threadcreate(netrecvthread, &cp[1], mainstacksize);
for(i = 0; i < nelem(m->pl); i++)
if(!isconnected(m->pl[i])){
sendp(playerq, m->pl[i^1]);
freeplayer(m->pl[i]);
goto Finish;
}
write(m->pl[0]->fd, "id\n", 3);
write(m->pl[1]->fd, "id\n", 3);
while((i = alt(a)) >= 0){
p = m->pl[i];
op = m->pl[i^1];
if(a[i].err != nil){
if(debug)
fprint(2, "[%d] alt: %s\n", getpid(), a[i].err);
write(op->fd, "win\n", 4);
sendp(playerq, op);
freeplayer(p);
break;
}
if(debug)
fprint(2, "[%d] said '%s'\n", i, s);
switch(p->state){
case Waiting0:
if(strncmp(s, "id", 2) == 0){
snprint(p->name, sizeof p->name, "%s", strlen(s) > 3? s+3: "???");
write(p->fd, "layout\n", 7);
p->state = Outlaying;
if(op->state == Outlaying){
fprint(p->fd, "oid %s\n", op->name);
fprint(op->fd, "oid %s\n", p->name);
}
}
break;
case Outlaying:
if(strncmp(s, "layout", 6) == 0)
if(gettokens(s+7, coords, nelem(coords), ",") == nelem(coords)){
if(debug)
fprint(2, "rcvd layout from %d\n", i);
for(j = 0; j < nelem(coords); j++){
cell = coords2cell(coords[j]);
orient = coords[j][strlen(coords[j])-1] == 'h'? OH: OV;
/* TODO keep track of the ships and report back on the first shot and when sunk */
settiles(p, cell, orient, shiplen(j), Tship);
}
p->state = Waiting;
if(op->state == Waiting){
if(debug){
fprint(2, "map%d:\n", i);
fprintmap(2, p);
fprint(2, "map%d:\n", i^1);
fprintmap(2, op);
}
n0 = truerand();
if(debug)
fprint(2, "let the game begin: %d plays, %d waits\n", n0%2, (n0+1)%2);
write(m->pl[n0%2]->fd, "play\n", 5);
m->pl[n0%2]->state = Playing;
write(m->pl[(n0+1)%2]->fd, "wait\n", 5);
}
}
break;
case Playing:
if(strncmp(s, "shoot", 5) == 0){
cell = coords2cell(s+6);
switch(gettile(op, cell)){
case Tship:
settile(op, cell, Thit);
write(p->fd, "hit\n", 4);
fprint(op->fd, "hit %s\n", cell2coords(cell));
if(countshipcells(op) < 17){
write(p->fd, "win\n", 4);
write(op->fd, "lose\n", 5);
sendp(playerq, p);
sendp(playerq, op);
goto Finish;
}
goto Swapturn;
case Twater:
settile(op, cell, Tmiss);
write(p->fd, "miss\n", 5);
fprint(op->fd, "miss %s\n", cell2coords(cell));
Swapturn:
write(p->fd, "wait\n", 5);
write(op->fd, "play\n", 5);
p->state = Waiting;
op->state = Playing;
break;
}
if(debug)
fprint(2, "%d plays, %d waits\n", i^1, i);
}
break;
}
free(s);
}
Finish:
if(debug)
fprint(2, "[%d] battleproc ending\n", getpid());
free(m);
chanfree(cp[0].c);
chanfree(cp[1].c);
threadkillgrp(threadgetgrp());
threadexits(nil);
}
void
matchmaker(void *)
{
Match *m;
Player *pl[2];
threadsetname("matchmaker");
for(;;){
pl[0] = recvp(playerq);
pl[1] = recvp(playerq);
pl[0]->state = Waiting0;
pl[1]->state = Waiting0;
memset(pl[0]->map, Twater, MAPW*MAPH);
memset(pl[1]->map, Twater, MAPW*MAPH);
m = emalloc(sizeof *m);
m->pl[0] = pl[0];
m->pl[1] = pl[1];
proccreate(battleproc, m, mainstacksize);
}
}
void
dolisten(char *addr)
{
char adir[40], ldir[40], aux[128], *s;
int acfd, lcfd, dfd, sfd;
Player *p;
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;
p->state = Waiting0;
sendp(playerq, 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;
GEOMfmtinstall();
addr = "tcp!*!3047";
ARGBEGIN{
case 'd':
debug++;
break;
case 'a':
addr = EARGF(usage());
break;
default: usage();
}ARGEND
if(argc != 0)
usage();
playerq = chancreate(sizeof(Player*), 8);
proccreate(matchmaker, nil, mainstacksize);
dolisten(addr);
}