Use more the EXTRACT_U_1() macro (40/n)

[tcpdump] / print-cnfp.c

diff --git a/print-cnfp.c b/print-cnfp.c
index 025b78c5c061fe799eb7ca7c7f801bafe0516baf..162b1f9f149fcb9a869ae7d63ecf6f8054a76ecd 100644 (file)
--- a/print-cnfp.c
+++ b/print-cnfp.c
@@ -30,135 +30,445 @@
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 

-/* Cisco NetFlow protocol */
+/* \summary: Cisco NetFlow protocol printer */

 
 

-#include <sys/types.h>
-#include <sys/time.h>
-#include <sys/socket.h>
-#include <netdb.h>
+/*
+ * Cisco NetFlow protocol
+ *
+ * See
+ *
+ *    http://www.cisco.com/c/en/us/td/docs/net_mgmt/netflow_collection_engine/3-6/user/guide/format.html#wp1005892
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif

 
 

-#include <netinet/in.h>
-#include <netinet/tcp.h>
+#include <netdissect-stdinc.h>

 
 #include <stdio.h>
 #include <string.h>
 
 
 #include <stdio.h>
 #include <string.h>
 

-#include "interface.h"
+#include "netdissect.h"
+#include "addrtoname.h"
+#include "extract.h"

 
 

-struct nfhdr {
-       u_int32_t       ver_cnt;        /* version [15], and # of records */
-       u_int32_t       msys_uptime;
-       u_int32_t       utc_sec;
-       u_int32_t       utc_nsec;
-       u_int32_t       sequence;       /* v5 flow sequence number */
-       u_int32_t       reserved;       /* v5 only */
+#include "tcp.h"
+#include "ipproto.h"
+
+struct nfhdr_v1 {
+       uint16_t        version;        /* version number */
+       uint16_t        count;          /* # of records */
+       uint32_t        msys_uptime;
+       uint32_t        utc_sec;
+       uint32_t        utc_nsec;

 };
 
 };
 

-struct nfrec {
-       in_addr_t       src_ina;
-       in_addr_t       dst_ina;
-       in_addr_t       nhop_ina;
-       u_int32_t       ifaces;         /* src,dst ifaces */
-       u_int32_t       packets;
-       u_int32_t       octets;
-       u_int32_t       start_time;     /* sys_uptime value */
-       u_int32_t       last_time;      /* sys_uptime value */
-       u_int32_t       ports;          /* src,dst ports */
-       u_int32_t       proto_tos;      /* proto, tos, pad, flags(v5) */
-       u_int32_t       asses;          /* v1: flags; v5: src,dst AS */
-       u_int32_t       masks;          /* src,dst addr prefix */
+struct nfrec_v1 {
+       struct in_addr  src_ina;
+       struct in_addr  dst_ina;
+       struct in_addr  nhop_ina;
+       uint16_t        input;          /* SNMP index of input interface */
+       uint16_t        output;         /* SNMP index of output interface */
+       uint32_t        packets;        /* packets in the flow */
+       uint32_t        octets;         /* layer 3 octets in the packets of the flow */
+       uint32_t        start_time;     /* sys_uptime value at start of flow */
+       uint32_t        last_time;      /* sys_uptime value when last packet of flow was received */
+       uint16_t        srcport;        /* TCP/UDP source port or equivalent */
+       uint16_t        dstport;        /* TCP/UDP source port or equivalent */
+       uint16_t        pad1;           /* pad */
+       uint8_t         proto;          /* IP protocol type */
+       uint8_t         tos;            /* IP type of service */
+       uint8_t         tcp_flags;      /* cumulative OR of TCP flags */
+       uint8_t         pad[3];         /* padding */
+       uint32_t        reserved;       /* unused */
+};

 
 

+struct nfhdr_v5 {
+       uint16_t        version;        /* version number */
+       uint16_t        count;          /* # of records */
+       uint32_t        msys_uptime;
+       uint32_t        utc_sec;
+       uint32_t        utc_nsec;
+       uint32_t        sequence;       /* flow sequence number */
+       uint8_t         engine_type;    /* type of flow-switching engine */
+       uint8_t         engine_id;      /* slot number of the flow-switching engine */
+       uint16_t        sampling_interval; /* sampling mode and interval */

 };
 
 };
 

-void
-cnfp_print(const u_char *cp, u_int len, const u_char *bp)
+struct nfrec_v5 {
+       struct in_addr  src_ina;
+       struct in_addr  dst_ina;
+       struct in_addr  nhop_ina;
+       uint16_t        input;          /* SNMP index of input interface */
+       uint16_t        output;         /* SNMP index of output interface */
+       uint32_t        packets;        /* packets in the flow */
+       uint32_t        octets;         /* layer 3 octets in the packets of the flow */
+       uint32_t        start_time;     /* sys_uptime value at start of flow */
+       uint32_t        last_time;      /* sys_uptime value when last packet of flow was received */
+       uint16_t        srcport;        /* TCP/UDP source port or equivalent */
+       uint16_t        dstport;        /* TCP/UDP source port or equivalent */
+       uint8_t         pad1;           /* pad */
+       uint8_t         tcp_flags;      /* cumulative OR of TCP flags */
+       uint8_t         proto;          /* IP protocol type */
+       uint8_t         tos;            /* IP type of service */
+       uint16_t        src_as;         /* AS number of the source */
+       uint16_t        dst_as;         /* AS number of the destination */
+       uint8_t         src_mask;       /* source address mask bits */
+       uint8_t         dst_mask;       /* destination address prefix mask bits */
+       uint16_t        pad2;
+       struct in_addr  peer_nexthop;   /* v6: IP address of the nexthop within the peer (FIB)*/
+};
+
+struct nfhdr_v6 {
+       uint16_t        version;        /* version number */
+       uint16_t        count;          /* # of records */
+       uint32_t        msys_uptime;
+       uint32_t        utc_sec;
+       uint32_t        utc_nsec;
+       uint32_t        sequence;       /* v5 flow sequence number */
+       uint32_t        reserved;       /* v5 only */
+};
+
+struct nfrec_v6 {
+       struct in_addr  src_ina;
+       struct in_addr  dst_ina;
+       struct in_addr  nhop_ina;
+       uint16_t        input;          /* SNMP index of input interface */
+       uint16_t        output;         /* SNMP index of output interface */
+       uint32_t        packets;        /* packets in the flow */
+       uint32_t        octets;         /* layer 3 octets in the packets of the flow */
+       uint32_t        start_time;     /* sys_uptime value at start of flow */
+       uint32_t        last_time;      /* sys_uptime value when last packet of flow was received */
+       uint16_t        srcport;        /* TCP/UDP source port or equivalent */
+       uint16_t        dstport;        /* TCP/UDP source port or equivalent */
+       uint8_t         pad1;           /* pad */
+       uint8_t         tcp_flags;      /* cumulative OR of TCP flags */
+       uint8_t         proto;          /* IP protocol type */
+       uint8_t         tos;            /* IP type of service */
+       uint16_t        src_as;         /* AS number of the source */
+       uint16_t        dst_as;         /* AS number of the destination */
+       uint8_t         src_mask;       /* source address mask bits */
+       uint8_t         dst_mask;       /* destination address prefix mask bits */
+       uint16_t        flags;
+       struct in_addr  peer_nexthop;   /* v6: IP address of the nexthop within the peer (FIB)*/
+};
+
+static void
+cnfp_v1_print(netdissect_options *ndo, const u_char *cp)

 {
 {

-       register const struct nfhdr *nh;
-       register const struct nfrec *nr;
-       register const struct ip *ip;
-       struct protoent *pent;
+       register const struct nfhdr_v1 *nh;
+       register const struct nfrec_v1 *nr;
+       const char *p_name;

        int nrecs, ver;
        int nrecs, ver;

+#if 0

        time_t t;
        time_t t;

-       char *p;
-
-       ip = (struct ip *)bp;
-       nh = (struct nfhdr *)cp;
-
-       if ((u_char *)(nh + 1) > snapend)
-               return;
-
-       nrecs = ntohl(nh->ver_cnt) & 0xffff;
-       ver = (ntohl(nh->ver_cnt) & 0xffff0000) >> 16;
-       t = ntohl(nh->utc_sec);
-/*     (p = ctime(&t))[24] = '\0'; */
-
-       printf("NetFlow v%x, %u.%03u uptime, %u.%09u, ", ver,
-              ntohl(nh->msys_uptime)/1000, ntohl(nh->msys_uptime)%1000,
-              ntohl(nh->utc_sec), ntohl(nh->utc_nsec));
-
-       if (ver == 5) {
-               printf("#%u, ", htonl(nh->sequence));
-               nr = (struct nfrec *)&nh[1];
-               snaplen -= 24;
-       } else {
-               nr = (struct nfrec *)&nh->sequence;
-               snaplen -= 16;
-       }
+#endif

 
 

-       printf("%2u recs", nrecs);
+       nh = (const struct nfhdr_v1 *)cp;
+       ND_TCHECK(*nh);

 
 

-       for (; nrecs-- && (u_char *)(nr + 1) <= snapend; nr++) {
-               char buf[5];
-               char asbuf[7];
+       ver = EXTRACT_BE_U_2(&nh->version);
+       nrecs = EXTRACT_BE_U_4(&nh->count);
+#if 0
+       /*
+        * This is seconds since the UN*X epoch, and is followed by
+        * nanoseconds.  XXX - format it, rather than just dumping the
+        * raw seconds-since-the-Epoch.
+        */
+       t = EXTRACT_BE_U_4(&nh->utc_sec);
+#endif

 
 

-               printf("\n  started %u.%03u, last %u.%03u",
-                       ntohl(nr->start_time)/1000, ntohl(nr->start_time)%1000,
-                       ntohl(nr->last_time)/1000, ntohl(nr->last_time)%1000);
+       ND_PRINT((ndo, "NetFlow v%x, %u.%03u uptime, %u.%09u, ", ver,
+              EXTRACT_BE_U_4(&nh->msys_uptime)/1000,
+              EXTRACT_BE_U_4(&nh->msys_uptime)%1000,
+              EXTRACT_BE_U_4(&nh->utc_sec), EXTRACT_BE_U_4(&nh->utc_nsec)));
+
+       nr = (const struct nfrec_v1 *)&nh[1];
+
+       ND_PRINT((ndo, "%2u recs", nrecs));
+
+       for (; nrecs != 0; nr++, nrecs--) {
+               char buf[20];
+               char asbuf[20];
+
+               /*
+                * Make sure we have the entire record.
+                */
+               ND_TCHECK(*nr);
+               ND_PRINT((ndo, "\n  started %u.%03u, last %u.%03u",
+                      EXTRACT_BE_U_4(&nr->start_time)/1000,
+                      EXTRACT_BE_U_4(&nr->start_time)%1000,
+                      EXTRACT_BE_U_4(&nr->last_time)/1000,
+                      EXTRACT_BE_U_4(&nr->last_time)%1000));

 
                asbuf[0] = buf[0] = '\0';
 
                asbuf[0] = buf[0] = '\0';

-               if (ver == 5) {
-                       sprintf(buf, "/%d", (ntohl(nr->masks) >> 24) & 0xff);
-                       sprintf(asbuf, "%d:", (ntohl(nr->asses) >> 16) & 0xffff);
-               }
-               printf("\n    %s%s%s:%u ", inet_ntoa(nr->src_ina), buf, asbuf,
-                       ntohl(nr->ports) >> 16);
+               ND_PRINT((ndo, "\n    %s%s%s:%u ", intoa(nr->src_ina.s_addr), buf, asbuf,
+                       EXTRACT_BE_U_2(&nr->srcport)));

 
 

-               if (ver == 5) {
-                       sprintf(buf, "/%d", (ntohl(nr->masks) >> 16) & 0xff);
-                       sprintf(asbuf, "%d:", ntohl(nr->asses) & 0xffff);
+               ND_PRINT((ndo, "> %s%s%s:%u ", intoa(nr->dst_ina.s_addr), buf, asbuf,
+                       EXTRACT_BE_U_2(&nr->dstport)));
+
+               ND_PRINT((ndo, ">> %s\n    ", intoa(nr->nhop_ina.s_addr)));
+
+               if (!ndo->ndo_nflag && (p_name = netdb_protoname(nr->proto)) != NULL)
+                       ND_PRINT((ndo, "%s ", p_name));
+               else
+                       ND_PRINT((ndo, "%u ", nr->proto));
+
+               /* tcp flags for tcp only */
+               if (nr->proto == IPPROTO_TCP) {
+                       int flags;
+                       flags = nr->tcp_flags;
+                       ND_PRINT((ndo, "%s%s%s%s%s%s%s",
+                               flags & TH_FIN  ? "F" : "",
+                               flags & TH_SYN  ? "S" : "",
+                               flags & TH_RST  ? "R" : "",
+                               flags & TH_PUSH ? "P" : "",
+                               flags & TH_ACK  ? "A" : "",
+                               flags & TH_URG  ? "U" : "",
+                               flags           ? " " : ""));

                }
                }

-               printf("> %s%s%s:%u ", inet_ntoa(nr->dst_ina), buf, asbuf,
-                       ntohl(nr->ports) & 0xffff);

 
 

-               printf(">> %s\n    ", inet_ntoa(nr->nhop_ina));
+               buf[0]='\0';
+               ND_PRINT((ndo, "tos %u, %u (%u octets) %s",
+                      nr->tos,
+                      EXTRACT_BE_U_4(&nr->packets),
+                      EXTRACT_BE_U_4(&nr->octets), buf));
+       }
+       return;
+
+trunc:
+       ND_PRINT((ndo, "[|cnfp]"));
+       return;
+}
+
+static void
+cnfp_v5_print(netdissect_options *ndo, const u_char *cp)
+{
+       register const struct nfhdr_v5 *nh;
+       register const struct nfrec_v5 *nr;
+       const char *p_name;
+       int nrecs, ver;
+#if 0
+       time_t t;
+#endif
+
+       nh = (const struct nfhdr_v5 *)cp;
+       ND_TCHECK(*nh);
+
+       ver = EXTRACT_BE_U_2(&nh->version);
+       nrecs = EXTRACT_BE_U_4(&nh->count);
+#if 0
+       /*
+        * This is seconds since the UN*X epoch, and is followed by
+        * nanoseconds.  XXX - format it, rather than just dumping the
+        * raw seconds-since-the-Epoch.
+        */
+       t = EXTRACT_BE_U_4(&nh->utc_sec);
+#endif
+
+       ND_PRINT((ndo, "NetFlow v%x, %u.%03u uptime, %u.%09u, ", ver,
+              EXTRACT_BE_U_4(&nh->msys_uptime)/1000,
+              EXTRACT_BE_U_4(&nh->msys_uptime)%1000,
+              EXTRACT_BE_U_4(&nh->utc_sec), EXTRACT_BE_U_4(&nh->utc_nsec)));
+
+       ND_PRINT((ndo, "#%u, ", EXTRACT_BE_U_4(&nh->sequence)));
+       nr = (const struct nfrec_v5 *)&nh[1];
+
+       ND_PRINT((ndo, "%2u recs", nrecs));
+
+       for (; nrecs != 0; nr++, nrecs--) {
+               char buf[20];
+               char asbuf[20];

 
 

-               pent = getprotobynumber((ntohl(nr->proto_tos) >> 8) & 0xff);
-               if (!pent || nflag)
-                       printf("%u ", (ntohl(nr->proto_tos) >> 8) & 0xff);
+               /*
+                * Make sure we have the entire record.
+                */
+               ND_TCHECK(*nr);
+               ND_PRINT((ndo, "\n  started %u.%03u, last %u.%03u",
+                      EXTRACT_BE_U_4(&nr->start_time)/1000,
+                      EXTRACT_BE_U_4(&nr->start_time)%1000,
+                      EXTRACT_BE_U_4(&nr->last_time)/1000,
+                      EXTRACT_BE_U_4(&nr->last_time)%1000));
+
+               asbuf[0] = buf[0] = '\0';
+               snprintf(buf, sizeof(buf), "/%u", nr->src_mask);
+               snprintf(asbuf, sizeof(asbuf), ":%u",
+                       EXTRACT_BE_U_2(&nr->src_as));
+               ND_PRINT((ndo, "\n    %s%s%s:%u ", intoa(nr->src_ina.s_addr), buf, asbuf,
+                       EXTRACT_BE_U_2(&nr->srcport)));
+
+               snprintf(buf, sizeof(buf), "/%d", nr->dst_mask);
+               snprintf(asbuf, sizeof(asbuf), ":%u",
+                        EXTRACT_BE_U_2(&nr->dst_as));
+               ND_PRINT((ndo, "> %s%s%s:%u ", intoa(nr->dst_ina.s_addr), buf, asbuf,
+                       EXTRACT_BE_U_2(&nr->dstport)));
+
+               ND_PRINT((ndo, ">> %s\n    ", intoa(nr->nhop_ina.s_addr)));
+
+               if (!ndo->ndo_nflag && (p_name = netdb_protoname(nr->proto)) != NULL)
+                       ND_PRINT((ndo, "%s ", p_name));

                else
                else

-                       printf("%s ", pent->p_name);
+                       ND_PRINT((ndo, "%u ", nr->proto));

 
                /* tcp flags for tcp only */
 
                /* tcp flags for tcp only */

-               if (pent && pent->p_proto == IPPROTO_TCP) {
+               if (nr->proto == IPPROTO_TCP) {

                        int flags;
                        int flags;

-                       if (ver == 1)
-                               flags = (ntohl(nr->asses) >> 24) & 0xff;
-                       else
-                               flags = (ntohl(nr->proto_tos) >> 16) & 0xff;
-                       if (flags & TH_FIN)     putchar('F');
-                       if (flags & TH_SYN)     putchar('S');
-                       if (flags & TH_RST)     putchar('R');
-                       if (flags & TH_PUSH)    putchar('P');
-                       if (flags & TH_ACK)     putchar('A');
-                       if (flags & TH_URG)     putchar('U');
-                       if (flags)
-                               putchar(' ');
+                       flags = nr->tcp_flags;
+                       ND_PRINT((ndo, "%s%s%s%s%s%s%s",
+                               flags & TH_FIN  ? "F" : "",
+                               flags & TH_SYN  ? "S" : "",
+                               flags & TH_RST  ? "R" : "",
+                               flags & TH_PUSH ? "P" : "",
+                               flags & TH_ACK  ? "A" : "",
+                               flags & TH_URG  ? "U" : "",
+                               flags           ? " " : ""));

                }
                }

-               printf("tos %u, %u (%u octets)", ntohl(nr->proto_tos) & 0xff,
-                       ntohl(nr->packets), ntohl(nr->octets));

 
 

+               buf[0]='\0';
+               ND_PRINT((ndo, "tos %u, %u (%u octets) %s",
+                      nr->tos,
+                      EXTRACT_BE_U_4(&nr->packets),
+                      EXTRACT_BE_U_4(&nr->octets), buf));
+       }
+       return;
+
+trunc:
+       ND_PRINT((ndo, "[|cnfp]"));
+       return;
+}
+
+static void
+cnfp_v6_print(netdissect_options *ndo, const u_char *cp)
+{
+       register const struct nfhdr_v6 *nh;
+       register const struct nfrec_v6 *nr;
+       const char *p_name;
+       int nrecs, ver;
+#if 0
+       time_t t;
+#endif
+
+       nh = (const struct nfhdr_v6 *)cp;
+       ND_TCHECK(*nh);
+
+       ver = EXTRACT_BE_U_2(&nh->version);
+       nrecs = EXTRACT_BE_U_4(&nh->count);
+#if 0
+       /*
+        * This is seconds since the UN*X epoch, and is followed by
+        * nanoseconds.  XXX - format it, rather than just dumping the
+        * raw seconds-since-the-Epoch.
+        */
+       t = EXTRACT_BE_U_4(&nh->utc_sec);
+#endif
+
+       ND_PRINT((ndo, "NetFlow v%x, %u.%03u uptime, %u.%09u, ", ver,
+              EXTRACT_BE_U_4(&nh->msys_uptime)/1000,
+              EXTRACT_BE_U_4(&nh->msys_uptime)%1000,
+              EXTRACT_BE_U_4(&nh->utc_sec), EXTRACT_BE_U_4(&nh->utc_nsec)));

 
 

+       ND_PRINT((ndo, "#%u, ", EXTRACT_BE_U_4(&nh->sequence)));
+       nr = (const struct nfrec_v6 *)&nh[1];
+
+       ND_PRINT((ndo, "%2u recs", nrecs));
+
+       for (; nrecs != 0; nr++, nrecs--) {
+               char buf[20];
+               char asbuf[20];
+
+               /*
+                * Make sure we have the entire record.
+                */
+               ND_TCHECK(*nr);
+               ND_PRINT((ndo, "\n  started %u.%03u, last %u.%03u",
+                      EXTRACT_BE_U_4(&nr->start_time)/1000,
+                      EXTRACT_BE_U_4(&nr->start_time)%1000,
+                      EXTRACT_BE_U_4(&nr->last_time)/1000,
+                      EXTRACT_BE_U_4(&nr->last_time)%1000));
+
+               asbuf[0] = buf[0] = '\0';
+               snprintf(buf, sizeof(buf), "/%u", nr->src_mask);
+               snprintf(asbuf, sizeof(asbuf), ":%u",
+                       EXTRACT_BE_U_2(&nr->src_as));
+               ND_PRINT((ndo, "\n    %s%s%s:%u ", intoa(nr->src_ina.s_addr), buf, asbuf,
+                       EXTRACT_BE_U_2(&nr->srcport)));
+
+               snprintf(buf, sizeof(buf), "/%d", nr->dst_mask);
+               snprintf(asbuf, sizeof(asbuf), ":%u",
+                        EXTRACT_BE_U_2(&nr->dst_as));
+               ND_PRINT((ndo, "> %s%s%s:%u ", intoa(nr->dst_ina.s_addr), buf, asbuf,
+                       EXTRACT_BE_U_2(&nr->dstport)));
+
+               ND_PRINT((ndo, ">> %s\n    ", intoa(nr->nhop_ina.s_addr)));
+
+               if (!ndo->ndo_nflag && (p_name = netdb_protoname(nr->proto)) != NULL)
+                       ND_PRINT((ndo, "%s ", p_name));
+               else
+                       ND_PRINT((ndo, "%u ", nr->proto));
+
+               /* tcp flags for tcp only */
+               if (nr->proto == IPPROTO_TCP) {
+                       int flags;
+                       flags = nr->tcp_flags;
+                       ND_PRINT((ndo, "%s%s%s%s%s%s%s",
+                               flags & TH_FIN  ? "F" : "",
+                               flags & TH_SYN  ? "S" : "",
+                               flags & TH_RST  ? "R" : "",
+                               flags & TH_PUSH ? "P" : "",
+                               flags & TH_ACK  ? "A" : "",
+                               flags & TH_URG  ? "U" : "",
+                               flags           ? " " : ""));
+               }
+
+               buf[0]='\0';
+               snprintf(buf, sizeof(buf), "(%u<>%u encaps)",
+                        (EXTRACT_BE_U_2(&nr->flags) >> 8) & 0xff,
+                        (EXTRACT_BE_U_2(&nr->flags)) & 0xff);
+               ND_PRINT((ndo, "tos %u, %u (%u octets) %s",
+                      nr->tos,
+                      EXTRACT_BE_U_4(&nr->packets),
+                      EXTRACT_BE_U_4(&nr->octets), buf));

        }
        }

+       return;

 
 

+trunc:
+       ND_PRINT((ndo, "[|cnfp]"));
+       return;

 }
 
 }
 

+void
+cnfp_print(netdissect_options *ndo, const u_char *cp)
+{
+       int ver;
+
+       /*
+        * First 2 bytes are the version number.
+        */
+       ND_TCHECK_2(cp);
+       ver = EXTRACT_BE_U_2(cp);
+       switch (ver) {
+
+       case 1:
+               cnfp_v1_print(ndo, cp);
+               break;
+
+       case 5:
+               cnfp_v5_print(ndo, cp);
+               break;
+
+       case 6:
+               cnfp_v6_print(ndo, cp);
+               break;
+
+       default:
+               ND_PRINT((ndo, "NetFlow v%x", ver));
+               break;
+       }
+       return;
+
+trunc:
+       ND_PRINT((ndo, "[|cnfp]"));
+       return;
+}