src/utils_dns.c: Import "rfc1035NameUnpack" from dnstop-20080321.
[collectd.git] / src / utils_dns.c
1 /*
2  * collectd - src/utils_dns.c
3  * Modifications Copyright (C) 2006  Florian octo Forster
4  * Copyright (C) 2002  The Measurement Factory, Inc.
5  * All rights reserved.
6  * 
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions are met:
9  * 
10  * 1. Redistributions of source code must retain the above copyright notice,
11  *    this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright notice,
13  *    this list of conditions and the following disclaimer in the documentation
14  *    and/or other materials provided with the distribution.
15  * 3. Neither the name of The Measurement Factory nor the names of its
16  *    contributors may be used to endorse or promote products derived from this
17  *    software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
20  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
23  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  *
31  * Authors:
32  *   The Measurement Factory, Inc. <http://www.measurement-factory.com/>
33  *   Florian octo Forster <octo at verplant.org>
34  */
35
36 #define _BSD_SOURCE
37
38 #include "collectd.h"
39 #include "plugin.h"
40 #include "common.h"
41
42 #if HAVE_NETINET_IN_SYSTM_H
43 # include <netinet/in_systm.h>
44 #endif
45 #if HAVE_NETINET_IN_H
46 # include <netinet/in.h>
47 #endif
48 #if HAVE_ARPA_INET_H
49 # include <arpa/inet.h>
50 #endif
51 #if HAVE_SYS_SOCKET_H
52 # include <sys/socket.h>
53 #endif
54
55 #if HAVE_ARPA_NAMESER_H
56 # include <arpa/nameser.h>
57 #endif
58 #if HAVE_ARPA_NAMESER_COMPAT_H
59 # include <arpa/nameser_compat.h>
60 #endif
61
62 #if HAVE_NET_IF_ARP_H
63 # include <net/if_arp.h>
64 #endif
65 #if HAVE_NET_IF_H
66 # include <net/if.h>
67 #endif
68 #if HAVE_NETINET_IF_ETHER_H
69 # include <netinet/if_ether.h>
70 #endif
71 #if HAVE_NET_PPP_DEFS_H
72 # include <net/ppp_defs.h>
73 #endif
74 #if HAVE_NET_IF_PPP_H
75 # include <net/if_ppp.h>
76 #endif
77
78 #if HAVE_NETDB_H
79 # include <netdb.h>
80 #endif
81
82 #if HAVE_NETINET_IP_H
83 # include <netinet/ip.h>
84 #endif
85 #ifdef HAVE_NETINET_IP_VAR_H
86 # include <netinet/ip_var.h>
87 #endif
88 #if HAVE_NETINET_IP6_H
89 # include <netinet/ip6.h>
90 #endif
91 #if HAVE_NETINET_UDP_H
92 # include <netinet/udp.h>
93 #endif
94
95 #if HAVE_PCAP_H
96 # include <pcap.h>
97 #endif
98
99 #define PCAP_SNAPLEN 1460
100 #ifndef ETHER_HDR_LEN
101 #define ETHER_ADDR_LEN 6
102 #define ETHER_TYPE_LEN 2
103 #define ETHER_HDR_LEN (ETHER_ADDR_LEN * 2 + ETHER_TYPE_LEN)
104 #endif
105 #ifndef ETHERTYPE_8021Q
106 # define ETHERTYPE_8021Q 0x8100
107 #endif
108 #ifndef ETHERTYPE_IPV6
109 # define ETHERTYPE_IPV6 0x86DD
110 #endif
111
112 #ifndef PPP_ADDRESS_VAL
113 # define PPP_ADDRESS_VAL 0xff   /* The address byte value */
114 #endif
115 #ifndef PPP_CONTROL_VAL
116 # define PPP_CONTROL_VAL 0x03   /* The control byte value */
117 #endif
118
119 #if HAVE_STRUCT_UDPHDR_UH_DPORT && HAVE_STRUCT_UDPHDR_UH_SPORT
120 # define UDP_DEST uh_dport
121 # define UDP_SRC  uh_dport
122 #elif HAVE_STRUCT_UDPHDR_DEST && HAVE_STRUCT_UDPHDR_SOURCE
123 # define UDP_DEST dest
124 # define UDP_SRC  source
125 #else
126 # error "`struct udphdr' is unusable."
127 #endif
128
129 #include "utils_dns.h"
130
131 /*
132  * Type definitions
133  */
134 struct ip_list_s
135 {
136     struct in6_addr addr;
137     void *data;
138     struct ip_list_s *next;
139 };
140 typedef struct ip_list_s ip_list_t;
141
142 typedef int (printer)(const char *, ...);
143
144 /*
145  * flags/features for non-interactive mode
146  */
147
148 #ifndef T_A6
149 #define T_A6 38
150 #endif
151 #ifndef T_SRV
152 #define T_SRV 33
153 #endif
154
155 /*
156  * Global variables
157  */
158 int qtype_counts[T_MAX];
159 int opcode_counts[OP_MAX];
160 int qclass_counts[C_MAX];
161
162 #if HAVE_PCAP_H
163 static pcap_t *pcap_obj = NULL;
164 #endif
165
166 static ip_list_t *IgnoreList = NULL;
167
168 #if HAVE_PCAP_H
169 static void (*Callback) (const rfc1035_header_t *) = NULL;
170
171 static int query_count_intvl = 0;
172 static int query_count_total = 0;
173 # ifdef __OpenBSD__
174 static struct bpf_timeval last_ts;
175 # else
176 static struct timeval last_ts;
177 # endif /* __OpenBSD__ */
178 #endif /* HAVE_PCAP_H */
179
180 static int cmp_in6_addr (const struct in6_addr *a,
181         const struct in6_addr *b)
182 {
183     int i;
184
185     assert (sizeof (struct in6_addr) == 16);
186
187     for (i = 0; i < 16; i++)
188         if (a->s6_addr[i] != b->s6_addr[i])
189             break;
190
191     if (i >= 16)
192         return (0);
193
194     return (a->s6_addr[i] > b->s6_addr[i] ? 1 : -1);
195 } /* int cmp_addrinfo */
196
197 static inline int ignore_list_match (const struct in6_addr *addr)
198 {
199     ip_list_t *ptr;
200
201     for (ptr = IgnoreList; ptr != NULL; ptr = ptr->next)
202         if (cmp_in6_addr (addr, &ptr->addr) == 0)
203             return (1);
204     return (0);
205 } /* int ignore_list_match */
206
207 static void ignore_list_add (const struct in6_addr *addr)
208 {
209     ip_list_t *new;
210
211     if (ignore_list_match (addr) != 0)
212         return;
213
214     new = malloc (sizeof (ip_list_t));
215     if (new == NULL)
216     {
217         perror ("malloc");
218         return;
219     }
220
221     memcpy (&new->addr, addr, sizeof (struct in6_addr));
222     new->next = IgnoreList;
223
224     IgnoreList = new;
225 } /* void ignore_list_add */
226
227 void ignore_list_add_name (const char *name)
228 {
229     struct addrinfo *ai_list;
230     struct addrinfo *ai_ptr;
231     struct in6_addr  addr;
232     int status;
233
234     status = getaddrinfo (name, NULL, NULL, &ai_list);
235     if (status != 0)
236         return;
237
238     for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
239     {
240         if (ai_ptr->ai_family == AF_INET)
241         {
242             memset (&addr, '\0', sizeof (addr));
243             addr.s6_addr[10] = 0xFF;
244             addr.s6_addr[11] = 0xFF;
245             memcpy (addr.s6_addr + 12, &((struct sockaddr_in *) ai_ptr->ai_addr)->sin_addr, 4);
246
247             ignore_list_add (&addr);
248         }
249         else
250         {
251             ignore_list_add (&((struct sockaddr_in6 *) ai_ptr->ai_addr)->sin6_addr);
252         }
253     } /* for */
254
255     freeaddrinfo (ai_list);
256 }
257
258 #if HAVE_PCAP_H
259 static void in6_addr_from_buffer (struct in6_addr *ia,
260         const void *buf, size_t buf_len,
261         int family)
262 {
263     memset (ia, 0, sizeof (struct in6_addr));
264     if ((AF_INET == family) && (sizeof (uint32_t) == buf_len))
265     {
266         ia->s6_addr[10] = 0xFF;
267         ia->s6_addr[11] = 0xFF;
268         memcpy (ia->s6_addr + 12, buf, buf_len);
269     }
270     else if ((AF_INET6 == family) && (sizeof (struct in6_addr) == buf_len))
271     {
272         memcpy (ia, buf, buf_len);
273     }
274 } /* void in6_addr_from_buffer */
275
276 void dnstop_set_pcap_obj (pcap_t *po)
277 {
278         pcap_obj = po;
279 }
280
281 void dnstop_set_callback (void (*cb) (const rfc1035_header_t *))
282 {
283         Callback = cb;
284 }
285
286 #define RFC1035_MAXLABELSZ 63
287 static int
288 rfc1035NameUnpack(const char *buf, size_t sz, off_t * off, char *name, size_t ns
289 )
290 {
291     off_t no = 0;
292     unsigned char c;
293     size_t len;
294     static int loop_detect = 0;
295     if (loop_detect > 2)
296         return 4;               /* compression loop */
297     if (ns <= 0)
298         return 4;               /* probably compression loop */
299     do {
300         if ((*off) >= sz)
301             break;
302         c = *(buf + (*off));
303         if (c > 191) {
304             /* blasted compression */
305             int rc;
306             unsigned short s;
307             off_t ptr;
308             memcpy(&s, buf + (*off), sizeof(s));
309             s = ntohs(s);
310             (*off) += sizeof(s);
311             /* Sanity check */
312             if ((*off) >= sz)
313                 return 1;       /* message too short */
314             ptr = s & 0x3FFF;
315             /* Make sure the pointer is inside this message */
316             if (ptr >= sz)
317                 return 2;       /* bad compression ptr */
318             if (ptr < DNS_MSG_HDR_SZ)
319                 return 2;       /* bad compression ptr */
320             loop_detect++;
321             rc = rfc1035NameUnpack(buf, sz, &ptr, name + no, ns - no);
322             loop_detect--;
323             return rc;
324         } else if (c > RFC1035_MAXLABELSZ) {
325             /*
326              * "(The 10 and 01 combinations are reserved for future use.)"
327              */
328             return 3;           /* reserved label/compression flags */
329             break;
330         } else {
331             (*off)++;
332             len = (size_t) c;
333             if (len == 0)
334                 break;
335             if (len > (ns - 1))
336                 len = ns - 1;
337             if ((*off) + len > sz)
338                 return 4;       /* message is too short */
339             if (no + len + 1 > ns)
340                 return 5;       /* qname would overflow name buffer */
341             memcpy(name + no, buf + (*off), len);
342             (*off) += len;
343             no += len;
344             *(name + (no++)) = '.';
345         }
346     } while (c > 0);
347     if (no > 0)
348         *(name + no - 1) = '\0';
349     /* make sure we didn't allow someone to overflow the name buffer */
350     assert(no <= ns);
351     return 0;
352 }
353
354 static int
355 handle_dns(const char *buf, int len)
356 {
357     rfc1035_header_t qh;
358     uint16_t us;
359     off_t offset;
360     char *t;
361     int x;
362
363     /* The DNS header is 12 bytes long */
364     if (len < 12)
365         return 0;
366
367     memcpy(&us, buf + 0, 2);
368     qh.id = ntohs(us);
369
370     memcpy(&us, buf + 2, 2);
371     us = ntohs(us);
372     qh.qr = (us >> 15) & 0x01;
373     qh.opcode = (us >> 11) & 0x0F;
374     qh.aa = (us >> 10) & 0x01;
375     qh.tc = (us >> 9) & 0x01;
376     qh.rd = (us >> 8) & 0x01;
377     qh.ra = (us >> 7) & 0x01;
378     qh.z  = (us >> 6) & 0x01;
379     qh.ad = (us >> 5) & 0x01;
380     qh.cd = (us >> 4) & 0x01;
381     qh.rcode = us & 0x0F;
382
383     memcpy(&us, buf + 4, 2);
384     qh.qdcount = ntohs(us);
385
386     memcpy(&us, buf + 6, 2);
387     qh.ancount = ntohs(us);
388
389     memcpy(&us, buf + 8, 2);
390     qh.nscount = ntohs(us);
391
392     memcpy(&us, buf + 10, 2);
393     qh.arcount = ntohs(us);
394
395     offset = 12;
396     memset(qh.qname, '\0', MAX_QNAME_SZ);
397     x = rfc1035NameUnpack(buf, len, &offset, qh.qname, MAX_QNAME_SZ);
398     if (0 != x)
399         return 0;
400     if ('\0' == qh.qname[0])
401         sstrncpy (qh.qname, ".", sizeof (qh.qname));
402     while ((t = strchr(qh.qname, '\n')))
403         *t = ' ';
404     while ((t = strchr(qh.qname, '\r')))
405         *t = ' ';
406     for (t = qh.qname; *t; t++)
407         *t = tolower((int) *t);
408
409     memcpy(&us, buf + offset, 2);
410     qh.qtype = ntohs(us);
411     memcpy(&us, buf + offset + 2, 2);
412     qh.qclass = ntohs(us);
413
414     qh.length = (uint16_t) len;
415
416     /* gather stats */
417     qtype_counts[qh.qtype]++;
418     qclass_counts[qh.qclass]++;
419     opcode_counts[qh.opcode]++;
420
421     if (Callback != NULL)
422             Callback (&qh);
423
424     return 1;
425 }
426
427 static int
428 handle_udp(const struct udphdr *udp, int len)
429 {
430     char buf[PCAP_SNAPLEN];
431     if ((ntohs (udp->UDP_DEST) != 53)
432                     && (ntohs (udp->UDP_SRC) != 53))
433         return 0;
434     memcpy(buf, udp + 1, len - sizeof(*udp));
435     if (0 == handle_dns(buf, len - sizeof(*udp)))
436         return 0;
437     return 1;
438 }
439
440 static int
441 handle_ipv6 (struct ip6_hdr *ipv6, int len)
442 {
443     char buf[PCAP_SNAPLEN];
444     unsigned int offset;
445     int nexthdr;
446
447     struct in6_addr s_addr;
448     struct in6_addr d_addr;
449     uint16_t payload_len;
450
451     if (0 > len)
452         return (0);
453
454     offset = sizeof (struct ip6_hdr);
455     nexthdr = ipv6->ip6_nxt;
456     s_addr = ipv6->ip6_src;
457     d_addr = ipv6->ip6_dst;
458     payload_len = ntohs (ipv6->ip6_plen);
459
460     if (ignore_list_match (&s_addr))
461             return (0);
462
463     /* Parse extension headers. This only handles the standard headers, as
464      * defined in RFC 2460, correctly. Fragments are discarded. */
465     while ((IPPROTO_ROUTING == nexthdr) /* routing header */
466             || (IPPROTO_HOPOPTS == nexthdr) /* Hop-by-Hop options. */
467             || (IPPROTO_FRAGMENT == nexthdr) /* fragmentation header. */
468             || (IPPROTO_DSTOPTS == nexthdr) /* destination options. */
469             || (IPPROTO_DSTOPTS == nexthdr) /* destination options. */
470             || (IPPROTO_AH == nexthdr) /* destination options. */
471             || (IPPROTO_ESP == nexthdr)) /* encapsulating security payload. */
472     {
473         struct ip6_ext ext_hdr;
474         uint16_t ext_hdr_len;
475
476         /* Catch broken packets */
477         if ((offset + sizeof (struct ip6_ext)) > (unsigned int)len)
478             return (0);
479
480         /* Cannot handle fragments. */
481         if (IPPROTO_FRAGMENT == nexthdr)
482             return (0);
483
484         memcpy (&ext_hdr, (char *) ipv6 + offset, sizeof (struct ip6_ext));
485         nexthdr = ext_hdr.ip6e_nxt;
486         ext_hdr_len = (8 * (ntohs (ext_hdr.ip6e_len) + 1));
487
488         /* This header is longer than the packets payload.. WTF? */
489         if (ext_hdr_len > payload_len)
490             return (0);
491
492         offset += ext_hdr_len;
493         payload_len -= ext_hdr_len;
494     } /* while */
495
496     /* Catch broken and empty packets */
497     if (((offset + payload_len) > (unsigned int)len)
498             || (payload_len == 0)
499             || (payload_len > PCAP_SNAPLEN))
500         return (0);
501
502     if (IPPROTO_UDP != nexthdr)
503         return (0);
504
505     memcpy (buf, (char *) ipv6 + offset, payload_len);
506     if (handle_udp ((struct udphdr *) buf, payload_len) == 0)
507         return (0);
508
509     return (1); /* Success */
510 } /* int handle_ipv6 */
511
512 static int
513 handle_ip(const struct ip *ip, int len)
514 {
515     char buf[PCAP_SNAPLEN];
516     int offset = ip->ip_hl << 2;
517     struct in6_addr s_addr;
518     struct in6_addr d_addr;
519
520     if (ip->ip_v == 6)
521         return (handle_ipv6 ((struct ip6_hdr *) ip, len));
522
523     in6_addr_from_buffer (&s_addr, &ip->ip_src.s_addr, sizeof (ip->ip_src.s_addr), AF_INET);
524     in6_addr_from_buffer (&d_addr, &ip->ip_dst.s_addr, sizeof (ip->ip_dst.s_addr), AF_INET);
525     if (ignore_list_match (&s_addr))
526             return (0);
527     if (IPPROTO_UDP != ip->ip_p)
528         return 0;
529     memcpy(buf, (void *) ip + offset, len - offset);
530     if (0 == handle_udp((struct udphdr *) buf, len - offset))
531         return 0;
532     return 1;
533 }
534
535 #if HAVE_NET_IF_PPP_H
536 static int
537 handle_ppp(const u_char * pkt, int len)
538 {
539     char buf[PCAP_SNAPLEN];
540     unsigned short us;
541     unsigned short proto;
542     if (len < 2)
543         return 0;
544     if (*pkt == PPP_ADDRESS_VAL && *(pkt + 1) == PPP_CONTROL_VAL) {
545         pkt += 2;               /* ACFC not used */
546         len -= 2;
547     }
548     if (len < 2)
549         return 0;
550     if (*pkt % 2) {
551         proto = *pkt;           /* PFC is used */
552         pkt++;
553         len--;
554     } else {
555         memcpy(&us, pkt, sizeof(us));
556         proto = ntohs(us);
557         pkt += 2;
558         len -= 2;
559     }
560     if (ETHERTYPE_IP != proto && PPP_IP != proto)
561         return 0;
562     memcpy(buf, pkt, len);
563     return handle_ip((struct ip *) buf, len);
564 }
565 #endif /* HAVE_NET_IF_PPP_H */
566
567 static int
568 handle_null(const u_char * pkt, int len)
569 {
570     unsigned int family;
571     memcpy(&family, pkt, sizeof(family));
572     if (AF_INET != family)
573         return 0;
574     return handle_ip((struct ip *) (pkt + 4), len - 4);
575 }
576
577 #ifdef DLT_LOOP
578 static int
579 handle_loop(const u_char * pkt, int len)
580 {
581     unsigned int family;
582     memcpy(&family, pkt, sizeof(family));
583     if (AF_INET != ntohl(family))
584         return 0;
585     return handle_ip((struct ip *) (pkt + 4), len - 4);
586 }
587
588 #endif
589
590 #ifdef DLT_RAW
591 static int
592 handle_raw(const u_char * pkt, int len)
593 {
594     return handle_ip((struct ip *) pkt, len);
595 }
596
597 #endif
598
599 static int
600 handle_ether(const u_char * pkt, int len)
601 {
602     char buf[PCAP_SNAPLEN];
603     struct ether_header *e = (void *) pkt;
604     unsigned short etype = ntohs(e->ether_type);
605     if (len < ETHER_HDR_LEN)
606         return 0;
607     pkt += ETHER_HDR_LEN;
608     len -= ETHER_HDR_LEN;
609     if (ETHERTYPE_8021Q == etype) {
610         etype = ntohs(*(unsigned short *) (pkt + 2));
611         pkt += 4;
612         len -= 4;
613     }
614     if ((ETHERTYPE_IP != etype)
615             && (ETHERTYPE_IPV6 != etype))
616         return 0;
617     memcpy(buf, pkt, len);
618     if (ETHERTYPE_IPV6 == etype)
619         return (handle_ipv6 ((struct ip6_hdr *) buf, len));
620     else
621         return handle_ip((struct ip *) buf, len);
622 }
623
624 #ifdef DLT_LINUX_SLL
625 static int
626 handle_linux_sll (const u_char *pkt, int len)
627 {
628     struct sll_header
629     {
630         uint16_t pkt_type;
631         uint16_t dev_type;
632         uint16_t addr_len;
633         uint8_t  addr[8];
634         uint16_t proto_type;
635     } *hdr;
636     uint16_t etype;
637
638     if ((0 > len) || ((unsigned int)len < sizeof (struct sll_header)))
639         return (0);
640
641     hdr  = (struct sll_header *) pkt;
642     pkt  = (u_char *) (hdr + 1);
643     len -= sizeof (struct sll_header);
644
645     etype = ntohs (hdr->proto_type);
646
647     if ((ETHERTYPE_IP != etype)
648             && (ETHERTYPE_IPV6 != etype))
649         return 0;
650
651     if (ETHERTYPE_IPV6 == etype)
652         return (handle_ipv6 ((struct ip6_hdr *) pkt, len));
653     else
654         return handle_ip((struct ip *) pkt, len);
655 }
656 #endif /* DLT_LINUX_SLL */
657
658 /* public function */
659 void handle_pcap(u_char *udata, const struct pcap_pkthdr *hdr, const u_char *pkt)
660 {
661     int status;
662
663     DEBUG ("handle_pcap (udata = %p, hdr = %p, pkt = %p): hdr->caplen = %i\n",
664                     (void *) udata, (void *) hdr, (void *) pkt,
665                     hdr->caplen);
666
667     if (hdr->caplen < ETHER_HDR_LEN)
668         return;
669
670     switch (pcap_datalink (pcap_obj))
671     {
672         case DLT_EN10MB:
673             status = handle_ether (pkt, hdr->caplen);
674             break;
675 #if HAVE_NET_IF_PPP_H
676         case DLT_PPP:
677             status = handle_ppp (pkt, hdr->caplen);
678             break;
679 #endif
680 #ifdef DLT_LOOP
681         case DLT_LOOP:
682             status = handle_loop (pkt, hdr->caplen);
683             break;
684 #endif
685 #ifdef DLT_RAW
686         case DLT_RAW:
687             status = handle_raw (pkt, hdr->caplen);
688             break;
689 #endif
690 #ifdef DLT_LINUX_SLL
691         case DLT_LINUX_SLL:
692             status = handle_linux_sll (pkt, hdr->caplen);
693             break;
694 #endif
695         case DLT_NULL:
696             status = handle_null (pkt, hdr->caplen);
697             break;
698
699         default:
700             ERROR ("handle_pcap: unsupported data link type %d\n",
701                     pcap_datalink(pcap_obj));
702             status = 0;
703             break;
704     } /* switch (pcap_datalink(pcap_obj)) */
705
706     if (0 == status)
707         return;
708
709     query_count_intvl++;
710     query_count_total++;
711     last_ts = hdr->ts;
712 }
713 #endif /* HAVE_PCAP_H */
714
715 const char *qtype_str(int t)
716 {
717     static char buf[32];
718     switch (t) {
719 #if (defined (__NAMESER)) && (__NAMESER >= 19991001)
720             case ns_t_a:        return ("A");
721             case ns_t_ns:       return ("NS");
722             case ns_t_md:       return ("MD");
723             case ns_t_mf:       return ("MF");
724             case ns_t_cname:    return ("CNAME");
725             case ns_t_soa:      return ("SOA");
726             case ns_t_mb:       return ("MB");
727             case ns_t_mg:       return ("MG");
728             case ns_t_mr:       return ("MR");
729             case ns_t_null:     return ("NULL");
730             case ns_t_wks:      return ("WKS");
731             case ns_t_ptr:      return ("PTR");
732             case ns_t_hinfo:    return ("HINFO");
733             case ns_t_minfo:    return ("MINFO");
734             case ns_t_mx:       return ("MX");
735             case ns_t_txt:      return ("TXT");
736             case ns_t_rp:       return ("RP");
737             case ns_t_afsdb:    return ("AFSDB");
738             case ns_t_x25:      return ("X25");
739             case ns_t_isdn:     return ("ISDN");
740             case ns_t_rt:       return ("RT");
741             case ns_t_nsap:     return ("NSAP");
742             case ns_t_nsap_ptr: return ("NSAP-PTR");
743             case ns_t_sig:      return ("SIG");
744             case ns_t_key:      return ("KEY");
745             case ns_t_px:       return ("PX");
746             case ns_t_gpos:     return ("GPOS");
747             case ns_t_aaaa:     return ("AAAA");
748             case ns_t_loc:      return ("LOC");
749             case ns_t_nxt:      return ("NXT");
750             case ns_t_eid:      return ("EID");
751             case ns_t_nimloc:   return ("NIMLOC");
752             case ns_t_srv:      return ("SRV");
753             case ns_t_atma:     return ("ATMA");
754             case ns_t_naptr:    return ("NAPTR");
755             case ns_t_kx:       return ("KX");
756             case ns_t_cert:     return ("CERT");
757             case ns_t_a6:       return ("A6");
758             case ns_t_dname:    return ("DNAME");
759             case ns_t_sink:     return ("SINK");
760             case ns_t_opt:      return ("OPT");
761 # if __NAMESER >= 19991006
762             case ns_t_tsig:     return ("TSIG");
763 # endif
764             case ns_t_ixfr:     return ("IXFR");
765             case ns_t_axfr:     return ("AXFR");
766             case ns_t_mailb:    return ("MAILB");
767             case ns_t_maila:    return ("MAILA");
768             case ns_t_any:      return ("ANY");
769             case ns_t_zxfr:     return ("ZXFR");
770 /* #endif __NAMESER >= 19991006 */
771 #elif (defined (__BIND)) && (__BIND >= 19950621)
772             case T_A:           return ("A"); /* 1 ... */
773             case T_NS:          return ("NS");
774             case T_MD:          return ("MD");
775             case T_MF:          return ("MF");
776             case T_CNAME:       return ("CNAME");
777             case T_SOA:         return ("SOA");
778             case T_MB:          return ("MB");
779             case T_MG:          return ("MG");
780             case T_MR:          return ("MR");
781             case T_NULL:        return ("NULL");
782             case T_WKS:         return ("WKS");
783             case T_PTR:         return ("PTR");
784             case T_HINFO:       return ("HINFO");
785             case T_MINFO:       return ("MINFO");
786             case T_MX:          return ("MX");
787             case T_TXT:         return ("TXT");
788             case T_RP:          return ("RP");
789             case T_AFSDB:       return ("AFSDB");
790             case T_X25:         return ("X25");
791             case T_ISDN:        return ("ISDN");
792             case T_RT:          return ("RT");
793             case T_NSAP:        return ("NSAP");
794             case T_NSAP_PTR:    return ("NSAP_PTR");
795             case T_SIG:         return ("SIG");
796             case T_KEY:         return ("KEY");
797             case T_PX:          return ("PX");
798             case T_GPOS:        return ("GPOS");
799             case T_AAAA:        return ("AAAA");
800             case T_LOC:         return ("LOC");
801             case T_NXT:         return ("NXT");
802             case T_EID:         return ("EID");
803             case T_NIMLOC:      return ("NIMLOC");
804             case T_SRV:         return ("SRV");
805             case T_ATMA:        return ("ATMA");
806             case T_NAPTR:       return ("NAPTR"); /* ... 35 */
807 #if (__BIND >= 19960801)
808             case T_KX:          return ("KX"); /* 36 ... */
809             case T_CERT:        return ("CERT");
810             case T_A6:          return ("A6");
811             case T_DNAME:       return ("DNAME");
812             case T_SINK:        return ("SINK");
813             case T_OPT:         return ("OPT");
814             case T_APL:         return ("APL");
815             case T_DS:          return ("DS");
816             case T_SSHFP:       return ("SSHFP");
817             case T_RRSIG:       return ("RRSIG");
818             case T_NSEC:        return ("NSEC");
819             case T_DNSKEY:      return ("DNSKEY"); /* ... 48 */
820             case T_TKEY:        return ("TKEY"); /* 249 */
821 #endif /* __BIND >= 19960801 */
822             case T_TSIG:        return ("TSIG"); /* 250 ... */
823             case T_IXFR:        return ("IXFR");
824             case T_AXFR:        return ("AXFR");
825             case T_MAILB:       return ("MAILB");
826             case T_MAILA:       return ("MAILA");
827             case T_ANY:         return ("ANY"); /* ... 255 */
828 #endif /* __BIND >= 19950621 */
829             default:
830                     ssnprintf (buf, sizeof (buf), "#%i", t);
831                     return (buf);
832     }; /* switch (t) */
833     /* NOTREACHED */
834     return (NULL);
835 }
836
837 const char *opcode_str (int o)
838 {
839     static char buf[30];
840     switch (o) {
841     case 0:
842         return "Query";
843         break;
844     case 1:
845         return "Iquery";
846         break;
847     case 2:
848         return "Status";
849         break;
850     case 4:
851         return "Notify";
852         break;
853     case 5:
854         return "Update";
855         break;
856     default:
857         ssnprintf(buf, sizeof (buf), "Opcode%d", o);
858         return buf;
859     }
860     /* NOTREACHED */
861 }
862
863 const char *rcode_str (int rcode)
864 {
865         static char buf[32];
866         switch (rcode)
867         {
868 #if (defined (__NAMESER)) && (__NAMESER >= 19991006)
869                 case ns_r_noerror:  return ("NOERROR");
870                 case ns_r_formerr:  return ("FORMERR");
871                 case ns_r_servfail: return ("SERVFAIL");
872                 case ns_r_nxdomain: return ("NXDOMAIN");
873                 case ns_r_notimpl:  return ("NOTIMPL");
874                 case ns_r_refused:  return ("REFUSED");
875                 case ns_r_yxdomain: return ("YXDOMAIN");
876                 case ns_r_yxrrset:  return ("YXRRSET");
877                 case ns_r_nxrrset:  return ("NXRRSET");
878                 case ns_r_notauth:  return ("NOTAUTH");
879                 case ns_r_notzone:  return ("NOTZONE");
880                 case ns_r_max:      return ("MAX");
881                 case ns_r_badsig:   return ("BADSIG");
882                 case ns_r_badkey:   return ("BADKEY");
883                 case ns_r_badtime:  return ("BADTIME");
884 /* #endif __NAMESER >= 19991006 */
885 #elif (defined (__BIND)) && (__BIND >= 19950621)
886                 case NOERROR:       return ("NOERROR");
887                 case FORMERR:       return ("FORMERR");
888                 case SERVFAIL:      return ("SERVFAIL");
889                 case NXDOMAIN:      return ("NXDOMAIN");
890                 case NOTIMP:        return ("NOTIMP");
891                 case REFUSED:       return ("REFUSED");
892 #if defined (YXDOMAIN) && defined (NXRRSET)
893                 case YXDOMAIN:      return ("YXDOMAIN");
894                 case YXRRSET:       return ("YXRRSET");
895                 case NXRRSET:       return ("NXRRSET");
896                 case NOTAUTH:       return ("NOTAUTH");
897                 case NOTZONE:       return ("NOTZONE");
898 #endif  /* RFC2136 rcodes */
899 #endif /* __BIND >= 19950621 */
900                 default:
901                         ssnprintf (buf, sizeof (buf), "RCode%i", rcode);
902                         return (buf);
903         }
904         /* Never reached */
905         return (NULL);
906 } /* const char *rcode_str (int rcode) */
907
908 #if 0
909 static int
910 main(int argc, char *argv[])
911 {
912     char errbuf[PCAP_ERRBUF_SIZE];
913     int x;
914     struct stat sb;
915     int readfile_state = 0;
916     struct bpf_program fp;
917
918     port53 = htons(53);
919     SubReport = Sources_report;
920     ignore_addr.s_addr = 0;
921     progname = strdup(strrchr(argv[0], '/') ? strchr(argv[0], '/') + 1 : argv[0]);
922     srandom(time(NULL));
923     ResetCounters();
924
925     while ((x = getopt(argc, argv, "ab:f:i:pst")) != -1) {
926         switch (x) {
927         case 'a':
928             anon_flag = 1;
929             break;
930         case 's':
931             sld_flag = 1;
932             break;
933         case 't':
934             nld_flag = 1;
935             break;
936         case 'p':
937             promisc_flag = 0;
938             break;
939         case 'b':
940             bpf_program_str = strdup(optarg);
941             break;
942         case 'i':
943             ignore_addr.s_addr = inet_addr(optarg);
944             break;
945         case 'f':
946             set_filter(optarg);
947             break;
948         default:
949             usage();
950             break;
951         }
952     }
953     argc -= optind;
954     argv += optind;
955
956     if (argc < 1)
957         usage();
958     device = strdup(argv[0]);
959
960     if (0 == stat(device, &sb))
961         readfile_state = 1;
962     if (readfile_state) {
963         pcap_obj = pcap_open_offline(device, errbuf);
964     } else {
965         pcap_obj = pcap_open_live(device, PCAP_SNAPLEN, promisc_flag, 1000, errbuf);
966     }
967     if (NULL == pcap_obj) {
968         fprintf(stderr, "pcap_open_*: %s\n", errbuf);
969         exit(1);
970     }
971
972     if (0 == isatty(1)) {
973         if (0 == readfile_state) {
974             fprintf(stderr, "Non-interactive mode requires savefile argument\n");
975             exit(1);
976         }
977         interactive = 0;
978         print_func = printf;
979     }
980
981     memset(&fp, '\0', sizeof(fp));
982     x = pcap_compile(pcap_obj, &fp, bpf_program_str, 1, 0);
983     if (x < 0) {
984         fprintf(stderr, "pcap_compile failed\n");
985         exit(1);
986     }
987     x = pcap_setfilter(pcap_obj, &fp);
988     if (x < 0) {
989         fprintf(stderr, "pcap_setfilter failed\n");
990         exit(1);
991     }
992
993     /*
994      * non-blocking call added for Mac OS X bugfix.  Sent by Max Horn.
995      * ref http://www.tcpdump.org/lists/workers/2002/09/msg00033.html
996      */
997     x = pcap_setnonblock(pcap_obj, 1, errbuf);
998     if (x < 0) {
999         fprintf(stderr, "pcap_setnonblock failed: %s\n", errbuf);
1000         exit(1);
1001     }
1002
1003     switch (pcap_datalink(pcap_obj)) {
1004     case DLT_EN10MB:
1005         handle_datalink = handle_ether;
1006         break;
1007 #if HAVE_NET_IF_PPP_H
1008     case DLT_PPP:
1009         handle_datalink = handle_ppp;
1010         break;
1011 #endif
1012 #ifdef DLT_LOOP
1013     case DLT_LOOP:
1014         handle_datalink = handle_loop;
1015         break;
1016 #endif
1017 #ifdef DLT_RAW
1018     case DLT_RAW:
1019         handle_datalink = handle_raw;
1020         break;
1021 #endif
1022     case DLT_NULL:
1023         handle_datalink = handle_null;
1024         break;
1025     default:
1026         fprintf(stderr, "unsupported data link type %d\n",
1027             pcap_datalink(pcap_obj));
1028         return 1;
1029         break;
1030     }
1031     if (interactive) {
1032         init_curses();
1033         while (0 == Quit) {
1034             if (readfile_state < 2) {
1035                 /*
1036                  * On some OSes select() might return 0 even when
1037                  * there are packets to process.  Thus, we always
1038                  * ignore its return value and just call pcap_dispatch()
1039                  * anyway.
1040                  */
1041                 if (0 == readfile_state)        /* interactive */
1042                     pcap_select(pcap_obj, 1, 0);
1043                 x = pcap_dispatch(pcap_obj, 50, handle_pcap, NULL);
1044             }
1045             if (0 == x && 1 == readfile_state) {
1046                 /* block on keyboard until user quits */
1047                 readfile_state++;
1048                 nodelay(w, 0);
1049             }
1050             keyboard();
1051             cron_pre();
1052             report();
1053             cron_post();
1054         }
1055         endwin();               /* klin, Thu Nov 28 08:56:51 2002 */
1056     } else {
1057         while (pcap_dispatch(pcap_obj, 50, handle_pcap, NULL))
1058                 (void) 0;
1059         cron_pre();
1060         Sources_report(); print_func("\n");
1061         Destinatioreport(); print_func("\n");
1062         Qtypes_report(); print_func("\n");
1063         Opcodes_report(); print_func("\n");
1064         Tld_report(); print_func("\n");
1065         Sld_report(); print_func("\n");
1066         Nld_report(); print_func("\n");
1067         SldBySource_report();
1068     }
1069
1070     pcap_close(pcap_obj);
1071     return 0;
1072 } /* static int main(int argc, char *argv[]) */
1073 #endif
1074 /*
1075  * vim:shiftwidth=4:tabstop=8:softtabstop=4
1076  */