]> The Tcpdump Group git mirrors - tcpdump/blobdiff - print-domain.c
Use nd_ types, add EXTRACT_ calls.
[tcpdump] / print-domain.c
index e150116d84eb8583e346fe08a1757c5dd6e490bf..8aff4e99dc1b140a98c4e3ca39b86ca9a704effe 100644 (file)
@@ -27,8 +27,6 @@
 
 #include <netdissect-stdinc.h>
 
-#include "nameser.h"
-
 #include <string.h>
 
 #include "netdissect.h"
@@ -36,6 +34,8 @@
 #include "addrtostr.h"
 #include "extract.h"
 
+#include "nameser.h"
+
 static const char *ns_ops[] = {
        "", " inv_q", " stat", " op3", " notify", " update", " op6", " op7",
        " op8", " updateA", " updateD", " updateDA",
@@ -52,9 +52,9 @@ static const char *ns_resp[] = {
 /* skip over a domain name */
 static const u_char *
 ns_nskip(netdissect_options *ndo,
-         register const u_char *cp)
+         const u_char *cp)
 {
-       register u_char i;
+       u_char i;
 
        if (!ND_TTEST_1(cp))
                return (NULL);
@@ -104,16 +104,16 @@ blabel_print(netdissect_options *ndo,
        /* print the bit string as a hex string */
        ND_PRINT((ndo, "\\[x"));
        for (bitp = cp + 1, b = bitlen; bitp < lim && b > 7; b -= 8, bitp++) {
-               ND_TCHECK(*bitp);
+               ND_TCHECK_1(bitp);
                ND_PRINT((ndo, "%02x", EXTRACT_U_1(bitp)));
        }
        if (b > 4) {
-               ND_TCHECK(*bitp);
+               ND_TCHECK_1(bitp);
                tc = EXTRACT_U_1(bitp);
                bitp++;
                ND_PRINT((ndo, "%02x", tc & (0xff << (8 - b))));
        } else if (b > 0) {
-               ND_TCHECK(*bitp);
+               ND_TCHECK_1(bitp);
                tc = EXTRACT_U_1(bitp);
                bitp++;
                ND_PRINT((ndo, "%1x", ((tc >> 4) & 0x0f) & (0x0f << (4 - b))));
@@ -129,7 +129,7 @@ static int
 labellen(netdissect_options *ndo,
          const u_char *cp)
 {
-       register u_int i;
+       u_int i;
 
        if (!ND_TTEST_1(cp))
                return(-1);
@@ -151,11 +151,11 @@ labellen(netdissect_options *ndo,
 
 const u_char *
 ns_nprint(netdissect_options *ndo,
-          register const u_char *cp, register const u_char *bp)
+          const u_char *cp, const u_char *bp)
 {
-       register u_int i, l;
-       register const u_char *rp = NULL;
-       register int compress = 0;
+       u_int i, l;
+       const u_char *rp = NULL;
+       int compress = 0;
        int elt;
        u_int offset, max_offset;
 
@@ -241,9 +241,9 @@ ns_nprint(netdissect_options *ndo,
 /* print a <character-string> */
 static const u_char *
 ns_cprint(netdissect_options *ndo,
-          register const u_char *cp)
+          const u_char *cp)
 {
-       register u_int i;
+       u_int i;
 
        if (!ND_TTEST_1(cp))
                return (NULL);
@@ -254,6 +254,8 @@ ns_cprint(netdissect_options *ndo,
        return (cp + i);
 }
 
+extern const struct tok ns_type2str[];
+
 /* http://www.iana.org/assignments/dns-parameters */
 const struct tok ns_type2str[] = {
        { T_A,          "A" },                  /* RFC 1035 */
@@ -320,6 +322,8 @@ const struct tok ns_type2str[] = {
        { 0,            NULL }
 };
 
+extern const struct tok ns_class2str[];
+
 const struct tok ns_class2str[] = {
        { C_IN,         "IN" },         /* Not used */
        { C_CHAOS,      "CHAOS" },
@@ -331,10 +335,10 @@ const struct tok ns_class2str[] = {
 /* print a query */
 static const u_char *
 ns_qprint(netdissect_options *ndo,
-          register const u_char *cp, register const u_char *bp, int is_mdns)
+          const u_char *cp, const u_char *bp, int is_mdns)
 {
-       register const u_char *np = cp;
-       register u_int i, class;
+       const u_char *np = cp;
+       u_int i, class;
 
        cp = ns_nskip(ndo, cp);
 
@@ -366,11 +370,11 @@ ns_qprint(netdissect_options *ndo,
 /* print a reply */
 static const u_char *
 ns_rprint(netdissect_options *ndo,
-          register const u_char *cp, register const u_char *bp, int is_mdns)
+          const u_char *cp, const u_char *bp, int is_mdns)
 {
-       register u_int i, class, opt_flags = 0;
-       register u_short typ, len;
-       register const u_char *rp;
+       u_int i, class, opt_flags = 0;
+       u_short typ, len;
+       const u_char *rp;
 
        if (ndo->ndo_vflag) {
                ND_PRINT((ndo, " "));
@@ -379,7 +383,7 @@ ns_rprint(netdissect_options *ndo,
        } else
                cp = ns_nskip(ndo, cp);
 
-       if (cp == NULL || !ND_TTEST2(*cp, 10))
+       if (cp == NULL || !ND_TTEST_LEN(cp, 10))
                return (ndo->ndo_snapend);
 
        /* print the type/qtype */
@@ -427,7 +431,7 @@ ns_rprint(netdissect_options *ndo,
 
        switch (typ) {
        case T_A:
-               if (!ND_TTEST2(*cp, sizeof(struct in_addr)))
+               if (!ND_TTEST_LEN(cp, sizeof(struct in_addr)))
                        return(NULL);
                ND_PRINT((ndo, " %s", intoa(htonl(EXTRACT_BE_U_4(cp)))));
                break;
@@ -452,7 +456,7 @@ ns_rprint(netdissect_options *ndo,
                ND_PRINT((ndo, " "));
                if ((cp = ns_nprint(ndo, cp, bp)) == NULL)
                        return(NULL);
-               if (!ND_TTEST2(*cp, 5 * 4))
+               if (!ND_TTEST_LEN(cp, 5 * 4))
                        return(NULL);
                ND_PRINT((ndo, " %u", EXTRACT_BE_U_4(cp)));
                cp += 4;
@@ -498,7 +502,7 @@ ns_rprint(netdissect_options *ndo,
            {
                char ntop_buf[INET6_ADDRSTRLEN];
 
-               if (!ND_TTEST2(*cp, sizeof(struct in6_addr)))
+               if (!ND_TTEST_LEN(cp, sizeof(struct in6_addr)))
                        return(NULL);
                ND_PRINT((ndo, " %s",
                    addrtostr6(cp, ntop_buf, sizeof(ntop_buf))));
@@ -520,7 +524,7 @@ ns_rprint(netdissect_options *ndo,
                        ND_PRINT((ndo, " %u(bad plen)", pbit));
                        break;
                } else if (pbit < 128) {
-                       if (!ND_TTEST2(*(cp + 1), sizeof(a) - pbyte))
+                       if (!ND_TTEST_LEN(cp + 1, sizeof(a) - pbyte))
                                return(NULL);
                        memset(&a, 0, sizeof(a));
                        memcpy(&a.s6_addr[pbyte], cp + 1, sizeof(a) - pbyte);
@@ -542,7 +546,7 @@ ns_rprint(netdissect_options *ndo,
                break;
 
        case T_UNSPECA:         /* One long string */
-               if (!ND_TTEST2(*cp, len))
+               if (!ND_TTEST_LEN(cp, len))
                        return(NULL);
                if (fn_printn(ndo, cp, len, ndo->ndo_snapend))
                        return(NULL);
@@ -585,38 +589,41 @@ ns_rprint(netdissect_options *ndo,
 
 void
 domain_print(netdissect_options *ndo,
-         register const u_char *bp, u_int length, int is_mdns)
+         const u_char *bp, u_int length, int is_mdns)
 {
-       register const HEADER *np;
-       register int qdcount, ancount, nscount, arcount;
-       register const u_char *cp;
+       const HEADER *np;
+       uint16_t flags;
+       u_int qdcount, ancount, nscount, arcount;
+       u_int i;
+       const u_char *cp;
        uint16_t b2;
 
        np = (const HEADER *)bp;
        ND_TCHECK(*np);
+       flags = EXTRACT_BE_U_2(np->flags);
        /* get the byte-order right */
-       qdcount = EXTRACT_BE_U_2(&np->qdcount);
-       ancount = EXTRACT_BE_U_2(&np->ancount);
-       nscount = EXTRACT_BE_U_2(&np->nscount);
-       arcount = EXTRACT_BE_U_2(&np->arcount);
+       qdcount = EXTRACT_BE_U_2(np->qdcount);
+       ancount = EXTRACT_BE_U_2(np->ancount);
+       nscount = EXTRACT_BE_U_2(np->nscount);
+       arcount = EXTRACT_BE_U_2(np->arcount);
 
-       if (DNS_QR(np)) {
+       if (DNS_QR(flags)) {
                /* this is a response */
                ND_PRINT((ndo, "%d%s%s%s%s%s%s",
-                       EXTRACT_BE_U_2(&np->id),
-                       ns_ops[DNS_OPCODE(np)],
-                       ns_resp[DNS_RCODE(np)],
-                       DNS_AA(np)? "*" : "",
-                       DNS_RA(np)? "" : "-",
-                       DNS_TC(np)? "|" : "",
-                       DNS_AD(np)? "$" : ""));
+                       EXTRACT_BE_U_2(np->id),
+                       ns_ops[DNS_OPCODE(flags)],
+                       ns_resp[DNS_RCODE(flags)],
+                       DNS_AA(flags)? "*" : "",
+                       DNS_RA(flags)? "" : "-",
+                       DNS_TC(flags)? "|" : "",
+                       DNS_AD(flags)? "$" : ""));
 
                if (qdcount != 1)
-                       ND_PRINT((ndo, " [%dq]", qdcount));
+                       ND_PRINT((ndo, " [%uq]", qdcount));
                /* Print QUESTION section on -vv */
                cp = (const u_char *)(np + 1);
-               while (qdcount--) {
-                       if (qdcount < EXTRACT_BE_U_2(&np->qdcount) - 1)
+               for (i = 0; i < qdcount; i++) {
+                       if (i != 0)
                                ND_PRINT((ndo, ","));
                        if (ndo->ndo_vflag > 1) {
                                ND_PRINT((ndo, " q:"));
@@ -628,126 +635,140 @@ domain_print(netdissect_options *ndo,
                                cp += 4;        /* skip QTYPE and QCLASS */
                        }
                }
-               ND_PRINT((ndo, " %d/%d/%d", ancount, nscount, arcount));
-               if (ancount--) {
+               ND_PRINT((ndo, " %u/%u/%u", ancount, nscount, arcount));
+               if (ancount) {
                        if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                goto trunc;
-                       while (cp < ndo->ndo_snapend && ancount--) {
+                       ancount--;
+                       while (cp < ndo->ndo_snapend && ancount) {
                                ND_PRINT((ndo, ","));
                                if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                        goto trunc;
+                               ancount--;
                        }
                }
-               if (ancount > 0)
+               if (ancount)
                        goto trunc;
                /* Print NS and AR sections on -vv */
                if (ndo->ndo_vflag > 1) {
-                       if (cp < ndo->ndo_snapend && nscount--) {
+                       if (cp < ndo->ndo_snapend && nscount) {
                                ND_PRINT((ndo, " ns:"));
                                if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                        goto trunc;
-                               while (cp < ndo->ndo_snapend && nscount--) {
+                               nscount--;
+                               while (cp < ndo->ndo_snapend && nscount) {
                                        ND_PRINT((ndo, ","));
                                        if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                                goto trunc;
+                                       nscount--;
                                }
                        }
-                       if (nscount > 0)
+                       if (nscount)
                                goto trunc;
-                       if (cp < ndo->ndo_snapend && arcount--) {
+                       if (cp < ndo->ndo_snapend && arcount) {
                                ND_PRINT((ndo, " ar:"));
                                if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                        goto trunc;
-                               while (cp < ndo->ndo_snapend && arcount--) {
+                               arcount--;
+                               while (cp < ndo->ndo_snapend && arcount) {
                                        ND_PRINT((ndo, ","));
                                        if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                                goto trunc;
+                                       arcount--;
                                }
                        }
-                       if (arcount > 0)
+                       if (arcount)
                                goto trunc;
                }
        }
        else {
                /* this is a request */
-               ND_PRINT((ndo, "%d%s%s%s", EXTRACT_BE_U_2(&np->id), ns_ops[DNS_OPCODE(np)],
-                         DNS_RD(np) ? "+" : "",
-                         DNS_CD(np) ? "%" : ""));
+               ND_PRINT((ndo, "%d%s%s%s", EXTRACT_BE_U_2(np->id), ns_ops[DNS_OPCODE(flags)],
+                         DNS_RD(flags) ? "+" : "",
+                         DNS_CD(flags) ? "%" : ""));
 
                /* any weirdness? */
                b2 = EXTRACT_BE_U_2(((const u_short *)np) + 1);
                if (b2 & 0x6cf)
                        ND_PRINT((ndo, " [b2&3=0x%x]", b2));
 
-               if (DNS_OPCODE(np) == IQUERY) {
+               if (DNS_OPCODE(flags) == IQUERY) {
                        if (qdcount)
-                               ND_PRINT((ndo, " [%dq]", qdcount));
+                               ND_PRINT((ndo, " [%uq]", qdcount));
                        if (ancount != 1)
-                               ND_PRINT((ndo, " [%da]", ancount));
+                               ND_PRINT((ndo, " [%ua]", ancount));
                }
                else {
                        if (ancount)
-                               ND_PRINT((ndo, " [%da]", ancount));
+                               ND_PRINT((ndo, " [%ua]", ancount));
                        if (qdcount != 1)
-                               ND_PRINT((ndo, " [%dq]", qdcount));
+                               ND_PRINT((ndo, " [%uq]", qdcount));
                }
                if (nscount)
-                       ND_PRINT((ndo, " [%dn]", nscount));
+                       ND_PRINT((ndo, " [%un]", nscount));
                if (arcount)
-                       ND_PRINT((ndo, " [%dau]", arcount));
+                       ND_PRINT((ndo, " [%uau]", arcount));
 
                cp = (const u_char *)(np + 1);
-               if (qdcount--) {
+               if (qdcount) {
                        cp = ns_qprint(ndo, cp, (const u_char *)np, is_mdns);
                        if (!cp)
                                goto trunc;
-                       while (cp < ndo->ndo_snapend && qdcount--) {
+                       qdcount--;
+                       while (cp < ndo->ndo_snapend && qdcount) {
                                cp = ns_qprint(ndo, (const u_char *)cp,
                                               (const u_char *)np,
                                               is_mdns);
                                if (!cp)
                                        goto trunc;
+                               qdcount--;
                        }
                }
-               if (qdcount > 0)
+               if (qdcount)
                        goto trunc;
 
                /* Print remaining sections on -vv */
                if (ndo->ndo_vflag > 1) {
-                       if (ancount--) {
+                       if (ancount) {
                                if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                        goto trunc;
-                               while (cp < ndo->ndo_snapend && ancount--) {
+                               ancount--;
+                               while (cp < ndo->ndo_snapend && ancount) {
                                        ND_PRINT((ndo, ","));
                                        if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                                goto trunc;
+                                       ancount--;
                                }
                        }
-                       if (ancount > 0)
+                       if (ancount)
                                goto trunc;
-                       if (cp < ndo->ndo_snapend && nscount--) {
+                       if (cp < ndo->ndo_snapend && nscount) {
                                ND_PRINT((ndo, " ns:"));
                                if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                        goto trunc;
-                               while (nscount-- && cp < ndo->ndo_snapend) {
+                               nscount--;
+                               while (cp < ndo->ndo_snapend && nscount) {
                                        ND_PRINT((ndo, ","));
                                        if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                                goto trunc;
+                                       nscount--;
                                }
                        }
                        if (nscount > 0)
                                goto trunc;
-                       if (cp < ndo->ndo_snapend && arcount--) {
+                       if (cp < ndo->ndo_snapend && arcount) {
                                ND_PRINT((ndo, " ar:"));
                                if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                        goto trunc;
-                               while (cp < ndo->ndo_snapend && arcount--) {
+                               arcount--;
+                               while (cp < ndo->ndo_snapend && arcount) {
                                        ND_PRINT((ndo, ","));
                                        if ((cp = ns_rprint(ndo, cp, bp, is_mdns)) == NULL)
                                                goto trunc;
+                                       arcount--;
                                }
                        }
-                       if (arcount > 0)
+                       if (arcount)
                                goto trunc;
                }
        }