RIPng: Use more ND_TCHECK_SIZE() macros

[tcpdump] / print-mpls.c

diff --git a/print-mpls.c b/print-mpls.c
index 504ce1975c439e67cc64ede765d050e2f0a1136f..1ade6e65d9b92b288e7a1a2159ac6de965df5ba8 100644 (file)
--- a/print-mpls.c
+++ b/print-mpls.c
@@ -26,11 +26,13 @@
  * SUCH DAMAGE.
  */
 
  * SUCH DAMAGE.
  */
 

+/* \summary: Multi-Protocol Label Switching (MPLS) printer */
+

 #ifdef HAVE_CONFIG_H
 #ifdef HAVE_CONFIG_H

-#include "config.h"
+#include <config.h>

 #endif
 
 #endif
 

-#include <netdissect-stdinc.h>
+#include "netdissect-stdinc.h"

 
 #include "netdissect.h"
 #include "extract.h"
 
 #include "netdissect.h"
 #include "extract.h"


@@ -63,23 +65,28 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
        enum mpls_packet_type pt = PT_UNKNOWN;
 
        p = bp;
        enum mpls_packet_type pt = PT_UNKNOWN;
 
        p = bp;

-       ND_PRINT((ndo, "MPLS"));
+       ND_PRINT("MPLS");

        do {
        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)) {
+                       ND_PRINT("[|MPLS], length %u", length);
+                       return;
+               }
+               label_entry = EXTRACT_BE_U_4(p);
+               ND_PRINT("%s(label %u",

                       (label_stack_depth && ndo->ndo_vflag) ? "\n\t" : " ",
                       (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]))
                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))
                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);
 
                p += sizeof(label_entry);

+               length -= sizeof(label_entry);

        } while (!MPLS_STACK(label_entry));
 
        /*
        } while (!MPLS_STACK(label_entry));
 
        /*


@@ -122,7 +129,12 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
                 * Cisco sends control-plane traffic MPLS-encapsulated in
                 * this fashion.
                 */
                 * Cisco sends control-plane traffic MPLS-encapsulated in
                 * this fashion.
                 */

-               switch(*p) {
+               ND_TCHECK_1(p);
+               if (length < 1) {
+                       /* nothing to print */
+                       return;
+               }
+               switch(EXTRACT_U_1(p)) {

 
                case 0x45:
                case 0x46:
 
                case 0x45:
                case 0x46:


@@ -174,22 +186,22 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
         */
        if (pt == PT_UNKNOWN) {
                if (!ndo->ndo_suppress_default_print)
         */
        if (pt == PT_UNKNOWN) {
                if (!ndo->ndo_suppress_default_print)

-                       ND_DEFAULTPRINT(p, length - (p - bp));
+                       ND_DEFAULTPRINT(p, length);

                return;
        }
                return;
        }

-       ND_PRINT((ndo, ndo->ndo_vflag ? "\n\t" : " "));
+       ND_PRINT(ndo->ndo_vflag ? "\n\t" : " ");

        switch (pt) {
 
        case PT_IPV4:
        switch (pt) {
 
        case PT_IPV4:

-               ip_print(ndo, p, length - (p - bp));
+               ip_print(ndo, p, length);

                break;
 
        case PT_IPV6:
                break;
 
        case PT_IPV6:

-               ip6_print(ndo, p, length - (p - bp));
+               ip6_print(ndo, p, length);

                break;
 
        case PT_OSI:
                break;
 
        case PT_OSI:

-               isoclns_print(ndo, p, length - (p - bp), length - (p - bp));
+               isoclns_print(ndo, p, length);

                break;
 
        default:
                break;
 
        default:


@@ -198,7 +210,7 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
        return;
 
 trunc:
        return;
 
 trunc:

-       ND_PRINT((ndo, "[|MPLS]"));
+       ND_PRINT("[|MPLS]");

 }
 
 
 }