Fix spaces

[tcpdump] / print-zeromq.c

diff --git a/print-zeromq.c b/print-zeromq.c
index e14106749af9ed01d6ab2f8242bdbcba0ea4ffb9..6e5e9dc8106a3f25d1624c9b4be1679f9c002d98 100644 (file)
--- a/print-zeromq.c
+++ b/print-zeromq.c
@@ -1,7 +1,4 @@
 /*
 /*

- * This file implements decoding of ZeroMQ network protocol(s).
- *
- *

  * Copyright (c) 2013 The TCPDUMP project
  * All rights reserved.
  *
  * Copyright (c) 2013 The TCPDUMP project
  * All rights reserved.
  *


@@ -28,16 +25,17 @@
  * POSSIBILITY OF SUCH DAMAGE.
  */
 
  * POSSIBILITY OF SUCH DAMAGE.
  */
 

+/* \summary: ZeroMQ Message Transport Protocol (ZMTP) 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"
 

-static const char tstr[] = " [|zmtp1]";

 
 /* Maximum number of ZMTP/1.0 frame body bytes (without the flags) to dump in
  * hex and ASCII under a single "-v" flag.
 
 /* Maximum number of ZMTP/1.0 frame body bytes (without the flags) to dump in
  * hex and ASCII under a single "-v" flag.


@@ -80,34 +78,34 @@ zmtp1_print_frame(netdissect_options *ndo, const u_char *cp, const u_char *ep)
        uint64_t body_len_declared, body_len_captured, header_len;
        uint8_t flags;
 
        uint64_t body_len_declared, body_len_captured, header_len;
        uint8_t flags;
 

-       ND_PRINT((ndo, "\n\t"));
-       ND_TCHECK2(*cp, 1); /* length/0xFF */
+       ND_PRINT("\n\t");
+       ND_TCHECK_1(cp); /* length/0xFF */

 
 

-       if (cp[0] != 0xFF) {
+       if (EXTRACT_U_1(cp) != 0xFF) {

                header_len = 1; /* length */
                header_len = 1; /* length */

-               body_len_declared = cp[0];
-               ND_PRINT((ndo, " frame flags+body  (8-bit) length %u", body_len_declared));
+               body_len_declared = EXTRACT_U_1(cp);
+               ND_PRINT(" frame flags+body  (8-bit) length %" PRIu64, body_len_declared);

        } else {
                header_len = 1 + 8; /* 0xFF, length */
        } else {
                header_len = 1 + 8; /* 0xFF, length */

-               ND_PRINT((ndo, " frame flags+body (64-bit) length"));
-               ND_TCHECK2(*cp, header_len); /* 0xFF, length */
-               body_len_declared = EXTRACT_64BITS(cp + 1);
-               ND_PRINT((ndo, " %" PRIu64, body_len_declared));
+               ND_PRINT(" frame flags+body (64-bit) length");
+               ND_TCHECK_LEN(cp, header_len); /* 0xFF, length */
+               body_len_declared = EXTRACT_BE_U_8(cp + 1);
+               ND_PRINT(" %" PRIu64, body_len_declared);

        }
        if (body_len_declared == 0)
                return cp + header_len; /* skip to the next frame */
        }
        if (body_len_declared == 0)
                return cp + header_len; /* skip to the next frame */

-       ND_TCHECK2(*cp, header_len + 1); /* ..., flags */
-       flags = cp[header_len];
+       ND_TCHECK_LEN(cp, header_len + 1); /* ..., flags */
+       flags = EXTRACT_U_1(cp + header_len);

 
        body_len_captured = ep - cp - header_len;
        if (body_len_declared > body_len_captured)
 
        body_len_captured = ep - cp - header_len;
        if (body_len_declared > body_len_captured)

-               ND_PRINT((ndo, " (%" PRIu64 " captured)", body_len_captured));
-       ND_PRINT((ndo, ", flags 0x%02x", flags));
+               ND_PRINT(" (%" PRIu64 " captured)", body_len_captured);
+       ND_PRINT(", flags 0x%02x", flags);

 
        if (ndo->ndo_vflag) {
                uint64_t body_len_printed = min(body_len_captured, body_len_declared);
 
 
        if (ndo->ndo_vflag) {
                uint64_t body_len_printed = min(body_len_captured, body_len_declared);
 

-               ND_PRINT((ndo, " (%s|%s|%s|%s|%s|%s|%s|%s)",
+               ND_PRINT(" (%s|%s|%s|%s|%s|%s|%s|%s)",

                        flags & 0x80 ? "MBZ" : "-",
                        flags & 0x40 ? "MBZ" : "-",
                        flags & 0x20 ? "MBZ" : "-",
                        flags & 0x80 ? "MBZ" : "-",
                        flags & 0x40 ? "MBZ" : "-",
                        flags & 0x20 ? "MBZ" : "-",


@@ -115,22 +113,29 @@ zmtp1_print_frame(netdissect_options *ndo, const u_char *cp, const u_char *ep)
                        flags & 0x08 ? "MBZ" : "-",
                        flags & 0x04 ? "MBZ" : "-",
                        flags & 0x02 ? "MBZ" : "-",
                        flags & 0x08 ? "MBZ" : "-",
                        flags & 0x04 ? "MBZ" : "-",
                        flags & 0x02 ? "MBZ" : "-",

-                       flags & 0x01 ? "MORE" : "-"));
+                       flags & 0x01 ? "MORE" : "-");

 
                if (ndo->ndo_vflag == 1)
                        body_len_printed = min(VBYTES + 1, body_len_printed);
                if (body_len_printed > 1) {
 
                if (ndo->ndo_vflag == 1)
                        body_len_printed = min(VBYTES + 1, body_len_printed);
                if (body_len_printed > 1) {

-                       ND_PRINT((ndo, ", first %" PRIu64 " byte(s) of body:", body_len_printed - 1));
+                       ND_PRINT(", first %" PRIu64 " byte(s) of body:", body_len_printed - 1);

                        hex_and_ascii_print(ndo, "\n\t ", cp + header_len + 1, body_len_printed - 1);
                        hex_and_ascii_print(ndo, "\n\t ", cp + header_len + 1, body_len_printed - 1);

-                       ND_PRINT((ndo, "\n"));
+                       ND_PRINT("\n");

                }
        }
 
                }
        }
 

-       ND_TCHECK2(*cp, header_len + body_len_declared); /* Next frame within the buffer ? */
-       return cp + header_len + body_len_declared;
+       /*
+        * Do not advance cp by the sum of header_len and body_len_declared
+        * before each offset has successfully passed ND_TCHECK_LEN() as the
+        * sum can roll over (9 + 0xfffffffffffffff7 = 0) and cause an
+        * infinite loop.
+        */
+       cp += header_len;
+       ND_TCHECK_LEN(cp, body_len_declared); /* Next frame within the buffer ? */
+       return cp + body_len_declared;

 
 trunc:
 
 trunc:

-       ND_PRINT((ndo, "%s", tstr));
+       nd_print_trunc(ndo);

        return ep;
 }
 
        return ep;
 }
 


@@ -139,7 +144,8 @@ zmtp1_print(netdissect_options *ndo, const u_char *cp, u_int len)
 {
        const u_char *ep = min(ndo->ndo_snapend, cp + len);
 
 {
        const u_char *ep = min(ndo->ndo_snapend, cp + len);
 

-       ND_PRINT((ndo, ": ZMTP/1.0"));
+       ndo->ndo_protocol = "zmtp1";
+       ND_PRINT(": ZMTP/1.0");

        while (cp < ep)
                cp = zmtp1_print_frame(ndo, cp, ep);
 }
        while (cp < ep)
                cp = zmtp1_print_frame(ndo, cp, ep);
 }


@@ -166,48 +172,49 @@ zmtp1_print_intermediate_part(netdissect_options *ndo, const u_char *cp, const u
        u_int frame_offset;
        uint64_t remaining_len;
 
        u_int frame_offset;
        uint64_t remaining_len;
 

-       ND_TCHECK2(*cp, 2);
-       frame_offset = EXTRACT_16BITS(cp);
-       ND_PRINT((ndo, "\n\t frame offset 0x%04x", frame_offset));
+       ND_TCHECK_2(cp);
+       frame_offset = EXTRACT_BE_U_2(cp);
+       ND_PRINT("\n\t frame offset 0x%04x", frame_offset);

        cp += 2;
        remaining_len = ndo->ndo_snapend - cp; /* without the frame length */
 
        if (frame_offset == 0xFFFF)
                frame_offset = len - 2; /* always within the declared length */
        else if (2 + frame_offset > len) {
        cp += 2;
        remaining_len = ndo->ndo_snapend - cp; /* without the frame length */
 
        if (frame_offset == 0xFFFF)
                frame_offset = len - 2; /* always within the declared length */
        else if (2 + frame_offset > len) {

-               ND_PRINT((ndo, " (exceeds datagram declared length)"));
+               ND_PRINT(" (exceeds datagram declared length)");

                goto trunc;
        }
 
        /* offset within declared length of the datagram */
        if (frame_offset) {
                goto trunc;
        }
 
        /* offset within declared length of the datagram */
        if (frame_offset) {

-               ND_PRINT((ndo, "\n\t frame intermediate part, %u bytes", frame_offset));
+               ND_PRINT("\n\t frame intermediate part, %u bytes", frame_offset);

                if (frame_offset > remaining_len)
                if (frame_offset > remaining_len)

-                       ND_PRINT((ndo, " (%"PRIu64" captured)", remaining_len));
+                       ND_PRINT(" (%"PRIu64" captured)", remaining_len);

                if (ndo->ndo_vflag) {
                        uint64_t len_printed = min(frame_offset, remaining_len);
 
                        if (ndo->ndo_vflag == 1)
                                len_printed = min(VBYTES, len_printed);
                        if (len_printed > 1) {
                if (ndo->ndo_vflag) {
                        uint64_t len_printed = min(frame_offset, remaining_len);
 
                        if (ndo->ndo_vflag == 1)
                                len_printed = min(VBYTES, len_printed);
                        if (len_printed > 1) {

-                               ND_PRINT((ndo, ", first %"PRIu64" byte(s):", len_printed));
+                               ND_PRINT(", first %"PRIu64" byte(s):", len_printed);

                                hex_and_ascii_print(ndo, "\n\t ", cp, len_printed);
                                hex_and_ascii_print(ndo, "\n\t ", cp, len_printed);

-                               ND_PRINT((ndo, "\n"));
+                               ND_PRINT("\n");

                        }
                }
        }
        return cp + frame_offset;
 
 trunc:
                        }
                }
        }
        return cp + frame_offset;
 
 trunc:

-       ND_PRINT((ndo, "%s", tstr));
+       nd_print_trunc(ndo);

        return cp + len;
 }
 
 void
        return cp + len;
 }
 
 void

-zmtp1_print_datagram(netdissect_options *ndo, const u_char *cp, const u_int len)
+zmtp1_datagram_print(netdissect_options *ndo, const u_char *cp, const u_int len)

 {
        const u_char *ep = min(ndo->ndo_snapend, cp + len);
 
 {
        const u_char *ep = min(ndo->ndo_snapend, cp + len);
 

+       ndo->ndo_protocol = "zmtp1";

        cp = zmtp1_print_intermediate_part(ndo, cp, len);
        while (cp < ep)
                cp = zmtp1_print_frame(ndo, cp, ep);
        cp = zmtp1_print_intermediate_part(ndo, cp, len);
        while (cp < ep)
                cp = zmtp1_print_frame(ndo, cp, ep);