nusb: add ethernet

This commit is contained in:
cinap_lenrek 2011-08-26 15:37:47 +02:00
parent 78ef115307
commit 6074cf1319
4 changed files with 904 additions and 1 deletions

View file

@ -5,6 +5,8 @@ if(! test -e '#σ/usb/usbevent')
if(! bind -a '#u' /dev) if(! bind -a '#u' /dev)
exit exit
mkdir -p -m 700 '#σc/usbnet'
@{ @{
rfork ne rfork ne
fn attach { fn attach {
@ -14,6 +16,7 @@ if(! bind -a '#u' /dev)
case *02 case *02
# serial and ethernet # serial and ethernet
nusb/serial $1 nusb/serial $1
nusb/ether $1
case *03 case *03
# handled /sys/src/9/boot/nusbrc # handled /sys/src/9/boot/nusbrc
# nusb/kb $1 # nusb/kb $1
@ -25,9 +28,13 @@ if(! bind -a '#u' /dev)
} }
} }
fn detach { fn detach {
# handled /sys/src/9/boot/nusbrc switch($4){
case *02
rm -f '#σ/usbnet/'^$1.*
}
} }
rc < '#σ/usb/usbevent' & rc < '#σ/usb/usbevent' &
} }
bind -a '#σ/usb' /dev bind -a '#σ/usb' /dev
bind -a '#σ/usbnet' /net

View file

@ -0,0 +1,883 @@
#include <u.h>
#include <libc.h>
#include <thread.h>
#include <bio.h>
#include <auth.h>
#include <fcall.h>
#include <9p.h>
#include "usb.h"
typedef struct Tab Tab;
typedef struct Qbuf Qbuf;
typedef struct Dq Dq;
typedef struct Conn Conn;
typedef struct Ehdr Ehdr;
typedef struct Stats Stats;
enum
{
Cdcunion = 6,
Scether = 6,
Fnether = 15,
};
enum
{
Qroot,
Qiface,
Qclone,
Qstats,
Qaddr,
Qndir,
Qctl,
Qdata,
Qtype,
Qmax,
};
struct Tab
{
char *name;
ulong mode;
};
Tab tab[] =
{
"/", DMDIR|0555,
"etherU", DMDIR|0555, /* calling it *ether* makes snoopy(8) happy */
"clone", 0666,
"stats", 0666,
"addr", 0444,
"%ud", DMDIR|0555,
"ctl", 0666,
"data", 0666,
"type", 0444,
};
struct Qbuf
{
Qbuf *next;
int ndata;
uchar data[];
};
struct Dq
{
QLock l;
Dq *next;
Req *r;
Req **rt;
Qbuf *q;
Qbuf **qt;
int nb;
};
struct Conn
{
QLock l;
int used;
int type;
int prom;
Dq *dq;
};
struct Ehdr
{
uchar d[6];
uchar s[6];
uchar type[2];
};
struct Stats
{
int in;
int out;
};
Conn conn[32];
int nconn = 0;
int debug;
ulong time0;
Dev *epin;
Dev *epout;
Stats stats;
uchar macaddr[6];
int maxpacket = 64;
char *uname;
#define PATH(type, n) ((type)|((n)<<8))
#define TYPE(path) (((uint)(path) & 0x000000FF)>>0)
#define NUM(path) (((uint)(path) & 0xFFFFFF00)>>8)
#define NUMCONN(c) (((long)(c)-(long)&conn[0])/sizeof(conn[0]))
static int
receivepacket(void *buf, int len);
static void
fillstat(Dir *d, uvlong path)
{
Tab *t;
memset(d, 0, sizeof(*d));
d->uid = estrdup9p(uname);
d->gid = estrdup9p(uname);
d->qid.path = path;
d->atime = d->mtime = time0;
t = &tab[TYPE(path)];
d->name = smprint(t->name, NUM(path));
d->qid.type = t->mode>>24;
d->mode = t->mode;
}
static void
fsattach(Req *r)
{
if(r->ifcall.aname && r->ifcall.aname[0]){
respond(r, "invalid attach specifier");
return;
}
if(uname == nil)
uname = estrdup9p(r->ifcall.uname);
r->fid->qid.path = PATH(Qroot, 0);
r->fid->qid.type = QTDIR;
r->fid->qid.vers = 0;
r->ofcall.qid = r->fid->qid;
respond(r, nil);
}
static void
fsstat(Req *r)
{
fillstat(&r->d, r->fid->qid.path);
respond(r, nil);
}
static int
rootgen(int i, Dir *d, void*)
{
i += Qroot+1;
if(i == Qiface){
fillstat(d, i);
return 0;
}
return -1;
}
static int
ifacegen(int i, Dir *d, void*)
{
i += Qiface+1;
if(i < Qndir){
fillstat(d, i);
return 0;
}
i -= Qndir;
if(i < nconn){
fillstat(d, PATH(Qndir, i));
return 0;
}
return -1;
}
static int
ndirgen(int i, Dir *d, void *aux)
{
i += Qndir+1;
if(i < Qmax){
fillstat(d, PATH(i, NUMCONN(aux)));
return 0;
}
return -1;
}
static char*
fswalk1(Fid *fid, char *name, Qid *qid)
{
int i, n;
char buf[32];
ulong path;
path = fid->qid.path;
if(!(fid->qid.type&QTDIR))
return "walk in non-directory";
if(strcmp(name, "..") == 0){
switch(TYPE(path)){
case Qroot:
return nil;
case Qiface:
qid->path = PATH(Qroot, 0);
qid->type = tab[Qroot].mode>>24;
return nil;
case Qndir:
qid->path = PATH(Qiface, 0);
qid->type = tab[Qiface].mode>>24;
return nil;
default:
return "bug in fswalk1";
}
}
for(i = TYPE(path)+1; i<nelem(tab); i++){
if(i==Qndir){
n = atoi(name);
snprint(buf, sizeof buf, "%d", n);
if(n < nconn && strcmp(buf, name) == 0){
qid->path = PATH(i, n);
qid->type = tab[i].mode>>24;
return nil;
}
break;
}
if(strcmp(name, tab[i].name) == 0){
qid->path = PATH(i, NUM(path));
qid->type = tab[i].mode>>24;
return nil;
}
if(tab[i].mode&DMDIR)
break;
}
return "directory entry not found";
}
static void
matchrq(Dq *d)
{
Req *r;
Qbuf *b;
while(r = d->r){
int n;
if((b = d->q) == nil)
break;
if((d->q = b->next) == nil)
d->qt = &d->q;
if((d->r = (Req*)r->aux) == nil)
d->rt = &d->r;
n = r->ifcall.count;
if(n > b->ndata)
n = b->ndata;
memmove(r->ofcall.data, b->data, n);
free(b);
r->ofcall.count = n;
respond(r, nil);
}
}
static void
readconndata(Req *r)
{
Dq *d;
d = r->fid->aux;
qlock(&d->l);
if(d->q==nil && d->nb){
qunlock(&d->l);
r->ofcall.count = 0;
respond(r, nil);
return;
}
// enqueue request
r->aux = nil;
*d->rt = r;
d->rt = (Req**)&r->aux;
matchrq(d);
qunlock(&d->l);
}
static void
writeconndata(Req *r)
{
Dq *d;
void *p;
int n;
d = r->fid->aux;
p = r->ifcall.data;
n = r->ifcall.count;
if((n == 11) && memcmp(p, "nonblocking", n)==0){
d->nb = 1;
goto out;
}
if(write(epout->dfd, p, n) < 0){
fprint(2, "write: %r\n");
} else {
/*
* this may not work with all CDC devices. the
* linux driver sends one more random byte
* instead of a zero byte transaction. maybe we
* should do the same?
*/
if(n % epout->maxpkt == 0)
write(epout->dfd, "", 0);
}
if(receivepacket(p, n) == 0)
stats.out++;
out:
r->ofcall.count = n;
respond(r, nil);
}
static char*
mac2str(uchar *m)
{
int i;
char *t = "0123456789abcdef";
static char buf[13];
buf[13] = 0;
for(i=0; i<6; i++){
buf[i*2] = t[m[i]>>4];
buf[i*2+1] = t[m[i]&0xF];
}
return buf;
}
static int
str2mac(uchar *m, char *s)
{
int i;
if(strlen(s) != 12)
return -1;
for(i=0; i<12; i++){
uchar v;
if(s[i] >= 'A' && s[i] <= 'F'){
v = 10 + s[i] - 'A';
} else if(s[i] >= 'a' && s[i] <= 'f'){
v = 10 + s[i] - 'a';
} else if(s[i] >= '0' && s[i] <= '9'){
v = s[i] - '0';
} else {
v = 0;
}
if(i&1){
m[i/2] |= v;
} else {
m[i/2] = v<<4;
}
}
return 0;
}
static void
fsread(Req *r)
{
char buf[200];
char e[ERRMAX];
ulong path;
path = r->fid->qid.path;
switch(TYPE(path)){
default:
snprint(e, sizeof e, "bug in fsread path=%lux", path);
respond(r, e);
break;
case Qroot:
dirread9p(r, rootgen, nil);
respond(r, nil);
break;
case Qiface:
dirread9p(r, ifacegen, nil);
respond(r, nil);
break;
case Qstats:
snprint(buf, sizeof(buf),
"in: %d\n"
"out: %d\n"
"mbps: %d\n"
"addr: %s\n",
stats.in, stats.out, 10, mac2str(macaddr));
readstr(r, buf);
respond(r, nil);
break;
case Qaddr:
readstr(r, mac2str(macaddr));
respond(r, nil);
break;
case Qndir:
dirread9p(r, ndirgen, &conn[NUM(path)]);
respond(r, nil);
break;
case Qctl:
snprint(buf, sizeof(buf), "%11d ", NUM(path));
readstr(r, buf);
respond(r, nil);
break;
case Qtype:
snprint(buf, sizeof(buf), "%11d ", conn[NUM(path)].type);
readstr(r, buf);
respond(r, nil);
break;
case Qdata:
readconndata(r);
break;
}
}
static void
fswrite(Req *r)
{
char e[ERRMAX];
ulong path;
char *p;
int n;
path = r->fid->qid.path;
switch(TYPE(path)){
case Qctl:
n = r->ifcall.count;
p = (char*)r->ifcall.data;
if((n == 11) && memcmp(p, "promiscuous", 11)==0)
conn[NUM(path)].prom = 1;
if((n > 8) && memcmp(p, "connect ", 8)==0){
char x[12];
if(n - 8 >= sizeof(x)){
respond(r, "invalid control msg");
return;
}
p += 8;
memcpy(x, p, n-8);
x[n-8] = 0;
conn[NUM(path)].type = atoi(p);
}
r->ofcall.count = n;
respond(r, nil);
break;
case Qdata:
writeconndata(r);
break;
default:
snprint(e, sizeof e, "bug in fswrite path=%lux", path);
respond(r, e);
}
}
static void
fsopen(Req *r)
{
static int need[4] = { 4, 2, 6, 1 };
ulong path;
int i, n;
Tab *t;
Dq *d;
Conn *c;
/*
* lib9p already handles the blatantly obvious.
* we just have to enforce the permissions we have set.
*/
path = r->fid->qid.path;
t = &tab[TYPE(path)];
n = need[r->ifcall.mode&3];
if((n&t->mode) != n){
respond(r, "permission denied");
return;
}
d = nil;
r->fid->aux = nil;
switch(TYPE(path)){
case Qclone:
for(i=0; i<nelem(conn); i++){
if(conn[i].used)
continue;
if(i >= nconn)
nconn = i+1;
path = PATH(Qctl, i);
goto CaseConn;
}
respond(r, "out of connections");
return;
case Qdata:
d = emalloc9p(sizeof(*d));
memset(d, 0, sizeof(*d));
d->qt = &d->q;
d->rt = &d->r;
r->fid->aux = d;
case Qndir:
case Qctl:
case Qtype:
CaseConn:
c = &conn[NUM(path)];
qlock(&c->l);
if(c->used++ == 0){
c->type = 0;
c->prom = 0;
}
if(d != nil){
d->next = c->dq;
c->dq = d;
}
qunlock(&c->l);
break;
}
r->fid->qid.path = path;
r->ofcall.qid.path = path;
respond(r, nil);
}
static void
fsflush(Req *r)
{
Req *o, **p;
Fid *f;
Dq *d;
o = r->oldreq;
f = o->fid;
if(TYPE(f->qid.path) == Qdata){
d = f->aux;
qlock(&d->l);
for(p=&d->r; *p; p=(Req**)&((*p)->aux)){
if(*p == o){
if((*p = (Req*)o->aux) == nil)
d->rt = p;
r->oldreq = nil;
respond(o, "interrupted");
break;
}
}
qunlock(&d->l);
}
respond(r, nil);
}
static void
fsdestroyfid(Fid *fid)
{
Conn *c;
Qbuf *b;
Dq **x, *d;
if(TYPE(fid->qid.path) >= Qndir){
c = &conn[NUM(fid->qid.path)];
qlock(&c->l);
if(d = fid->aux){
fid->aux = nil;
for(x=&c->dq; *x; x=&((*x)->next)){
if(*x == d){
*x = d->next;
break;
}
}
qlock(&d->l);
while(b = d->q){
d->q = b->next;
free(b);
}
qunlock(&d->l);
}
if(TYPE(fid->qid.path) == Qctl)
c->prom = 0;
c->used--;
qunlock(&c->l);
}
}
static void
setalt(Dev *d, int ifcid, int altid)
{
if(usbcmd(d, Rh2d|Rstd|Riface, Rsetiface, altid, ifcid, nil, 0) < 0)
dprint(2, "%s: setalt ifc %d alt %d: %r\n", argv0, ifcid, altid);
}
static int
ifaceinit(Dev *d, Iface *ifc, int *ei, int *eo)
{
int i, epin, epout;
Ep *ep;
if(ifc == nil)
return -1;
epin = epout = -1;
for(i = 0; (epin < 0 || epout < 0) && i < nelem(ifc->ep); i++)
if((ep = ifc->ep[i]) != nil && ep->type == Ebulk){
if(ep->dir == Eboth || ep->dir == Ein)
if(epin == -1)
epin = ep->id;
if(ep->dir == Eboth || ep->dir == Eout)
if(epout == -1)
epout = ep->id;
}
if(epin == -1 || epout == -1)
return -1;
for(i = 0; i < nelem(ifc->altc); i++)
if(ifc->altc[i] != nil)
setalt(d, ifc->id, i);
*ei = epin;
*eo = epout;
return 0;
}
int
findendpoints(Dev *d, int *ei, int *eo)
{
int i, j, ctlid, datid;
Iface *ctlif, *datif;
Usbdev *ud;
Desc *desc;
Conf *c;
ud = d->usb;
*ei = *eo = -1;
/* look for union descriptor with ethernet ctrl interface */
for(i = 0; i < nelem(ud->ddesc); i++){
if((desc = ud->ddesc[i]) == nil)
continue;
if(desc->data.bLength < 5 || desc->data.bbytes[0] != Cdcunion)
continue;
ctlid = desc->data.bbytes[1];
datid = desc->data.bbytes[2];
if((c = desc->conf) == nil)
continue;
ctlif = datif = nil;
for(j = 0; j < nelem(c->iface); j++){
if(c->iface[j] == nil)
continue;
if(c->iface[j]->id == ctlid)
ctlif = c->iface[j];
if(c->iface[j]->id == datid)
datif = c->iface[j];
if(datif != nil && ctlif != nil){
if(Subclass(ctlif->csp) == Scether)
if(ifaceinit(d, datif, ei, eo) != -1)
return 0;
break;
}
}
}
/* try any other one that seems to be ok */
for(i = 0; i < nelem(ud->conf); i++)
if((c = ud->conf[i]) != nil)
for(j = 0; j < nelem(c->iface); j++)
if(ifaceinit(d,c->iface[j],ei,eo) != -1)
return 0;
return -1;
}
static int
getmac(Dev *d)
{
int i;
Usbdev *ud;
uchar *b;
Desc *dd;
char *mac;
ud = d->usb;
for(i = 0; i < nelem(ud->ddesc); i++)
if((dd = ud->ddesc[i]) != nil){
b = (uchar*)&dd->data;
if(b[1] == Dfunction && b[2] == Fnether){
mac = loaddevstr(d, b[3]);
if(mac != nil && strlen(mac) != 12){
free(mac);
mac = nil;
}
if(mac != nil){
str2mac(macaddr, mac);
free(mac);
return 0;
}
}
}
return -1;
}
static int
inote(void *, char *msg)
{
if(strstr(msg, "interrupt"))
return 1;
return 0;
}
static int
receivepacket(void *buf, int len)
{
int i;
int t;
Ehdr *h;
if(len < sizeof(*h))
return -1;
h = (Ehdr*)buf;
t = (h->type[0]<<8)|h->type[1];
for(i=0; i<nconn; i++){
Qbuf *b;
Conn *c;
Dq *d;
c = &conn[i];
qlock(&c->l);
if(!c->used)
goto next;
if(c->type > 0)
if(c->type != t)
goto next;
if(!c->prom && !(h->d[0]&1))
if(memcmp(h->d, macaddr, 6))
goto next;
for(d=c->dq; d; d=d->next){
int n;
n = len;
if(c->type == -2 && n > 64)
n = 64;
b = emalloc9p(sizeof(*b) + n);
b->ndata = n;
memcpy(b->data, buf, n);
qlock(&d->l);
// enqueue buffer
b->next = nil;
*d->qt = b;
d->qt = &b->next;
matchrq(d);
qunlock(&d->l);
}
next:
qunlock(&c->l);
}
return 0;
}
static void
usbreadproc(void *)
{
char err[ERRMAX];
uchar buf[4*1024];
atnotify(inote, 1);
threadsetname("usbreadproc");
for(;;){
int n;
n = read(epin->dfd, buf, sizeof(buf));
if(n < 0){
rerrstr(err, sizeof(err));
if(strstr(err, "interrupted") || strstr(err, "timed out"))
continue;
fprint(2, "usbreadproc: %s\n", err);
threadexitsall(err);
}
if(n == 0)
continue;
if(receivepacket(buf, n) == 0)
stats.in++;
}
}
Srv fs =
{
.attach= fsattach,
.destroyfid= fsdestroyfid,
.walk1= fswalk1,
.open= fsopen,
.read= fsread,
.write= fswrite,
.stat= fsstat,
.flush= fsflush,
};
static void
usage(void)
{
fprint(2, "usage: %s [-dD] devid\n", argv0);
exits("usage");
}
void
threadmain(int argc, char **argv)
{
char s[64];
int ei, eo;
Dev *d;
ARGBEGIN {
case 'd':
debug = 1;
break;
case 'D':
chatty9p++;
break;
default:
usage();
} ARGEND;
if(argc != 1)
usage();
d = getdev(atoi(*argv));
if(configdev(d) < 0)
sysfatal("configdev: %r");
if(findendpoints(d, &ei, &eo) < 0)
sysfatal("no endpoints found");
if(getmac(d) < 0)
sysfatal("can't get mac address");
if((epin = openep(d, ei)) == nil)
sysfatal("openep: %r");
if(ei == eo){
incref(epin);
epout = epin;
opendevdata(epin, ORDWR);
} else {
if((epout = openep(d, eo)) == nil)
sysfatal("openep: %r");
opendevdata(epin, OREAD);
opendevdata(epout, OWRITE);
}
proccreate(usbreadproc, nil, 8*1024);
atnotify(inote, 1);
time0 = time(0);
tab[Qiface].name = smprint("etherU%d", d->id);
snprint(s, sizeof(s), "%d.ether", d->id);
closedev(d);
threadpostsharesrv(&fs, nil, "usbnet", s);
}

View file

@ -0,0 +1,12 @@
</$objtype/mkfile
BIN=/$objtype/bin/nusb
LIB=../lib/usb.a$O
TARG=ether
HFILES=
OFILES=ether.$O
</sys/src/cmd/mkone
CFLAGS=-I../lib $CFLAGS

View file

@ -4,6 +4,7 @@ DIRS=\
lib\ lib\
kb\ kb\
audio\ audio\
ether\
usbd\ usbd\
disk\ disk\
serial\ serial\