ref: d32578073676ce90769c378cbd5c61db22c1b49e
dir: /sys/src/cmd/nusb/joy/joy.c/
/*
* USB Human Interaction Device: game controller.
*
* Game controller events are written to stdout.
* May be used in conjunction with a script to
* translate to keyboard events and pipe to emulator
* (example /sys/src/games/nes/joynes).
*
*/
#include <u.h>
#include <libc.h>
#include <thread.h>
#include "usb.h"
#include "hid.h"
typedef struct KDev KDev;
struct KDev
{
Dev* dev; /* usb device*/
Dev* ep; /* endpoint to get events */
/* report descriptor */
int nrep;
uchar rep[512];
};
static void
kbfree(KDev *kd)
{
if(kd->ep != nil)
closedev(kd->ep);
if(kd->dev != nil)
closedev(kd->dev);
free(kd);
}
static void
kbfatal(KDev *kd, char *sts)
{
if(sts != nil)
fprint(2, "%s: fatal: %s\n", argv0, sts);
else
fprint(2, "%s: exiting\n", argv0);
kbfree(kd);
threadexits(sts);
}
static int debug, kbd;
static int
signext(int v, int bits)
{
int s;
s = sizeof(v)*8 - bits;
v <<= s;
v >>= s;
return v;
}
static int
getbits(uchar *p, uchar *e, int bits, int off)
{
int v, m;
p += off/8;
off %= 8;
v = 0;
m = 1;
if(p < e){
while(bits--){
if(*p & (1<<off))
v |= m;
if(++off == 8){
if(++p >= e)
break;
off = 0;
}
m <<= 1;
}
}
return v;
}
enum {
Ng = RepCnt+1,
UsgCnt = Delim+1, /* fake */
Nl = UsgCnt+1,
Nu = 256,
};
static uchar*
repparse1(uchar *d, uchar *e, int g[], int l[], int c,
void (*f)(int t, int v, int g[], int l[], int c, void *a), void *a)
{
int z, k, t, v, i;
while(d < e){
v = 0;
t = *d++;
z = t & 3, t >>= 2;
k = t & 3, t >>= 2;
switch(z){
case 3:
d += 4;
if(d > e) continue;
v = d[-4] | d[-3]<<8 | d[-2]<<16 | d[-1]<<24;
break;
case 2:
d += 2;
if(d > e) continue;
v = d[-2] | d[-1]<<8;
break;
case 1:
d++;
if(d > e) continue;
v = d[-1];
break;
}
switch(k){
case 0: /* main item*/
switch(t){
case Collection:
memset(l, 0, Nl*sizeof(l[0]));
d = repparse1(d, e, g, l, v, f, a);
continue;
case CollectionEnd:
return d;
case Input:
case Output:
case Feature:
if(l[UsgCnt] == 0 && l[UsagMin] != 0 && l[UsagMin] < l[UsagMax])
for(i=l[UsagMin]; i<=l[UsagMax] && l[UsgCnt] < Nu; i++)
l[Nl + l[UsgCnt]++] = i;
for(i=0; i<g[RepCnt]; i++){
if(i < l[UsgCnt])
l[Usage] = l[Nl + i];
(*f)(t, v, g, l, c, a);
}
break;
}
memset(l, 0, Nl*sizeof(l[0]));
continue;
case 1: /* global item */
if(t == Push){
int w[Ng];
memmove(w, g, sizeof(w));
d = repparse1(d, e, w, l, c, f, a);
} else if(t == Pop){
return d;
} else if(t < Ng){
if(t == RepId)
v &= 0xFF;
else if(t == UsagPg)
v &= 0xFFFF;
else if(t != RepSize && t != RepCnt){
v = signext(v, (z == 3) ? 32 : 8*z);
}
g[t] = v;
}
continue;
case 2: /* local item */
if(l[Delim] != 0)
continue;
if(t == Delim){
l[Delim] = 1;
} else if(t < Delim){
if(z != 3 && (t == Usage || t == UsagMin || t == UsagMax))
v = (v & 0xFFFF) | (g[UsagPg] << 16);
l[t] = v;
if(t == Usage && l[UsgCnt] < Nu)
l[Nl + l[UsgCnt]++] = v;
}
continue;
case 3: /* long item */
if(t == 15)
d += v & 0xFF;
continue;
}
}
return d;
}
/*
* parse the report descriptor and call f for every (Input, Output
* and Feature) main item as often as it would appear in the report
* data packet.
*/
static void
repparse(uchar *d, uchar *e,
void (*f)(int t, int v, int g[], int l[], int c, void *a), void *a)
{
int l[Nl+Nu], g[Ng];
memset(l, 0, sizeof(l));
memset(g, 0, sizeof(g));
repparse1(d, e, g, l, 0, f, a);
}
static int
setproto(KDev *f, Iface *iface)
{
int proto;
proto = Bootproto;
f->nrep = usbcmd(f->dev, Rd2h|Rstd|Riface, Rgetdesc, Dreport<<8, iface->id,
f->rep, sizeof(f->rep));
if(f->nrep > 0){
if(debug){
int i;
fprint(2, "report descriptor:");
for(i = 0; i < f->nrep; i++){
if(i%8 == 0)
fprint(2, "\n\t");
fprint(2, "%#2.2ux ", f->rep[i]);
}
fprint(2, "\n");
}
proto = Reportproto;
}else
kbfatal(f, "no report");
/*
* if a HID's subclass code is 1 (boot mode), it will support
* setproto, otherwise it is not guaranteed to.
*/
if(Subclass(iface->csp) != 1)
return 0;
return usbcmd(f->dev, Rh2d|Rclass|Riface, Setproto, proto, iface->id, nil, 0);
}
static void
sethipri(void)
{
char fn[64];
int fd;
snprint(fn, sizeof(fn), "/proc/%d/ctl", getpid());
fd = open(fn, OWRITE);
if(fd < 0)
return;
fprint(fd, "pri 13");
close(fd);
}
typedef struct Joy Joy;
struct Joy
{
int axes[Maxaxes];
int oldaxes[Maxaxes];
u64int btns;
int o;
uchar *e;
uchar p[128];
};
static void
joyparse(int t, int f, int g[], int l[], int, void *a)
{
int v, i;
Joy *p = a;
u64int m;
if(t != Input)
return;
if(g[RepId] != 0){
if(p->p[0] != g[RepId]){
p->o = 0;
return;
}
if(p->o < 8)
p->o = 8; /* skip report id byte */
}
v = getbits(p->p, p->e, g[RepSize], p->o);
if(g[LogiMin] < 0)
v = signext(v, g[RepSize]);
if((f & (Fvar|Farray)) == Fvar && v >= g[LogiMin] && v <= g[LogiMax]){
switch(l[Usage]){
case 0x010030:
case 0x010031:
case 0x010032:
i = l[Usage] - 0x010030;
if((f & (Fabs|Frel)) == Fabs)
p->axes[i] = v;
else
p->axes[i] += v;
break;
}
if((l[Usage] >> 16) == 0x09){
m = 1ULL << (l[Usage] & 0xff);
p->btns &= ~m;
if(v != 0)
p->btns |= m;
}
}
p->o += g[RepSize];
}
static int
kbdbusy(uchar* buf, int n)
{
int i;
for(i = 1; i < n; i++)
if(buf[i] == 0 || buf[i] != buf[0])
return 0;
return 1;
}
static void
joywork(void *a)
{
char err[ERRMAX];
int i, c, nerrs;
KDev* f = a;
Joy p;
u64int lastb;
threadsetname("joy %s", f->ep->dir);
sethipri();
memset(&p, 0, sizeof(p));
lastb = 0;
nerrs = 0;
for(;;){
if(f->ep == nil)
kbfatal(f, nil);
if(f->ep->maxpkt < 1 || f->ep->maxpkt > sizeof(p.p))
kbfatal(f, "joy: weird mouse maxpkt");
memset(p.p, 0, sizeof(p.p));
c = read(f->ep->dfd, p.p, f->ep->maxpkt);
if(c <= 0){
if(c < 0)
rerrstr(err, sizeof(err));
else
strcpy(err, "zero read");
if(++nerrs < 3){
fprint(2, "%s: joy: %s: read: %s\n", argv0, f->ep->dir, err);
continue;
}
kbfatal(f, err);
}
nerrs = 0;
p.o = 0;
p.e = p.p + c;
repparse(f->rep, f->rep+f->nrep, joyparse, &p);
for(i = 0; i < Maxaxes; i++){
if(p.axes[i] != p.oldaxes[i])
print("axis %d %d\n", i, p.axes[i]);
p.oldaxes[i] = p.axes[i];
}
for(i = 0; i < 64; i++)
if(((lastb ^ p.btns) & (1ULL<<i)) != 0)
print("%s %d\n", (p.btns & (1ULL<<i)) != 0 ? "down" : "up", i);
lastb = p.btns;
}
}
/* apply quirks for special devices */
static void
quirks(Dev *d)
{
int ret;
uchar buf[17];
/* sony dualshock 3 (ps3) controller requires special enable command */
if(d->usb->vid == 0x054c && d->usb->did == 0x0268){
ret = usbcmd(d, Rd2h|Rclass|Riface, Getreport, (0x3<<8) | 0xF2, 0, buf, sizeof(buf));
if(ret < 0)
sysfatal("failed to enable ps3 controller: %r");
}
}
static void
kbstart(Dev *d, Ep *ep, void (*f)(void*))
{
KDev *kd;
kd = emallocz(sizeof(KDev), 1);
incref(d);
kd->dev = d;
if(setproto(kd, ep->iface) < 0){
fprint(2, "%s: %s: setproto: %r\n", argv0, d->dir);
goto Err;
}
kd->ep = openep(kd->dev, ep);
if(kd->ep == nil){
fprint(2, "%s: %s: openep %d: %r\n", argv0, d->dir, ep->id);
goto Err;
}
if(opendevdata(kd->ep, OREAD) < 0){
fprint(2, "%s: %s: opendevdata: %r\n", argv0, kd->ep->dir);
goto Err;
}
quirks(kd->dev);
f(kd);
return;
Err:
kbfree(kd);
}
static void
usage(void)
{
fprint(2, "usage: %s [-d] devid\n", argv0);
threadexits("usage");
}
void
threadmain(int argc, char* argv[])
{
int i;
Dev *d;
Ep *ep;
Usbdev *ud;
ARGBEGIN{
case 'd':
debug++;
break;
default:
usage();
}ARGEND;
if(argc != 1)
usage();
d = getdev(*argv);
if(d == nil)
sysfatal("getdev: %r");
ud = d->usb;
ep = nil;
for(i = 0; i < nelem(ud->ep); i++){
if((ep = ud->ep[i]) == nil)
continue;
if(ep->type == Eintr && ep->dir == Ein && ep->iface->csp == JoyCSP)
break;
}
if(ep == nil)
sysfatal("no suitable endpoint found");
kbstart(d, ep, joywork);
threadexits(nil);
}