new goodies

src/cmd/ip/snoopy/il.c

@@ -0,0 +1,146 @@
+#include <u.h>
+#include <libc.h>
+#include <ip.h>
+#include "dat.h"
+#include "protos.h"
+
+typedef struct Hdr	Hdr;
+struct Hdr
+{
+	uchar	sum[2];		/* Checksum including header */
+	uchar	len[2];		/* Packet length */
+	uchar	type;		/* Packet type */
+	uchar	spec;		/* Special */
+	uchar	sport[2];	/* Src port */
+	uchar	dport[2];	/* Dst port */
+	uchar	id[4];		/* Sequence id */
+	uchar	ack[4];		/* Acked sequence */
+};
+
+enum
+{
+	ILLEN= 18,
+};
+
+enum
+{
+	Os,
+	Od,
+	Osd,
+};
+
+static Field p_fields[] = 
+{
+	{"s",		Fnum,	Os,	"source port",	} ,
+	{"d",		Fnum,	Od,	"dest port",	} ,
+	{"a",		Fnum,	Osd,	"source/dest port",	} ,
+	{"sd",		Fnum,	Osd,	"source/dest port",	} ,
+	{0}
+};
+
+static Mux p_mux[] =
+{
+	{"ninep",	17007, },	/* exportfs */
+	{"ninep",	17008, },	/* 9fs */
+	{"ninep",	17005, },	/* ocpu */
+	{"ninep",	17010, },	/* ncpu */
+	{"ninep",	17013, },	/* cpu */
+	{0},
+};
+
+static void
+p_compile(Filter *f)
+{
+	Mux *m;
+
+	if(f->op == '='){
+		compile_cmp(udp.name, f, p_fields);
+		return;
+	}
+	for(m = p_mux; m->name != nil; m++)
+		if(strcmp(f->s, m->name) == 0){
+			f->pr = m->pr;
+			f->ulv = m->val;
+			f->subop = Osd;
+			return;
+		}
+	sysfatal("unknown il field or protocol: %s", f->s);
+}
+
+static int
+p_filter(Filter *f, Msg *m)
+{
+	Hdr *h;
+
+	if(m->pe - m->ps < ILLEN)
+		return 0;
+	h = (Hdr*)m->ps;
+	m->ps += ILLEN;
+
+	switch(f->subop){
+	case Os:
+		return NetS(h->sport) == f->ulv;
+	case Od:
+		return NetS(h->dport) == f->ulv;
+	case Osd:
+		return NetS(h->sport) == f->ulv || NetS(h->dport) == f->ulv;
+	}
+	return 0;
+}
+
+char *pktnames[] = 
+{
+	"Sync",	
+	"Data",
+	"Dataquery",
+	"Ack",
+	"Query",
+	"State",
+	"Close"
+};
+
+static char*
+pkttype(int t)
+{
+	static char b[10];
+	
+	if(t > 6){
+		sprint(b, "%d", t);
+		return b;
+	}
+	return pktnames[t];
+}
+
+static int
+p_seprint(Msg *m)
+{
+	Hdr *h;
+	int dport, sport;
+
+	if(m->pe - m->ps < ILLEN)
+		return -1;
+	h = (Hdr*)m->ps;
+	m->ps += ILLEN;
+
+	dport = NetS(h->dport);
+	sport = NetS(h->sport);
+	demux(p_mux, sport, dport, m, &dump);
+
+	m->p = seprint(m->p, m->e, "s=%d d=%d t=%s id=%lud ack=%lud spec=%d ck=%4.4ux ln=%d",
+			sport, dport, pkttype(h->type),
+			(ulong)NetL(h->id), (ulong)NetL(h->ack),
+			h->spec,
+			NetS(h->sum), NetS(h->len));
+	return 0;
+}
+
+Proto il =
+{
+	"il",
+	p_compile,
+	p_filter,
+	p_seprint,
+	p_mux,
+	p_fields,
+	defaultframer,
+};