X-Git-Url: https://git.tcpdump.org/tcpdump/blobdiff_plain/c1c3c77463d592cc576eaa491e604752044ca55a..10ac80fdecfa9b9b7d259d8f50d0b72ef1b18f12:/print-mpls.c

diff --git a/print-mpls.c b/print-mpls.c
index 027b6729..4351507e 100644
--- a/print-mpls.c
+++ b/print-mpls.c
@@ -26,14 +26,16 @@
  * SUCH DAMAGE.
  */
 
+/* \summary: Multi-Protocol Label Switching (MPLS) printer */
+
 #ifdef HAVE_CONFIG_H
-#include "config.h"
+#include <config.h>
 #endif
 
-#include <tcpdump-stdinc.h>
+#include "netdissect-stdinc.h"
 
 #include "netdissect.h"
-#include "extract.h"			/* must come after interface.h */
+#include "extract.h"
 #include "mpls.h"
 
 static const char *mpls_labelname[] = {
@@ -62,24 +64,28 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
 	uint16_t label_stack_depth = 0;
 	enum mpls_packet_type pt = PT_UNKNOWN;
 
+	ndo->ndo_protocol = "mpls";
 	p = bp;
-	ND_PRINT((ndo, "MPLS"));
+	nd_print_protocol_caps(ndo);
 	do {
-		ND_TCHECK2(*p, sizeof(label_entry));
-		label_entry = EXTRACT_32BITS(p);
-		ND_PRINT((ndo, "%s(label %u",
+		ND_TCHECK_LEN(p, sizeof(label_entry));
+		if (length < sizeof(label_entry))
+			goto trunc;
+		label_entry = GET_BE_U_4(p);
+		ND_PRINT("%s(label %u",
 		       (label_stack_depth && ndo->ndo_vflag) ? "\n\t" : " ",
-       		       MPLS_LABEL(label_entry)));
+       		       MPLS_LABEL(label_entry));
 		label_stack_depth++;
 		if (ndo->ndo_vflag &&
 		    MPLS_LABEL(label_entry) < sizeof(mpls_labelname) / sizeof(mpls_labelname[0]))
-			ND_PRINT((ndo, " (%s)", mpls_labelname[MPLS_LABEL(label_entry)]));
-		ND_PRINT((ndo, ", exp %u", MPLS_EXP(label_entry)));
+			ND_PRINT(" (%s)", mpls_labelname[MPLS_LABEL(label_entry)]);
+		ND_PRINT(", exp %u", MPLS_EXP(label_entry));
 		if (MPLS_STACK(label_entry))
-			ND_PRINT((ndo, ", [S]"));
-		ND_PRINT((ndo, ", ttl %u)", MPLS_TTL(label_entry)));
+			ND_PRINT(", [S]");
+		ND_PRINT(", ttl %u)", MPLS_TTL(label_entry));
 
 		p += sizeof(label_entry);
+		length -= sizeof(label_entry);
 	} while (!MPLS_STACK(label_entry));
 
 	/*
@@ -122,7 +128,12 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
 		 * Cisco sends control-plane traffic MPLS-encapsulated in
 		 * this fashion.
 		 */
-		switch(*p) {
+		ND_TCHECK_1(p);
+		if (length < 1) {
+			/* nothing to print */
+			return;
+		}
+		switch(GET_U_1(p)) {
 
 		case 0x45:
 		case 0x46:
@@ -172,39 +183,29 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
 	/*
 	 * Print the payload.
 	 */
-	if (pt == PT_UNKNOWN) {
-		if (!ndo->ndo_suppress_default_print)
-			ND_DEFAULTPRINT(p, length - (p - bp));
-		return;
-	}
-	ND_PRINT((ndo, ndo->ndo_vflag ? "\n\t" : " "));
 	switch (pt) {
+	case PT_UNKNOWN:
+		if (!ndo->ndo_suppress_default_print)
+			ND_DEFAULTPRINT(p, length);
+		break;
 
 	case PT_IPV4:
-		ip_print(ndo, p, length - (p - bp));
+		ND_PRINT(ndo->ndo_vflag ? "\n\t" : " ");
+		ip_print(ndo, p, length);
 		break;
 
 	case PT_IPV6:
-		ip6_print(ndo, p, length - (p - bp));
+		ND_PRINT(ndo->ndo_vflag ? "\n\t" : " ");
+		ip6_print(ndo, p, length);
 		break;
 
 	case PT_OSI:
-		isoclns_print(ndo, p, length - (p - bp), length - (p - bp));
-		break;
-
-	default:
+		ND_PRINT(ndo->ndo_vflag ? "\n\t" : " ");
+		isoclns_print(ndo, p, length);
 		break;
 	}
 	return;
 
 trunc:
-	ND_PRINT((ndo, "[|MPLS]"));
+	nd_print_trunc(ndo);
 }
-
-
-/*
- * Local Variables:
- * c-style: whitesmith
- * c-basic-offset: 8
- * End:
- */