SMB: Add two missing bounds checks

[tcpdump] / print-mpls.c

diff --git a/print-mpls.c b/print-mpls.c
index e45f2826ffb33db9c9fd063ba2adba14a4240de9..bdc66c642360bc7710da437abd20a04f744509cc 100644 (file)
--- a/print-mpls.c
+++ b/print-mpls.c
@@ -69,11 +69,9 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
        ND_PRINT("MPLS");
        do {
                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);
+               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));
@@ -135,7 +133,7 @@ mpls_print(netdissect_options *ndo, const u_char *bp, u_int length)
                        /* nothing to print */
                        return;
                }
-               switch(EXTRACT_U_1(p)) {
+               switch(GET_U_1(p)) {
 
                case 0x45:
                case 0x46: