Handle very large -f files by rejecting them.

[tcpdump] / print-frag6.c

diff --git a/print-frag6.c b/print-frag6.c
index 0cc99d50e775633d608f2656badae0c8f990cc76..fbcabc5b00cdc82a97ca76e7f15ca94823f40561 100644 (file)
--- a/print-frag6.c
+++ b/print-frag6.c
@@ -19,23 +19,21 @@
  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  */
 
+/* \summary: IPv6 fragmentation header printer */
+
 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif
 
-#ifdef INET6
-
-#include <tcpdump-stdinc.h>
+#include <netdissect-stdinc.h>
 
-#include <stdio.h>
+#include "netdissect.h"
+#include "extract.h"
 
 #include "ip6.h"
 
-#include "interface.h"
-#include "extract.h"
-
 int
-frag6_print(register const u_char *bp, register const u_char *bp2)
+frag6_print(netdissect_options *ndo, register const u_char *bp, register const u_char *bp2)
 {
        register const struct ip6_frag *dp;
        register const struct ip6_hdr *ip6;
@@ -43,34 +41,30 @@ frag6_print(register const u_char *bp, register const u_char *bp2)
        dp = (const struct ip6_frag *)bp;
        ip6 = (const struct ip6_hdr *)bp2;
 
-       TCHECK(dp->ip6f_offlg);
+       ND_TCHECK(*dp);
 
-       if (vflag) {
-               printf("frag (0x%08x:%d|%ld)",
+       if (ndo->ndo_vflag) {
+               ND_PRINT((ndo, "frag (0x%08x:%d|%ld)",
                       EXTRACT_32BITS(&dp->ip6f_ident),
                       EXTRACT_16BITS(&dp->ip6f_offlg) & IP6F_OFF_MASK,
                       sizeof(struct ip6_hdr) + EXTRACT_16BITS(&ip6->ip6_plen) -
-                              (long)(bp - bp2) - sizeof(struct ip6_frag));
+                              (long)(bp - bp2) - sizeof(struct ip6_frag)));
        } else {
-               printf("frag (%d|%ld)",
+               ND_PRINT((ndo, "frag (%d|%ld)",
                       EXTRACT_16BITS(&dp->ip6f_offlg) & IP6F_OFF_MASK,
                       sizeof(struct ip6_hdr) + EXTRACT_16BITS(&ip6->ip6_plen) -
-                              (long)(bp - bp2) - sizeof(struct ip6_frag));
+                              (long)(bp - bp2) - sizeof(struct ip6_frag)));
        }
 
-#if 1
        /* it is meaningless to decode non-first fragment */
        if ((EXTRACT_16BITS(&dp->ip6f_offlg) & IP6F_OFF_MASK) != 0)
                return -1;
        else
-#endif
        {
-               fputs(" ", stdout);
+               ND_PRINT((ndo, " "));
                return sizeof(struct ip6_frag);
        }
 trunc:
-       fputs("[|frag]", stdout);
+       ND_PRINT((ndo, "[|frag]"));
        return -1;
-#undef TCHECK
 }
-#endif /* INET6 */