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 */
36 #include "collectd.h"
38 #if HAVE_NETINET_IN_H
39 # include <netinet/in.h>
40 #endif
41 #if HAVE_PCAP_H
42 # include <pcap.h>
43 #endif
44 #if HAVE_ARPA_INET_H
45 # include <arpa/inet.h>
46 #endif
48 #if HAVE_ARPA_NAMESER_H
49 # include <arpa/nameser.h>
50 #elif HAVE_ARPA_NAMESER_COMPAT_H
51 # include <arpa/nameser_compat.h>
52 #endif
54 #if HAVE_NET_IF_ARP_H
55 # include <net/if_arp.h>
56 #endif
57 #if HAVE_NET_IF_H
58 # include <net/if.h>
59 #endif
60 #if HAVE_NETINET_IF_ETHER_H
61 # include <netinet/if_ether.h>
62 #endif
63 #if HAVE_NET_IF_PPP_H
64 # include <net/if_ppp.h>
65 #endif
67 #if HAVE_SYS_SOCKET_H
68 # include <sys/socket.h>
69 #endif
70 #if HAVE_NETDB_H
71 # include <netdb.h>
72 #endif
74 #if HAVE_NETINET_IN_SYSTM_H
75 # include <netinet/in_systm.h>
76 #endif
77 #if HAVE_NETINET_IN_H
78 # include <netinet/in.h>
79 #endif
80 #if HAVE_NETINET_IP_H
81 # include <netinet/ip.h>
82 #endif
83 #ifdef HAVE_NETINET_IP_VAR_H
84 # include <netinet/ip_var.h>
85 #endif
86 #if HAVE_NETINET_IP6_H
87 # include <netinet/ip6.h>
88 #endif
89 #if HAVE_NETINET_UDP_H
90 # include <netinet/udp.h>
91 #endif
93 #define PCAP_SNAPLEN 1460
94 #ifndef ETHER_HDR_LEN
95 #define ETHER_ADDR_LEN 6
96 #define ETHER_TYPE_LEN 2
97 #define ETHER_HDR_LEN (ETHER_ADDR_LEN * 2 + ETHER_TYPE_LEN)
98 #endif
99 #ifndef ETHERTYPE_8021Q
100 # define ETHERTYPE_8021Q 0x8100
101 #endif
102 #ifndef ETHERTYPE_IPV6
103 # define ETHERTYPE_IPV6 0x86DD
104 #endif
106 #ifndef PPP_ADDRESS_VAL
107 # define PPP_ADDRESS_VAL 0xff /* The address byte value */
108 #endif
109 #ifndef PPP_CONTROL_VAL
110 # define PPP_CONTROL_VAL 0x03 /* The control byte value */
111 #endif
113 #ifdef __linux__
114 #define uh_sport source
115 #define uh_dport dest
116 #endif
118 #include "utils_dns.h"
120 /*
121 * Type definitions
122 */
123 struct ip_list_s
124 {
125 struct in6_addr addr;
126 void *data;
127 struct ip_list_s *next;
128 };
129 typedef struct ip_list_s ip_list_t;
131 typedef int (printer)(const char *, ...);
133 /*
134 * flags/features for non-interactive mode
135 */
137 #ifndef T_A6
138 #define T_A6 38
139 #endif
140 #ifndef T_SRV
141 #define T_SRV 33
142 #endif
144 /*
145 * Global variables
146 */
147 int qtype_counts[T_MAX];
148 int opcode_counts[OP_MAX];
149 int qclass_counts[C_MAX];
151 #if HAVE_PCAP_H
152 static pcap_t *pcap_obj = NULL;
153 #endif
155 static ip_list_t *IgnoreList = NULL;
157 #if HAVE_PCAP_H
158 static void (*Callback) (const rfc1035_header_t *) = NULL;
160 static int query_count_intvl = 0;
161 static int query_count_total = 0;
162 # ifdef __OpenBSD__
163 static struct bpf_timeval last_ts;
164 # else
165 static struct timeval last_ts;
166 # endif /* __OpenBSD__ */
167 #endif /* HAVE_PCAP_H */
169 static int cmp_in6_addr (const struct in6_addr *a,
170 const struct in6_addr *b)
171 {
172 int i;
174 assert (sizeof (struct in6_addr) == 16);
176 for (i = 0; i < 16; i++)
177 if (a->s6_addr[i] != b->s6_addr[i])
178 break;
180 if (i >= 16)
181 return (0);
183 return (a->s6_addr[i] > b->s6_addr[i] ? 1 : -1);
184 } /* int cmp_addrinfo */
186 static inline int ignore_list_match (const struct in6_addr *addr)
187 {
188 ip_list_t *ptr;
190 for (ptr = IgnoreList; ptr != NULL; ptr = ptr->next)
191 if (cmp_in6_addr (addr, &ptr->addr) == 0)
192 return (1);
193 return (0);
194 } /* int ignore_list_match */
196 static void ignore_list_add (const struct in6_addr *addr)
197 {
198 ip_list_t *new;
200 if (ignore_list_match (addr) != 0)
201 return;
203 new = malloc (sizeof (ip_list_t));
204 if (new == NULL)
205 {
206 perror ("malloc");
207 return;
208 }
210 memcpy (&new->addr, addr, sizeof (struct in6_addr));
211 new->next = IgnoreList;
213 IgnoreList = new;
214 } /* void ignore_list_add */
216 void ignore_list_add_name (const char *name)
217 {
218 struct addrinfo *ai_list;
219 struct addrinfo *ai_ptr;
220 struct in6_addr addr;
221 int status;
223 status = getaddrinfo (name, NULL, NULL, &ai_list);
224 if (status != 0)
225 return;
227 for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
228 {
229 if (ai_ptr->ai_family == AF_INET)
230 {
231 memset (&addr, '\0', sizeof (addr));
232 addr.s6_addr[10] = 0xFF;
233 addr.s6_addr[11] = 0xFF;
234 memcpy (addr.s6_addr + 12, &((struct sockaddr_in *) ai_ptr->ai_addr)->sin_addr, 4);
236 ignore_list_add (&addr);
237 }
238 else
239 {
240 ignore_list_add (&((struct sockaddr_in6 *) ai_ptr->ai_addr)->sin6_addr);
241 }
242 } /* for */
244 freeaddrinfo (ai_list);
245 }
247 #if HAVE_PCAP_H
248 static void in6_addr_from_buffer (struct in6_addr *ia,
249 const void *buf, size_t buf_len,
250 int family)
251 {
252 memset (ia, 0, sizeof (struct in6_addr));
253 if ((AF_INET == family) && (sizeof (uint32_t) == buf_len))
254 {
255 ia->s6_addr[10] = 0xFF;
256 ia->s6_addr[11] = 0xFF;
257 memcpy (ia->s6_addr + 12, buf, buf_len);
258 }
259 else if ((AF_INET6 == family) && (sizeof (struct in6_addr) == buf_len))
260 {
261 memcpy (ia, buf, buf_len);
262 }
263 } /* void in6_addr_from_buffer */
265 void dnstop_set_pcap_obj (pcap_t *po)
266 {
267 pcap_obj = po;
268 }
270 void dnstop_set_callback (void (*cb) (const rfc1035_header_t *))
271 {
272 Callback = cb;
273 }
275 #define RFC1035_MAXLABELSZ 63
276 static int
277 rfc1035NameUnpack(const char *buf, size_t sz, off_t * off, char *name, size_t ns
278 )
279 {
280 off_t no = 0;
281 unsigned char c;
282 size_t len;
283 assert(ns > 0);
284 do {
285 if ((*off) >= sz)
286 break;
287 c = *(buf + (*off));
288 if (c > 191) {
289 /* blasted compression */
290 unsigned short s;
291 off_t ptr;
292 memcpy(&s, buf + (*off), sizeof(s));
293 s = ntohs(s);
294 (*off) += sizeof(s);
295 /* Sanity check */
296 if ((*off) >= sz)
297 return 1;
298 ptr = s & 0x3FFF;
299 /* Make sure the pointer is inside this message */
300 if (ptr >= sz)
301 return 2;
302 return rfc1035NameUnpack(buf, sz, &ptr, name + no, ns - no);
303 } else if (c > RFC1035_MAXLABELSZ) {
304 /*
305 * "(The 10 and 01 combinations are reserved for future use.)"
306 */
307 break;
308 return 3;
309 } else {
310 (*off)++;
311 len = (size_t) c;
312 if (len == 0)
313 break;
314 if (len > (ns - 1))
315 len = ns - 1;
316 if ((*off) + len > sz) /* message is too short */
317 return 4;
318 memcpy(name + no, buf + (*off), len);
319 (*off) += len;
320 no += len;
321 *(name + (no++)) = '.';
322 }
323 } while (c > 0);
324 *(name + no - 1) = '\0';
325 /* make sure we didn't allow someone to overflow the name buffer */
326 assert(no <= ns);
327 return 0;
328 }
330 static int
331 handle_dns(const char *buf, int len,
332 const struct in6_addr *s_addr,
333 const struct in6_addr *d_addr)
334 {
335 rfc1035_header_t qh;
336 uint16_t us;
337 off_t offset;
338 char *t;
339 int x;
341 /* The DNS header is 12 bytes long */
342 if (len < 12)
343 return 0;
345 memcpy(&us, buf + 0, 2);
346 qh.id = ntohs(us);
348 memcpy(&us, buf + 2, 2);
349 us = ntohs(us);
350 fprintf (stderr, "Bytes 0, 1: 0x%04hx\n", us);
351 qh.qr = (us >> 15) & 0x01;
352 qh.opcode = (us >> 11) & 0x0F;
353 qh.aa = (us >> 10) & 0x01;
354 qh.tc = (us >> 9) & 0x01;
355 qh.rd = (us >> 8) & 0x01;
356 qh.ra = (us >> 7) & 0x01;
357 qh.z = (us >> 6) & 0x01;
358 qh.ad = (us >> 5) & 0x01;
359 qh.cd = (us >> 4) & 0x01;
360 qh.rcode = us & 0x0F;
362 memcpy(&us, buf + 4, 2);
363 qh.qdcount = ntohs(us);
365 memcpy(&us, buf + 6, 2);
366 qh.ancount = ntohs(us);
368 memcpy(&us, buf + 8, 2);
369 qh.nscount = ntohs(us);
371 memcpy(&us, buf + 10, 2);
372 qh.arcount = ntohs(us);
374 offset = 12;
375 memset(qh.qname, '\0', MAX_QNAME_SZ);
376 x = rfc1035NameUnpack(buf, len, &offset, qh.qname, MAX_QNAME_SZ);
377 if (0 != x)
378 return 0;
379 if ('\0' == qh.qname[0])
380 strcpy(qh.qname, ".");
381 while ((t = strchr(qh.qname, '\n')))
382 *t = ' ';
383 while ((t = strchr(qh.qname, '\r')))
384 *t = ' ';
385 for (t = qh.qname; *t; t++)
386 *t = tolower(*t);
388 memcpy(&us, buf + offset, 2);
389 qh.qtype = ntohs(us);
390 memcpy(&us, buf + offset + 2, 2);
391 qh.qclass = ntohs(us);
393 qh.length = (uint16_t) len;
395 /* gather stats */
396 qtype_counts[qh.qtype]++;
397 qclass_counts[qh.qclass]++;
398 opcode_counts[qh.opcode]++;
400 if (Callback != NULL)
401 Callback (&qh);
403 return 1;
404 }
406 static int
407 handle_udp(const struct udphdr *udp, int len,
408 const struct in6_addr *s_addr,
409 const struct in6_addr *d_addr)
410 {
411 char buf[PCAP_SNAPLEN];
412 if ((ntohs (udp->uh_dport) != 53)
413 && (ntohs (udp->uh_sport) != 53))
414 return 0;
415 memcpy(buf, udp + 1, len - sizeof(*udp));
416 if (0 == handle_dns(buf, len - sizeof(*udp), s_addr, d_addr))
417 return 0;
418 return 1;
419 }
421 static int
422 handle_ipv6 (struct ip6_hdr *ipv6, int len)
423 {
424 char buf[PCAP_SNAPLEN];
425 int offset;
426 int nexthdr;
428 struct in6_addr s_addr;
429 struct in6_addr d_addr;
430 uint16_t payload_len;
432 offset = sizeof (struct ip6_hdr);
433 nexthdr = ipv6->ip6_nxt;
434 s_addr = ipv6->ip6_src;
435 d_addr = ipv6->ip6_dst;
436 payload_len = ntohs (ipv6->ip6_plen);
438 if (ignore_list_match (&s_addr))
439 return (0);
441 /* Parse extension headers. This only handles the standard headers, as
442 * defined in RFC 2460, correctly. Fragments are discarded. */
443 while ((IPPROTO_ROUTING == nexthdr) /* routing header */
444 || (IPPROTO_HOPOPTS == nexthdr) /* Hop-by-Hop options. */
445 || (IPPROTO_FRAGMENT == nexthdr) /* fragmentation header. */
446 || (IPPROTO_DSTOPTS == nexthdr) /* destination options. */
447 || (IPPROTO_DSTOPTS == nexthdr) /* destination options. */
448 || (IPPROTO_AH == nexthdr) /* destination options. */
449 || (IPPROTO_ESP == nexthdr)) /* encapsulating security payload. */
450 {
451 struct ip6_ext ext_hdr;
452 uint16_t ext_hdr_len;
454 /* Catch broken packets */
455 if ((offset + sizeof (struct ip6_ext)) > len)
456 return (0);
458 /* Cannot handle fragments. */
459 if (IPPROTO_FRAGMENT == nexthdr)
460 return (0);
462 memcpy (&ext_hdr, (char *) ipv6 + offset, sizeof (struct ip6_ext));
463 nexthdr = ext_hdr.ip6e_nxt;
464 ext_hdr_len = (8 * (ntohs (ext_hdr.ip6e_len) + 1));
466 /* This header is longer than the packets payload.. WTF? */
467 if (ext_hdr_len > payload_len)
468 return (0);
470 offset += ext_hdr_len;
471 payload_len -= ext_hdr_len;
472 } /* while */
474 /* Catch broken and empty packets */
475 if (((offset + payload_len) > len)
476 || (payload_len == 0)
477 || (payload_len > PCAP_SNAPLEN))
478 return (0);
480 if (IPPROTO_UDP != nexthdr)
481 return (0);
483 memcpy (buf, (char *) ipv6 + offset, payload_len);
484 if (handle_udp ((struct udphdr *) buf, payload_len, &s_addr, &d_addr) == 0)
485 return (0);
487 return (1); /* Success */
488 } /* int handle_ipv6 */
490 static int
491 handle_ip(const struct ip *ip, int len)
492 {
493 char buf[PCAP_SNAPLEN];
494 int offset = ip->ip_hl << 2;
495 struct in6_addr s_addr;
496 struct in6_addr d_addr;
498 if (ip->ip_v == 6)
499 return (handle_ipv6 ((struct ip6_hdr *) ip, len));
501 in6_addr_from_buffer (&s_addr, &ip->ip_src.s_addr, sizeof (ip->ip_src.s_addr), AF_INET);
502 in6_addr_from_buffer (&d_addr, &ip->ip_dst.s_addr, sizeof (ip->ip_dst.s_addr), AF_INET);
503 if (ignore_list_match (&s_addr))
504 return (0);
505 if (IPPROTO_UDP != ip->ip_p)
506 return 0;
507 memcpy(buf, (void *) ip + offset, len - offset);
508 if (0 == handle_udp((struct udphdr *) buf, len - offset, &s_addr, &d_addr))
509 return 0;
510 return 1;
511 }
513 #if HAVE_NET_IF_PPP_H
514 static int
515 handle_ppp(const u_char * pkt, int len)
516 {
517 char buf[PCAP_SNAPLEN];
518 unsigned short us;
519 unsigned short proto;
520 if (len < 2)
521 return 0;
522 if (*pkt == PPP_ADDRESS_VAL && *(pkt + 1) == PPP_CONTROL_VAL) {
523 pkt += 2; /* ACFC not used */
524 len -= 2;
525 }
526 if (len < 2)
527 return 0;
528 if (*pkt % 2) {
529 proto = *pkt; /* PFC is used */
530 pkt++;
531 len--;
532 } else {
533 memcpy(&us, pkt, sizeof(us));
534 proto = ntohs(us);
535 pkt += 2;
536 len -= 2;
537 }
538 if (ETHERTYPE_IP != proto && PPP_IP != proto)
539 return 0;
540 memcpy(buf, pkt, len);
541 return handle_ip((struct ip *) buf, len);
542 }
543 #endif /* HAVE_NET_IF_PPP_H */
545 static int
546 handle_null(const u_char * pkt, int len)
547 {
548 unsigned int family;
549 memcpy(&family, pkt, sizeof(family));
550 if (AF_INET != family)
551 return 0;
552 return handle_ip((struct ip *) (pkt + 4), len - 4);
553 }
555 #ifdef DLT_LOOP
556 static int
557 handle_loop(const u_char * pkt, int len)
558 {
559 unsigned int family;
560 memcpy(&family, pkt, sizeof(family));
561 if (AF_INET != ntohl(family))
562 return 0;
563 return handle_ip((struct ip *) (pkt + 4), len - 4);
564 }
566 #endif
568 #ifdef DLT_RAW
569 static int
570 handle_raw(const u_char * pkt, int len)
571 {
572 return handle_ip((struct ip *) pkt, len);
573 }
575 #endif
577 static int
578 handle_ether(const u_char * pkt, int len)
579 {
580 char buf[PCAP_SNAPLEN];
581 struct ether_header *e = (void *) pkt;
582 unsigned short etype = ntohs(e->ether_type);
583 if (len < ETHER_HDR_LEN)
584 return 0;
585 pkt += ETHER_HDR_LEN;
586 len -= ETHER_HDR_LEN;
587 if (ETHERTYPE_8021Q == etype) {
588 etype = ntohs(*(unsigned short *) (pkt + 2));
589 pkt += 4;
590 len -= 4;
591 }
592 if ((ETHERTYPE_IP != etype)
593 && (ETHERTYPE_IPV6 != etype))
594 return 0;
595 memcpy(buf, pkt, len);
596 if (ETHERTYPE_IPV6 == etype)
597 return (handle_ipv6 ((struct ip6_hdr *) buf, len));
598 else
599 return handle_ip((struct ip *) buf, len);
600 }
602 /* public function */
603 void handle_pcap(u_char *udata, const struct pcap_pkthdr *hdr, const u_char *pkt)
604 {
605 int status;
607 fprintf (stderr, "handle_pcap (udata = %p, hdr = %p, pkt = %p): hdr->caplen = %i\n",
608 (void *) udata, (void *) hdr, (void *) pkt,
609 hdr->caplen);
611 if (hdr->caplen < ETHER_HDR_LEN)
612 return;
614 switch (pcap_datalink (pcap_obj))
615 {
616 case DLT_EN10MB:
617 status = handle_ether (pkt, hdr->caplen);
618 break;
619 #if HAVE_NET_IF_PPP_H
620 case DLT_PPP:
621 status = handle_ppp (pkt, hdr->caplen);
622 break;
623 #endif
624 #ifdef DLT_LOOP
625 case DLT_LOOP:
626 status = handle_loop (pkt, hdr->caplen);
627 break;
628 #endif
629 #ifdef DLT_RAW
630 case DLT_RAW:
631 status = handle_raw (pkt, hdr->caplen);
632 break;
633 #endif
634 case DLT_NULL:
635 status = handle_null (pkt, hdr->caplen);
636 break;
638 default:
639 fprintf (stderr, "unsupported data link type %d\n",
640 pcap_datalink(pcap_obj));
641 status = 0;
642 break;
643 } /* switch (pcap_datalink(pcap_obj)) */
645 if (0 == status)
646 return;
648 query_count_intvl++;
649 query_count_total++;
650 last_ts = hdr->ts;
651 }
652 #endif /* HAVE_PCAP_H */
654 const char *qtype_str(int t)
655 {
656 static char buf[32];
657 switch (t) {
658 case ns_t_a: return ("A");
659 case ns_t_ns: return ("NS");
660 case ns_t_md: return ("MD");
661 case ns_t_mf: return ("MF");
662 case ns_t_cname: return ("CNAME");
663 case ns_t_soa: return ("SOA");
664 case ns_t_mb: return ("MB");
665 case ns_t_mg: return ("MG");
666 case ns_t_mr: return ("MR");
667 case ns_t_null: return ("NULL");
668 case ns_t_wks: return ("WKS");
669 case ns_t_ptr: return ("PTR");
670 case ns_t_hinfo: return ("HINFO");
671 case ns_t_minfo: return ("MINFO");
672 case ns_t_mx: return ("MX");
673 case ns_t_txt: return ("TXT");
674 case ns_t_rp: return ("RP");
675 case ns_t_afsdb: return ("AFSDB");
676 case ns_t_x25: return ("X25");
677 case ns_t_isdn: return ("ISDN");
678 case ns_t_rt: return ("RT");
679 case ns_t_nsap: return ("NSAP");
680 case ns_t_nsap_ptr: return ("NSAP-PTR");
681 case ns_t_sig: return ("SIG");
682 case ns_t_key: return ("KEY");
683 case ns_t_px: return ("PX");
684 case ns_t_gpos: return ("GPOS");
685 case ns_t_aaaa: return ("AAAA");
686 case ns_t_loc: return ("LOC");
687 case ns_t_nxt: return ("NXT");
688 case ns_t_eid: return ("EID");
689 case ns_t_nimloc: return ("NIMLOC");
690 case ns_t_srv: return ("SRV");
691 case ns_t_atma: return ("ATMA");
692 case ns_t_naptr: return ("NAPTR");
693 case ns_t_kx: return ("KX");
694 case ns_t_cert: return ("CERT");
695 case ns_t_a6: return ("A6");
696 case ns_t_dname: return ("DNAME");
697 case ns_t_sink: return ("SINK");
698 case ns_t_opt: return ("OPT");
699 case ns_t_tsig: return ("TSIG");
700 case ns_t_ixfr: return ("IXFR");
701 case ns_t_axfr: return ("AXFR");
702 case ns_t_mailb: return ("MAILB");
703 case ns_t_maila: return ("MAILA");
704 case ns_t_any: return ("ANY");
705 case ns_t_zxfr: return ("ZXFR");
706 default:
707 snprintf (buf, 32, "#%i", t);
708 buf[31] = '\0';
709 return (buf);
710 }; /* switch (t) */
711 /* NOTREACHED */
712 return (NULL);
713 }
715 const char *opcode_str (int o)
716 {
717 static char buf[30];
718 switch (o) {
719 case 0:
720 return "Query";
721 break;
722 case 1:
723 return "Iquery";
724 break;
725 case 2:
726 return "Status";
727 break;
728 case 4:
729 return "Notify";
730 break;
731 case 5:
732 return "Update";
733 break;
734 default:
735 snprintf(buf, 30, "Opcode%d", o);
736 return buf;
737 }
738 /* NOTREACHED */
739 }
741 const char *rcode_str (int rcode)
742 {
743 static char buf[32];
744 switch (rcode)
745 {
746 case ns_r_noerror: return ("NOERROR");
747 case ns_r_formerr: return ("FORMERR");
748 case ns_r_servfail: return ("SERVFAIL");
749 case ns_r_nxdomain: return ("NXDOMAIN");
750 case ns_r_notimpl: return ("NOTIMPL");
751 case ns_r_refused: return ("REFUSED");
752 case ns_r_yxdomain: return ("YXDOMAIN");
753 case ns_r_yxrrset: return ("YXRRSET");
754 case ns_r_nxrrset: return ("NXRRSET");
755 case ns_r_notauth: return ("NOTAUTH");
756 case ns_r_notzone: return ("NOTZONE");
757 case ns_r_max: return ("MAX");
758 case ns_r_badsig: return ("BADSIG");
759 case ns_r_badkey: return ("BADKEY");
760 case ns_r_badtime: return ("BADTIME");
761 default:
762 snprintf (buf, 32, "RCode%i", rcode);
763 buf[31] = '\0';
764 return (buf);
765 }
766 /* Never reached */
767 return (NULL);
768 } /* const char *rcode_str (int rcode) */
770 #if 0
771 static int
772 main(int argc, char *argv[])
773 {
774 char errbuf[PCAP_ERRBUF_SIZE];
775 int x;
776 struct stat sb;
777 int readfile_state = 0;
778 struct bpf_program fp;
780 port53 = htons(53);
781 SubReport = Sources_report;
782 ignore_addr.s_addr = 0;
783 progname = strdup(strrchr(argv[0], '/') ? strchr(argv[0], '/') + 1 : argv[0]);
784 srandom(time(NULL));
785 ResetCounters();
787 while ((x = getopt(argc, argv, "ab:f:i:pst")) != -1) {
788 switch (x) {
789 case 'a':
790 anon_flag = 1;
791 break;
792 case 's':
793 sld_flag = 1;
794 break;
795 case 't':
796 nld_flag = 1;
797 break;
798 case 'p':
799 promisc_flag = 0;
800 break;
801 case 'b':
802 bpf_program_str = strdup(optarg);
803 break;
804 case 'i':
805 ignore_addr.s_addr = inet_addr(optarg);
806 break;
807 case 'f':
808 set_filter(optarg);
809 break;
810 default:
811 usage();
812 break;
813 }
814 }
815 argc -= optind;
816 argv += optind;
818 if (argc < 1)
819 usage();
820 device = strdup(argv[0]);
822 if (0 == stat(device, &sb))
823 readfile_state = 1;
824 if (readfile_state) {
825 pcap_obj = pcap_open_offline(device, errbuf);
826 } else {
827 pcap_obj = pcap_open_live(device, PCAP_SNAPLEN, promisc_flag, 1000, errbuf);
828 }
829 if (NULL == pcap_obj) {
830 fprintf(stderr, "pcap_open_*: %s\n", errbuf);
831 exit(1);
832 }
834 if (0 == isatty(1)) {
835 if (0 == readfile_state) {
836 fprintf(stderr, "Non-interactive mode requires savefile argument\n");
837 exit(1);
838 }
839 interactive = 0;
840 print_func = printf;
841 }
843 memset(&fp, '\0', sizeof(fp));
844 x = pcap_compile(pcap_obj, &fp, bpf_program_str, 1, 0);
845 if (x < 0) {
846 fprintf(stderr, "pcap_compile failed\n");
847 exit(1);
848 }
849 x = pcap_setfilter(pcap_obj, &fp);
850 if (x < 0) {
851 fprintf(stderr, "pcap_setfilter failed\n");
852 exit(1);
853 }
855 /*
856 * non-blocking call added for Mac OS X bugfix. Sent by Max Horn.
857 * ref http://www.tcpdump.org/lists/workers/2002/09/msg00033.html
858 */
859 x = pcap_setnonblock(pcap_obj, 1, errbuf);
860 if (x < 0) {
861 fprintf(stderr, "pcap_setnonblock failed: %s\n", errbuf);
862 exit(1);
863 }
865 switch (pcap_datalink(pcap_obj)) {
866 case DLT_EN10MB:
867 handle_datalink = handle_ether;
868 break;
869 #if HAVE_NET_IF_PPP_H
870 case DLT_PPP:
871 handle_datalink = handle_ppp;
872 break;
873 #endif
874 #ifdef DLT_LOOP
875 case DLT_LOOP:
876 handle_datalink = handle_loop;
877 break;
878 #endif
879 #ifdef DLT_RAW
880 case DLT_RAW:
881 handle_datalink = handle_raw;
882 break;
883 #endif
884 case DLT_NULL:
885 handle_datalink = handle_null;
886 break;
887 default:
888 fprintf(stderr, "unsupported data link type %d\n",
889 pcap_datalink(pcap_obj));
890 return 1;
891 break;
892 }
893 if (interactive) {
894 init_curses();
895 while (0 == Quit) {
896 if (readfile_state < 2) {
897 /*
898 * On some OSes select() might return 0 even when
899 * there are packets to process. Thus, we always
900 * ignore its return value and just call pcap_dispatch()
901 * anyway.
902 */
903 if (0 == readfile_state) /* interactive */
904 pcap_select(pcap_obj, 1, 0);
905 x = pcap_dispatch(pcap_obj, 50, handle_pcap, NULL);
906 }
907 if (0 == x && 1 == readfile_state) {
908 /* block on keyboard until user quits */
909 readfile_state++;
910 nodelay(w, 0);
911 }
912 keyboard();
913 cron_pre();
914 report();
915 cron_post();
916 }
917 endwin(); /* klin, Thu Nov 28 08:56:51 2002 */
918 } else {
919 while (pcap_dispatch(pcap_obj, 50, handle_pcap, NULL))
920 (void) 0;
921 cron_pre();
922 Sources_report(); print_func("\n");
923 Destinatioreport(); print_func("\n");
924 Qtypes_report(); print_func("\n");
925 Opcodes_report(); print_func("\n");
926 Tld_report(); print_func("\n");
927 Sld_report(); print_func("\n");
928 Nld_report(); print_func("\n");
929 SldBySource_report();
930 }
932 pcap_close(pcap_obj);
933 return 0;
934 } /* static int main(int argc, char *argv[]) */
935 #endif
936 /*
937 * vim:shiftwidth=4:tabstop=8:softtabstop=4
938 */