X-Git-Url: https://git.tcpdump.org/tcpdump/blobdiff_plain/546558eabd81cfc36a81a4df728fdfea0d83b41a..d2777156522f139a858bd6b5b51e364826bc95a7:/print-wb.c

diff --git a/print-wb.c b/print-wb.c
index 42686906..35b5a196 100644
--- a/print-wb.c
+++ b/print-wb.c
@@ -27,30 +27,29 @@
 
 #include "netdissect-stdinc.h"
 
+#define ND_LONGJMP_FROM_TCHECK
 #include "netdissect.h"
 #include "addrtoname.h"
 #include "extract.h"
 
-static const char tstr[] = "[|wb]";
-
-/* XXX need to add byte-swapping macros! */
-/* XXX - you mean like the ones in "extract.h"? */
 
+#if 0
 /*
  * Largest packet size.  Everything should fit within this space.
  * For instance, multiline objects are sent piecewise.
  */
 #define MAXFRAMESIZE 1024
+#endif
 
 /*
  * Multiple drawing ops can be sent in one packet.  Each one starts on a
  * an even multiple of DOP_ALIGN bytes, which must be a power of two.
  */
 #define DOP_ALIGN 4
-#define DOP_ROUNDUP(x)	((((int)(x)) + (DOP_ALIGN - 1)) & ~(DOP_ALIGN - 1))
+#define DOP_ROUNDUP(x)	roundup2(x, DOP_ALIGN)
 #define DOP_NEXT(d)\
 	((const struct dophdr *)((const u_char *)(d) + \
-				DOP_ROUNDUP(EXTRACT_BE_U_2((d)->dh_len) + sizeof(*(d)))))
+				DOP_ROUNDUP(GET_BE_U_2((d)->dh_len) + sizeof(*(d)))))
 
 /*
  * Format of the whiteboard packet header.
@@ -73,6 +72,7 @@ struct pkt_hdr {
 #define PT_PREQ         5       /* page vector request */
 #define PT_PREP         7       /* page vector reply */
 
+#if 0
 #ifdef PF_USER
 #undef PF_USER			/* {Digital,Tru64} UNIX define this, alas */
 #endif
@@ -80,6 +80,7 @@ struct pkt_hdr {
 /* flags */
 #define PF_USER		0x01	/* hint that packet has interactive data */
 #define PF_VIS		0x02	/* only visible ops wanted */
+#endif
 
 struct PageID {
 	nd_uint32_t p_sid;		/* session id of initiator */
@@ -110,7 +111,23 @@ struct dophdr {
 #define DT_REF          13
 #define DT_SKIP         14
 #define DT_HOLE         15
-#define DT_MAXTYPE      15
+static const struct tok dop_str[] = {
+	{ DT_RECT,   "RECT"   },
+	{ DT_LINE,   "LINE"   },
+	{ DT_ML,     "ML"     },
+	{ DT_DEL,    "DEL"    },
+	{ DT_XFORM,  "XFORM"  },
+	{ DT_ELL,    "ELL"    },
+	{ DT_CHAR,   "CHAR"   },
+	{ DT_STR,    "STR"    },
+	{ DT_NOP,    "NOP"    },
+	{ DT_PSCODE, "PSCODE" },
+	{ DT_PSCOMP, "PSCOMP" },
+	{ DT_REF,    "REF"    },
+	{ DT_SKIP,   "SKIP"   },
+	{ DT_HOLE,   "HOLE"   },
+	{ 0, NULL }
+};
 
 /*
  * A drawing operation.
@@ -181,45 +198,43 @@ wb_id(netdissect_options *ndo,
       const struct pkt_id *id, u_int len)
 {
 	u_int i;
-	const char *cp;
+	const u_char *sitename;
 	const struct id_off *io;
 	char c;
 	u_int nid;
 
 	ND_PRINT(" wb-id:");
-	if (len < sizeof(*id) || !ND_TTEST_SIZE(id))
+	if (len < sizeof(*id))
 		return (-1);
 	len -= sizeof(*id);
 
 	ND_PRINT(" %u/%s:%u (max %u/%s:%u) ",
-	       EXTRACT_BE_U_4(id->pi_ps.slot),
-	       ipaddr_string(ndo, id->pi_ps.page.p_sid),
-	       EXTRACT_BE_U_4(id->pi_ps.page.p_uid),
-	       EXTRACT_BE_U_4(id->pi_mslot),
-	       ipaddr_string(ndo, id->pi_mpage.p_sid),
-	       EXTRACT_BE_U_4(id->pi_mpage.p_uid));
-
-	nid = EXTRACT_BE_U_2(id->pi_ps.nid);
+	       GET_BE_U_4(id->pi_ps.slot),
+	       GET_IPADDR_STRING(id->pi_ps.page.p_sid),
+	       GET_BE_U_4(id->pi_ps.page.p_uid),
+	       GET_BE_U_4(id->pi_mslot),
+	       GET_IPADDR_STRING(id->pi_mpage.p_sid),
+	       GET_BE_U_4(id->pi_mpage.p_uid));
+	/* now the rest of the fixed-size part of struct pkt_id */
+	ND_TCHECK_SIZE(id);
+
+	nid = GET_BE_U_2(id->pi_ps.nid);
+	if (len < sizeof(*io) * nid)
+		return (-1);
 	len -= sizeof(*io) * nid;
 	io = (const struct id_off *)(id + 1);
-	cp = (const char *)(io + nid);
-	if (ND_TTEST_LEN(cp, len)) {
-		ND_PRINT("\"");
-		fn_print(ndo, (const u_char *)cp, (const u_char *)cp + len);
-		ND_PRINT("\"");
-	}
+	sitename = (const u_char *)(io + nid);
 
 	c = '<';
-	for (i = 0; i < nid && ND_TTEST_SIZE(io); ++io, ++i) {
+	for (i = 0; i < nid; ++io, ++i) {
 		ND_PRINT("%c%s:%u",
-		    c, ipaddr_string(ndo, io->id), EXTRACT_BE_U_4(io->off));
+		    c, GET_IPADDR_STRING(io->id), GET_BE_U_4(io->off));
 		c = ',';
 	}
-	if (i >= nid) {
-		ND_PRINT(">");
-		return (0);
-	}
-	return (-1);
+	ND_PRINT("> \"");
+	nd_printjnp(ndo, sitename, len);
+	ND_PRINT("\"");
+	return (0);
 }
 
 static int
@@ -227,15 +242,15 @@ wb_rreq(netdissect_options *ndo,
         const struct pkt_rreq *rreq, u_int len)
 {
 	ND_PRINT(" wb-rreq:");
-	if (len < sizeof(*rreq) || !ND_TTEST_SIZE(rreq))
+	if (len < sizeof(*rreq))
 		return (-1);
 
 	ND_PRINT(" please repair %s %s:%u<%u:%u>",
-	       ipaddr_string(ndo, rreq->pr_id),
-	       ipaddr_string(ndo, rreq->pr_page.p_sid),
-	       EXTRACT_BE_U_4(rreq->pr_page.p_uid),
-	       EXTRACT_BE_U_4(rreq->pr_sseq),
-	       EXTRACT_BE_U_4(rreq->pr_eseq));
+	       GET_IPADDR_STRING(rreq->pr_id),
+	       GET_IPADDR_STRING(rreq->pr_page.p_sid),
+	       GET_BE_U_4(rreq->pr_page.p_uid),
+	       GET_BE_U_4(rreq->pr_sseq),
+	       GET_BE_U_4(rreq->pr_eseq));
 	return (0);
 }
 
@@ -244,13 +259,15 @@ wb_preq(netdissect_options *ndo,
         const struct pkt_preq *preq, u_int len)
 {
 	ND_PRINT(" wb-preq:");
-	if (len < sizeof(*preq) || !ND_TTEST_SIZE(preq))
+	if (len < sizeof(*preq))
 		return (-1);
 
 	ND_PRINT(" need %u/%s:%u",
-	       EXTRACT_BE_U_4(preq->pp_low),
-	       ipaddr_string(ndo, preq->pp_page.p_sid),
-	       EXTRACT_BE_U_4(preq->pp_page.p_uid));
+	       GET_BE_U_4(preq->pp_low),
+	       GET_IPADDR_STRING(preq->pp_page.p_sid),
+	       GET_BE_U_4(preq->pp_page.p_uid));
+	/* now the rest of the fixed-size part of struct pkt_req */
+	ND_TCHECK_SIZE(preq);
 	return (0);
 }
 
@@ -260,55 +277,36 @@ wb_prep(netdissect_options *ndo,
 {
 	u_int n;
 	const struct pgstate *ps;
-	const u_char *ep = ndo->ndo_snapend;
 
 	ND_PRINT(" wb-prep:");
-	if (len < sizeof(*prep) || !ND_TTEST_SIZE(prep))
+	if (len < sizeof(*prep))
 		return (-1);
-	n = EXTRACT_BE_U_4(prep->pp_n);
+	n = GET_BE_U_4(prep->pp_n);
 	ps = (const struct pgstate *)(prep + 1);
-	while (n != 0 && ND_TTEST_SIZE(ps)) {
+	while (n != 0) {
 		const struct id_off *io, *ie;
 		char c = '<';
 
 		ND_PRINT(" %u/%s:%u",
-		    EXTRACT_BE_U_4(ps->slot),
-		    ipaddr_string(ndo, ps->page.p_sid),
-		    EXTRACT_BE_U_4(ps->page.p_uid));
+		    GET_BE_U_4(ps->slot),
+		    GET_IPADDR_STRING(ps->page.p_sid),
+		    GET_BE_U_4(ps->page.p_uid));
+		/* now the rest of the fixed-size part of struct pgstate */
+		ND_TCHECK_SIZE(ps);
 		io = (const struct id_off *)(ps + 1);
-		for (ie = io + EXTRACT_U_1(ps->nid); io < ie && ND_TTEST_SIZE(io); ++io) {
-			ND_PRINT("%c%s:%u", c, ipaddr_string(ndo, io->id),
-			    EXTRACT_BE_U_4(io->off));
+		for (ie = io + GET_U_1(ps->nid); io < ie; ++io) {
+			ND_PRINT("%c%s:%u", c, GET_IPADDR_STRING(io->id),
+			    GET_BE_U_4(io->off));
 			c = ',';
 		}
 		ND_PRINT(">");
 		ps = (const struct pgstate *)io;
 		n--;
 	}
-	return ((const u_char *)ps <= ep? 0 : -1);
+	return 0;
 }
 
-
-static const char *dopstr[] = {
-	"dop-0!",
-	"dop-1!",
-	"RECT",
-	"LINE",
-	"ML",
-	"DEL",
-	"XFORM",
-	"ELL",
-	"CHAR",
-	"STR",
-	"NOP",
-	"PSCODE",
-	"PSCOMP",
-	"REF",
-	"SKIP",
-	"HOLE",
-};
-
-static int
+static void
 wb_dops(netdissect_options *ndo, const struct pkt_dop *dop,
         uint32_t ss, uint32_t es)
 {
@@ -318,31 +316,22 @@ wb_dops(netdissect_options *ndo, const struct pkt_dop *dop,
 	for ( ; ss <= es; ++ss) {
 		u_int t;
 
-		if (!ND_TTEST_SIZE(dh)) {
-			ND_PRINT("%s", tstr);
-			break;
-		}
-		t = EXTRACT_U_1(dh->dh_type);
-
-		if (t > DT_MAXTYPE)
-			ND_PRINT(" dop-%u!", t);
-		else {
-			ND_PRINT(" %s", dopstr[t]);
-			if (t == DT_SKIP || t == DT_HOLE) {
-				uint32_t ts = EXTRACT_BE_U_4(dh->dh_ts);
-				ND_PRINT("%u", ts - ss + 1);
-				if (ss > ts || ts > es) {
-					ND_PRINT("[|]");
-					if (ts < ss)
-						return (0);
-				}
-				ss = ts;
+		t = GET_U_1(dh->dh_type);
+
+		ND_PRINT(" %s", tok2str(dop_str, "dop-%u!", t));
+		if (t == DT_SKIP || t == DT_HOLE) {
+			uint32_t ts = GET_BE_U_4(dh->dh_ts);
+			ND_PRINT("%u", ts - ss + 1);
+			if (ss > ts || ts > es) {
+				ND_PRINT("[|]");
+				if (ts < ss)
+					return;
 			}
+			ss = ts;
 		}
 		dh = DOP_NEXT(dh);
 	}
 	ND_PRINT(" >");
-	return (0);
 }
 
 static int
@@ -352,21 +341,21 @@ wb_rrep(netdissect_options *ndo,
 	const struct pkt_dop *dop = &rrep->pr_dop;
 
 	ND_PRINT(" wb-rrep:");
-	if (len < sizeof(*rrep) || !ND_TTEST_SIZE(rrep))
+	if (len < sizeof(*rrep))
 		return (-1);
 	len -= sizeof(*rrep);
 
 	ND_PRINT(" for %s %s:%u<%u:%u>",
-	    ipaddr_string(ndo, rrep->pr_id),
-	    ipaddr_string(ndo, dop->pd_page.p_sid),
-	    EXTRACT_BE_U_4(dop->pd_page.p_uid),
-	    EXTRACT_BE_U_4(dop->pd_sseq),
-	    EXTRACT_BE_U_4(dop->pd_eseq));
+	    GET_IPADDR_STRING(rrep->pr_id),
+	    GET_IPADDR_STRING(dop->pd_page.p_sid),
+	    GET_BE_U_4(dop->pd_page.p_uid),
+	    GET_BE_U_4(dop->pd_sseq),
+	    GET_BE_U_4(dop->pd_eseq));
 
 	if (ndo->ndo_vflag)
-		return (wb_dops(ndo, dop,
-		    EXTRACT_BE_U_4(dop->pd_sseq),
-		    EXTRACT_BE_U_4(dop->pd_eseq)));
+		wb_dops(ndo, dop,
+		        GET_BE_U_4(dop->pd_sseq),
+		        GET_BE_U_4(dop->pd_eseq));
 	return (0);
 }
 
@@ -375,20 +364,20 @@ wb_drawop(netdissect_options *ndo,
           const struct pkt_dop *dop, u_int len)
 {
 	ND_PRINT(" wb-dop:");
-	if (len < sizeof(*dop) || !ND_TTEST_SIZE(dop))
+	if (len < sizeof(*dop))
 		return (-1);
 	len -= sizeof(*dop);
 
 	ND_PRINT(" %s:%u<%u:%u>",
-	    ipaddr_string(ndo, dop->pd_page.p_sid),
-	    EXTRACT_BE_U_4(dop->pd_page.p_uid),
-	    EXTRACT_BE_U_4(dop->pd_sseq),
-	    EXTRACT_BE_U_4(dop->pd_eseq));
+	    GET_IPADDR_STRING(dop->pd_page.p_sid),
+	    GET_BE_U_4(dop->pd_page.p_uid),
+	    GET_BE_U_4(dop->pd_sseq),
+	    GET_BE_U_4(dop->pd_eseq));
 
 	if (ndo->ndo_vflag)
-		return (wb_dops(ndo, dop,
-				EXTRACT_BE_U_4(dop->pd_sseq),
-				EXTRACT_BE_U_4(dop->pd_eseq)));
+		wb_dops(ndo, dop,
+		        GET_BE_U_4(dop->pd_sseq),
+		        GET_BE_U_4(dop->pd_eseq));
 	return (0);
 }
 
@@ -401,18 +390,18 @@ wb_print(netdissect_options *ndo,
 {
 	const struct pkt_hdr *ph;
 	uint8_t type;
+	int print_result;
 
 	ndo->ndo_protocol = "wb";
 	ph = (const struct pkt_hdr *)hdr;
-	if (len < sizeof(*ph) || !ND_TTEST_SIZE(ph)) {
-		ND_PRINT("%s", tstr);
-		return;
-	}
+	if (len < sizeof(*ph))
+		goto invalid;
+	ND_TCHECK_SIZE(ph);
 	len -= sizeof(*ph);
 
-	if (EXTRACT_U_1(ph->ph_flags))
+	if (GET_U_1(ph->ph_flags))
 		ND_PRINT("*");
-	type = EXTRACT_U_1(ph->ph_type);
+	type = GET_U_1(ph->ph_type);
 	switch (type) {
 
 	case PT_KILL:
@@ -420,43 +409,37 @@ wb_print(netdissect_options *ndo,
 		return;
 
 	case PT_ID:
-		if (wb_id(ndo, (const struct pkt_id *)(ph + 1), len) >= 0)
-			return;
-		ND_PRINT("%s", tstr);
+		print_result = wb_id(ndo, (const struct pkt_id *)(ph + 1), len);
 		break;
 
 	case PT_RREQ:
-		if (wb_rreq(ndo, (const struct pkt_rreq *)(ph + 1), len) >= 0)
-			return;
-		ND_PRINT("%s", tstr);
+		print_result = wb_rreq(ndo, (const struct pkt_rreq *)(ph + 1), len);
 		break;
 
 	case PT_RREP:
-		if (wb_rrep(ndo, (const struct pkt_rrep *)(ph + 1), len) >= 0)
-			return;
-		ND_PRINT("%s", tstr);
+		print_result = wb_rrep(ndo, (const struct pkt_rrep *)(ph + 1), len);
 		break;
 
 	case PT_DRAWOP:
-		if (wb_drawop(ndo, (const struct pkt_dop *)(ph + 1), len) >= 0)
-			return;
-		ND_PRINT("%s", tstr);
+		print_result = wb_drawop(ndo, (const struct pkt_dop *)(ph + 1), len);
 		break;
 
 	case PT_PREQ:
-		if (wb_preq(ndo, (const struct pkt_preq *)(ph + 1), len) >= 0)
-			return;
-		ND_PRINT("%s", tstr);
+		print_result = wb_preq(ndo, (const struct pkt_preq *)(ph + 1), len);
 		break;
 
 	case PT_PREP:
-		if (wb_prep(ndo, (const struct pkt_prep *)(ph + 1), len) >= 0)
-			return;
-		ND_PRINT("%s", tstr);
+		print_result = wb_prep(ndo, (const struct pkt_prep *)(ph + 1), len);
 		break;
 
 	default:
 		ND_PRINT(" wb-%u!", type);
-		return;
+		print_result = -1;
 	}
+	if (print_result < 0)
+		goto invalid;
+	return;
+
+invalid:
+	nd_print_invalid(ndo);
 }