ref: 3a7a70853c38b79291760a4a0b923d1964139c90
dir: /com.c/
#include <u.h>
#include <libc.h>
#include <draw.h>
#include <fcall.h>
#include "dat.h"
#include "fns.h"
enum{
Hdrsz = 2,
};
typedef struct Header Header;
struct Header{
ushort size;
};
static int
vunpack(uchar *p, uchar *e, char *fmt, va_list a)
{
int n, sz;
sz = 0;
for(;;){
n = 0;
switch(*fmt++){
default: sysfatal("vunpack: unknown format %c", fmt[-1]);
error: werrstr("vunpack: truncated message"); return -1;
case 0: return sz;
case ' ': break;
case 'h': n = sizeof(u8int); if(p + n > e) goto error;
*va_arg(a, int*) = GBIT8(p); p += n;
break;
case 's': n = sizeof(u16int); if(p + n > e) goto error;
*va_arg(a, int*) = GBIT16(p); p += n;
break;
case 'd': n = sizeof(u32int); if(p + n > e) goto error;
*va_arg(a, int*) = GBIT32(p); p += n;
break;
case 'l': n = sizeof(u32int); if(p + n > e) goto error;
*va_arg(a, long*) = GBIT32(p); p += n;
break;
case 'v': n = sizeof(u64int); if(p + n > e) goto error;
*va_arg(a, vlong*) = GBIT64(p); p += n;
break;
}
sz += n;
}
}
static int
unpack(uchar *p, uchar *e, char *fmt, ...)
{
int n;
va_list a;
va_start(a, fmt);
n = vunpack(p, e, fmt, a);
va_end(a);
return n;
}
static Mobj *
mobjfromreq(Mobj *r)
{
Mobj *mo;
if((mo = derefmobj(r->idx, r->uuid)) == nil)
return nil;
if(!eqpt(mo->Point, r->Point)){
werrstr("phase error: req mobj at %P, found %M", r->Point, mo);
return nil;
}
return mo;
}
static int
reqgather(uchar *p, uchar *e)
{
int n;
Mobj reqm, reqt, *mo, *tgt;
if((n = unpack(p, e, "dldd dldd",
&reqm.idx, &reqm.uuid, &reqm.x, &reqm.y,
&reqt.idx, &reqt.uuid, &reqt.x, &reqt.y)) < 0)
return -1;
if((mo = mobjfromreq(&reqm)) == nil)
return -1;
if((mo->o->f & Fgather) == 0){
werrstr("reqgather: object %M not a gatherer", mo);
return -1;
}
if((mo->o->f & Fimmutable) || mo->o->speed == 0.0){
werrstr("reqgather: object %M can't move", mo);
return -1;
}
if((tgt = mobjfromreq(&reqt)) == nil)
return -1;
if((tgt->o->f & Fresource) == 0){
werrstr("reqgather: target %M not a resource", tgt);
return -1;
}
if(pushgathercommand(mo, tgt) < 0)
return -1;
return n;
}
static int
reqmovenear(uchar *p, uchar *e)
{
int n;
Point click;
Mobj reqm, reqt, *mo, *tgt;
if((n = unpack(p, e, "dldd dd dldd",
&reqm.idx, &reqm.uuid, &reqm.x, &reqm.y,
&click.x, &click.y,
&reqt.idx, &reqt.uuid, &reqt.x, &reqt.y)) < 0)
return -1;
if(eqpt(reqm.Point, reqt.Point) || eqpt(reqm.Point, click)){
dprint("reqmovenear: %P [%#ux,%ld] → %P [%#ux,%ld] (%P), not moving to itself\n", reqm.Point, reqm.idx, reqm.uuid, reqt.Point, reqt.idx, reqt.uuid, click);
return n;
}
if((mo = mobjfromreq(&reqm)) == nil)
return -1;
if((mo->o->f & Fimmutable) || mo->o->speed == 0.0){
werrstr("reqmovenear: object %M can't move", mo);
return -1;
}
if((tgt = mobjfromreq(&reqt)) == nil)
return -1;
if(click.x >= nodemapwidth || click.y >= nodemapheight){
werrstr("reqmovenear: invalid location %d,%d", click.x, click.y);
return -1;
}
if(pushmovecommand(click, mo, tgt) < 0)
return -1;
return n;
}
static int
reqmove(uchar *p, uchar *e)
{
int n;
Point tgt;
Mobj reqm, *mo;
if((n = unpack(p, e, "dldd dd",
&reqm.idx, &reqm.uuid, &reqm.x, &reqm.y,
&tgt.x, &tgt.y)) < 0)
return -1;
if(eqpt(reqm.Point, tgt)){
dprint("reqmove: %P [%#ux,%ld] → %P, not moving to itself\n", reqm.Point, reqm.idx, reqm.uuid, tgt);
return n;
}
if((mo = mobjfromreq(&reqm)) == nil)
return -1;
if((mo->o->f & Fimmutable) || mo->o->speed == 0.0){
werrstr("reqmove: object %M can't move", mo);
return -1;
}
if(tgt.x >= nodemapwidth || tgt.y >= nodemapheight){
werrstr("reqmove: invalid target %d,%d", tgt.x, tgt.y);
return -1;
}
if(pushmovecommand(tgt, mo, nil) < 0)
return -1;
return n;
}
static int
reqstop(uchar *p, uchar *e)
{
int n;
Mobj reqm, *mo;
if((n = unpack(p, e, "dldd",
&reqm.idx, &reqm.uuid, &reqm.x, &reqm.y)) < 0)
return -1;
if((mo = mobjfromreq(&reqm)) == nil)
return -1;
clearcommands(mo);
return n;
}
static int
reqpause(uchar *, uchar *)
{
pause ^= 1;
return 0;
}
static int
readhdr(Msg *m, Header *h)
{
if(unpack(m->buf, m->buf + m->sz, "s", &h->size) < 0
|| h->size <= 0
|| h->size != m->sz - Hdrsz){
werrstr("readhdr: malformed message");
return -1;
}
return 0;
}
int
parsemsg(Msg *m)
{
int n, type;
uchar *p, *e;
int (*fn)(uchar*, uchar*);
Header h;
if(readhdr(m, &h) < 0){
dprint("parsemsg: %r\n");
return -1;
}
p = m->buf + Hdrsz;
e = m->buf + m->sz;
while(p < e){
type = *p++;
switch(type){
case CTpause: fn = reqpause; break;
case CTstop: fn = reqstop; break;
case CTmove: fn = reqmove; break;
case CTmovenear: fn = reqmovenear; break;
case CTgather: fn = reqgather; break;
case CTeom:
if(p < e)
fprint(2, "parsemsg: trailing data\n");
return 0;
default: fprint(2, "parsemsg: invalid message type %ux\n", type); return -1;
}
if((n = fn(p, e)) < 0){
fprint(2, "parsemsg: %r\n");
return -1;
}else
p += n;
}
return 0;
}
static int
vpack(uchar *p, uchar *e, char *fmt, va_list a)
{
int n, sz;
uchar u[8];
u32int v;
u64int w;
sz = 0;
for(;;){
n = 0;
switch(*fmt++){
default: sysfatal("unknown format %c", fmt[-1]);
copy: if(p + n > e) sysfatal("vpack: buffer overflow");
memcpy(p, u, n); p += n; break;
case 0: return sz;
case ' ': break;
case 'h': v = va_arg(a, int); PBIT8(u, v); n = sizeof(u8int); goto copy;
case 's': v = va_arg(a, int); PBIT16(u, v); n = sizeof(u16int); goto copy;
case 'd': v = va_arg(a, int); PBIT32(u, v); n = sizeof(u32int); goto copy;
case 'l': v = va_arg(a, long); PBIT32(u, v); n = sizeof(u32int); goto copy;
case 'v': w = va_arg(a, vlong); PBIT64(u, w); n = sizeof(u64int); goto copy;
}
sz += n;
}
}
static void
newmsg(Msg *m)
{
m->sz += Hdrsz;
}
static int
pack(uchar *p, uchar *e, char *fmt, ...)
{
int n;
va_list a;
va_start(a, fmt);
n = vpack(p, e, fmt, a);
va_end(a);
return n;
}
static int
packmsg(Msg *m, char *fmt, ...)
{
int n;
va_list a;
if(m->sz == 0)
newmsg(m);
va_start(a, fmt);
n = vpack(m->buf + m->sz, m->buf + sizeof m->buf, fmt, a);
va_end(a);
if(n >= 0)
m->sz += n;
return n;
}
void
endmsg(Msg *m)
{
packmsg(m, "h", CTeom);
pack(m->buf, m->buf + Hdrsz, "s", m->sz - Hdrsz);
}
int
sendgather(Mobj *mo, Mobj *tgt)
{
Msg *m;
m = getclbuf();
if(packmsg(m, "h dldd dldd", CTgather,
mo->idx, mo->uuid, mo->x, mo->y,
tgt->idx, tgt->uuid, tgt->x, tgt->y) < 0){
fprint(2, "sendgather: %r\n");
return -1;
}
return 0;
}
int
sendmovenear(Mobj *mo, Point click, Mobj *tgt)
{
Msg *m;
m = getclbuf();
if(packmsg(m, "h dldd dd dldd", CTmovenear,
mo->idx, mo->uuid, mo->x, mo->y,
click.x, click.y,
tgt->idx, tgt->uuid, tgt->x, tgt->y) < 0){
fprint(2, "sendmovenear: %r\n");
return -1;
}
return 0;
}
int
sendmove(Mobj *mo, Point tgt)
{
Msg *m;
m = getclbuf();
if(packmsg(m, "h dldd dd", CTmove,
mo->idx, mo->uuid, mo->x, mo->y,
tgt.x, tgt.y) < 0){
fprint(2, "sendmove: %r\n");
return -1;
}
return 0;
}
int
sendstop(Mobj *mo)
{
Msg *m;
m = getclbuf();
if(packmsg(m, "h dldd", CTstop,
mo->idx, mo->uuid, mo->x, mo->y) < 0){
fprint(2, "sendstop: %r\n");
return -1;
}
return 0;
}
int
sendpause(void)
{
Msg *m;
m = getclbuf();
if(packmsg(m, "h", CTpause) < 0){
fprint(2, "sendpause: %r\n");
return -1;
}
return 0;
}