/* * cec — coraid ethernet console * Copyright © Coraid, Inc. 2006-2008. * All Rights Reserved. */#include <u.h>#include <libc.h>#include <ip.h>		/* really! */#include <ctype.h>#include "cec.h" enum {	Tinita		= 0,	Tinitb,	Tinitc,	Tdata,	Tack,	Tdiscover,	Toffer,	Treset, 	Hdrsz		= 18,	Eaddrlen	= 6,}; typedef struct{	uchar	ea[Eaddrlen];	int	major;	char	name[28];} Shelf; int 	conn(int);void 	gettingkilled(int);int 	pickone(void);void 	probe(void);void	sethdr(Pkt *, int);int	shelfidx(void); Shelf	*con;Shelf	tab[1000]; char	*host;char	*srv;char	*svc; char	pflag; int	ntab;int	shelf = -1; uchar	bcast[] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};uchar	contag;uchar 	esc = '';uchar	ea[Eaddrlen];uchar	unsetea[Eaddrlen]; extern 	int fd;		/* set in netopen */ voidpost(char *srv, int fd){	char buf[32];	int f; 	if((f = create(srv, OWRITE, 0666)) == -1)		sysfatal("create %s: %r", srv);	snprint(buf, sizeof buf, "%d", fd);	if(write(f, buf, strlen(buf)) != strlen(buf))		sysfatal("write %s: %r", srv);	close(f);} voiddosrv(char *s){	int p[2]; 	if(pipe(p) < 0)		sysfatal("pipe: %r");	if (srv[0] != '/')		svc = smprint("/srv/%s", s);	else		svc = smprint("%s", s);	post(svc, p[0]);	close(p[0]);	dup(p[1], 0);	dup(p[1], 1); 	switch(rfork(RFFDG|RFPROC|RFNAMEG|RFNOTEG)){	case -1:		sysfatal("fork: %r");	case 0:		break;	default:		exits("");	}	close(2);} voidusage(void){	fprint(2, "usage: cec [-dp] [-c esc] [-e ea] [-h host] [-s shelf] "		"[-S srv] interface\n");	exits0("usage");} voidcatch(void*, char *note){	if(strcmp(note, "alarm") == 0)		noted(NCONT);	noted(NDFLT);} intnilea(uchar *ea){	return memcmp(ea, unsetea, Eaddrlen) == 0;} voidmain(int argc, char **argv){	int r, n; 	ARGBEGIN{	case 'S':		srv = EARGF(usage());		break;	case 'c':		esc = tolower(*(EARGF(usage()))) - 'a' + 1;		if(esc == 0 || esc >= ' ')			usage();		break;	case 'd':		debug++;		break;	case 'e':		if(parseether(ea, EARGF(usage())) == -1)			usage();		pflag = 1;		break;	case 'h':		host = EARGF(usage());		break;	case 'p':		pflag = 1;		break;	case 's':		shelf = atoi(EARGF(usage()));		break;	default:		usage();	}ARGEND	if(argc == 0)		*argv = "/net/ether0";	else if(argc != 1)		usage(); 	fmtinstall('E', eipfmt);	if(srv != nil)		dosrv(srv);	r = netopen(*argv);	if(r == -1){		fprint(2, "cec: can't netopen %s\n", *argv);		exits0("open");	}	notify(catch);	probe();	for(;;){		n = 0;		if(shelf == -1 && host == 0 && nilea(ea))			n = pickone();		rawon();		conn(n);		rawoff();		if(pflag == 0){			if(shelf != -1)				exits0("shelf not found");			if(host)				exits0("host not found");			if(!nilea(ea))				exits0("ea not found");		} else if(shelf != -1 || host || !nilea(ea))			exits0("");	}} voidtimewait(int ms){	alarm(ms);} intdidtimeout(void){	char buf[ERRMAX]; 	rerrstr(buf, sizeof buf);	if(strcmp(buf, "interrupted") == 0){		werrstr(buf, 0);		return 1;	}	return 0;} ushorthtons(ushort h){	ushort n;	uchar *p; 	p = (uchar*)&n;	p[0] = h >> 8;	p[1] = h;	return n;} ushortntohs(int h){	ushort n;	uchar *p; 	n = h;	p = (uchar*)&n;	return p[0] << 8 | p[1];} inttcmp(void *a, void *b){	Shelf *s, *t;	int d; 	s = a;	t = b;	d = s->major - t->major;	if(d == 0)		d = strcmp(s->name, t->name);	if(d == 0)		d = memcmp(s->ea, t->ea, Eaddrlen);	return d;} voidprobe(void){	char *sh, *other;	int n;	Pkt q;	Shelf *p; 	do {		ntab = 0;		memset(q.dst, 0xff, Eaddrlen);		memset(q.src, 0, Eaddrlen);		q.etype = htons(Etype);		q.type = Tdiscover;		q.len = 0;		q.conn = 0;		q.seq = 0;		netsend(&q, 60);		timewait(Iowait);		while((n = netget(&q, sizeof q)) >= 0){			if((n <= 0 && didtimeout()) || ntab == nelem(tab))				break;			if(n < 60 || q.len == 0 || q.type != Toffer)				continue;			q.data[q.len] = 0;			sh = strtok((char *)q.data, " \t");			if(sh == nil)				continue;			if(!nilea(ea) && memcmp(ea, q.src, Eaddrlen) != 0)				continue;			if(shelf != -1 && atoi(sh) != shelf)				continue;			other = strtok(nil, "\x1");			if(other == 0)				other = "";			if(host && strcmp(host, other) != 0)				continue;			p = tab + ntab++;			memcpy(p->ea, q.src, Eaddrlen);			p->major = atoi(sh);			p->name[0] = 0;			if(p->name)				snprint(p->name, sizeof p->name, "%s", other);		}		alarm(0);	} while (ntab == 0 && pflag);	if(ntab == 0){		fprint(2, "none found.\n");		exits0("none found");	}	qsort(tab, ntab, sizeof tab[0], tcmp);} voidshowtable(void){	int i; 	for(i = 0; i < ntab; i++)		print("%2d   %5d %E %s\n", i, tab[i].major, tab[i].ea, tab[i].name);} intpickone(void){	char buf[80];	int n, i; 	for(;;){		showtable();		print("[#qp]: ");		switch(n = read(0, buf, sizeof buf)){		case 1:			if(buf[0] == '\n')				continue;			/* fall through */		case 2:			if(buf[0] == 'p'){				probe();				break;			}			if(buf[0] == 'q')				/* fall through */		case 0:		case -1:				exits0(0);			break;		}		if(isdigit(buf[0])){			buf[n] = 0;			i = atoi(buf);			if(i >= 0 && i < ntab)				break;		}	}	return i;} voidsethdr(Pkt *pp, int type){	memmove(pp->dst, con->ea, Eaddrlen);	memset(pp->src, 0, Eaddrlen);	pp->etype = htons(Etype);	pp->type = type;	pp->len = 0;	pp->conn = contag;} voidethclose(void){	static Pkt msg; 	sethdr(&msg, Treset);	timewait(Iowait);	netsend(&msg, 60);	alarm(0);	con = 0;} intethopen(void){	Pkt tpk, rpk;	int i, n; 	contag = (getpid() >> 8) ^ (getpid() & 0xff);	sethdr(&tpk, Tinita);	sethdr(&rpk, 0);	for(i = 0; i < 3 && rpk.type != Tinitb; i++){		netsend(&tpk, 60);		timewait(Iowait);		n = netget(&rpk, 1000);		alarm(0);		if(n < 0)			return -1;	}	if(rpk.type != Tinitb)		return -1;	sethdr(&tpk, Tinitc);	netsend(&tpk, 60);	return 0;} charescape(void){	char buf[64];	int r; 	for(;;){		fprint(2, ">>> ");		buf[0] = '.';		rawoff();		r = read(0, buf, sizeof buf - 1);		rawon();		if(r == -1)			exits0("kbd: %r");		switch(buf[0]){		case 'i':		case 'q':		case '.':			return buf[0];		}		fprint(2, "	(q)uit, (i)nterrupt, (.)continue\n");	}} /* * this is a bit too aggressive.  it really needs to replace only \n\r with \n. */static uchar crbuf[256]; voidnocrwrite(int fd, uchar *buf, int n){	int i, j, c; 	j = 0;	for(i = 0; i < n; i++)		if((c = buf[i]) != '\r')			crbuf[j++] = c;	write(fd, crbuf, j);} intdoloop(void){	int unacked, retries, set[2];	uchar c, tseq, rseq;	uchar ea[Eaddrlen];	Pkt tpk, spk;	Mux *m; 	memmove(ea, con->ea, Eaddrlen);	retries = 0;	unacked = 0;	tseq = 0;	rseq = -1;	set[0] = 0;	set[1] = fd;top:	if ((m = mux(set)) == 0)		exits0("mux: %r");	for (; ; )		switch (muxread(m, &spk)) {		case -1:			if (unacked == 0)				break;			if (retries-- == 0) {				fprint(2, "Connection timed out\n");				muxfree(m);				return 0;			}			netsend(&tpk, Hdrsz + unacked);			break;		case Fkbd:			c = spk.data[0];			if (c == esc) {				muxfree(m);				switch (escape()) {				case 'q':					tpk.len = 0;					tpk.type = Treset;					netsend(&tpk, 60);					return 0;				case '.':					goto top;				case 'i':					if ((m = mux(set)) == 0)						exits0("mux: %r");					break;				}			}			sethdr(&tpk, Tdata);			memcpy(tpk.data, spk.data, spk.len);			tpk.len = spk.len;			tpk.seq = ++tseq;			unacked = spk.len;			retries = 2;			netsend(&tpk, Hdrsz + spk.len);			break;		case Fcec:			if (memcmp(spk.src, ea, Eaddrlen) != 0 ||			    ntohs(spk.etype) != Etype)				continue;			if (spk.type == Toffer &&			    memcmp(spk.dst, bcast, Eaddrlen) != 0) {				muxfree(m);				return 1;			}			if (spk.conn != contag)

Lines 1–500 of 558. Load more