2 * collectd - src/utils_dns.c
3 * Modifications Copyright (C) 2006 Florian octo Forster
4 * Copyright (C) 2002 The Measurement Factory, Inc.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are met:
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.
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.
32 * The Measurement Factory, Inc. <http://www.measurement-factory.com/>
33 * Florian octo Forster <octo at verplant.org>
38 #if HAVE_NETINET_IN_SYSTM_H
39 # include <netinet/in_systm.h>
42 # include <netinet/in.h>
45 # include <arpa/inet.h>
48 # include <sys/socket.h>
51 #if HAVE_ARPA_NAMESER_H
52 # include <arpa/nameser.h>
53 #elif HAVE_ARPA_NAMESER_COMPAT_H
54 # include <arpa/nameser_compat.h>
58 # include <net/if_arp.h>
63 #if HAVE_NETINET_IF_ETHER_H
64 # include <netinet/if_ether.h>
66 #if HAVE_NET_PPP_DEFS_H
67 # include <net/ppp_defs.h>
70 # include <net/if_ppp.h>
78 # include <netinet/ip.h>
80 #ifdef HAVE_NETINET_IP_VAR_H
81 # include <netinet/ip_var.h>
83 #if HAVE_NETINET_IP6_H
84 # include <netinet/ip6.h>
86 #if HAVE_NETINET_UDP_H
87 # include <netinet/udp.h>
94 #define PCAP_SNAPLEN 1460
96 #define ETHER_ADDR_LEN 6
97 #define ETHER_TYPE_LEN 2
98 #define ETHER_HDR_LEN (ETHER_ADDR_LEN * 2 + ETHER_TYPE_LEN)
100 #ifndef ETHERTYPE_8021Q
101 # define ETHERTYPE_8021Q 0x8100
103 #ifndef ETHERTYPE_IPV6
104 # define ETHERTYPE_IPV6 0x86DD
107 #ifndef PPP_ADDRESS_VAL
108 # define PPP_ADDRESS_VAL 0xff /* The address byte value */
110 #ifndef PPP_CONTROL_VAL
111 # define PPP_CONTROL_VAL 0x03 /* The control byte value */
114 #if HAVE_STRUCT_UDPHDR_UH_DPORT && HAVE_STRUCT_UDPHDR_UH_SPORT
115 # define UDP_DEST uh_dport
116 # define UDP_SRC uh_dport
117 #elif HAVE_STRUCT_UDPHDR_DEST && HAVE_STRUCT_UDPHDR_SOURCE
118 # define UDP_DEST dest
119 # define UDP_SRC source
121 # error "`struct udphdr' is unusable."
124 #include "utils_dns.h"
131 struct in6_addr addr;
133 struct ip_list_s *next;
135 typedef struct ip_list_s ip_list_t;
137 typedef int (printer)(const char *, ...);
140 * flags/features for non-interactive mode
153 int qtype_counts[T_MAX];
154 int opcode_counts[OP_MAX];
155 int qclass_counts[C_MAX];
158 static pcap_t *pcap_obj = NULL;
161 static ip_list_t *IgnoreList = NULL;
164 static void (*Callback) (const rfc1035_header_t *) = NULL;
166 static int query_count_intvl = 0;
167 static int query_count_total = 0;
169 static struct bpf_timeval last_ts;
171 static struct timeval last_ts;
172 # endif /* __OpenBSD__ */
173 #endif /* HAVE_PCAP_H */
175 static int cmp_in6_addr (const struct in6_addr *a,
176 const struct in6_addr *b)
180 assert (sizeof (struct in6_addr) == 16);
182 for (i = 0; i < 16; i++)
183 if (a->s6_addr[i] != b->s6_addr[i])
189 return (a->s6_addr[i] > b->s6_addr[i] ? 1 : -1);
190 } /* int cmp_addrinfo */
192 static inline int ignore_list_match (const struct in6_addr *addr)
196 for (ptr = IgnoreList; ptr != NULL; ptr = ptr->next)
197 if (cmp_in6_addr (addr, &ptr->addr) == 0)
200 } /* int ignore_list_match */
202 static void ignore_list_add (const struct in6_addr *addr)
206 if (ignore_list_match (addr) != 0)
209 new = malloc (sizeof (ip_list_t));
216 memcpy (&new->addr, addr, sizeof (struct in6_addr));
217 new->next = IgnoreList;
220 } /* void ignore_list_add */
222 void ignore_list_add_name (const char *name)
224 struct addrinfo *ai_list;
225 struct addrinfo *ai_ptr;
226 struct in6_addr addr;
229 status = getaddrinfo (name, NULL, NULL, &ai_list);
233 for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
235 if (ai_ptr->ai_family == AF_INET)
237 memset (&addr, '\0', sizeof (addr));
238 addr.s6_addr[10] = 0xFF;
239 addr.s6_addr[11] = 0xFF;
240 memcpy (addr.s6_addr + 12, &((struct sockaddr_in *) ai_ptr->ai_addr)->sin_addr, 4);
242 ignore_list_add (&addr);
246 ignore_list_add (&((struct sockaddr_in6 *) ai_ptr->ai_addr)->sin6_addr);
250 freeaddrinfo (ai_list);
254 static void in6_addr_from_buffer (struct in6_addr *ia,
255 const void *buf, size_t buf_len,
258 memset (ia, 0, sizeof (struct in6_addr));
259 if ((AF_INET == family) && (sizeof (uint32_t) == buf_len))
261 ia->s6_addr[10] = 0xFF;
262 ia->s6_addr[11] = 0xFF;
263 memcpy (ia->s6_addr + 12, buf, buf_len);
265 else if ((AF_INET6 == family) && (sizeof (struct in6_addr) == buf_len))
267 memcpy (ia, buf, buf_len);
269 } /* void in6_addr_from_buffer */
271 void dnstop_set_pcap_obj (pcap_t *po)
276 void dnstop_set_callback (void (*cb) (const rfc1035_header_t *))
281 #define RFC1035_MAXLABELSZ 63
283 rfc1035NameUnpack(const char *buf, size_t sz, off_t * off, char *name, size_t ns
295 /* blasted compression */
298 memcpy(&s, buf + (*off), sizeof(s));
305 /* Make sure the pointer is inside this message */
308 return rfc1035NameUnpack(buf, sz, &ptr, name + no, ns - no);
309 } else if (c > RFC1035_MAXLABELSZ) {
311 * "(The 10 and 01 combinations are reserved for future use.)"
322 if ((*off) + len > sz) /* message is too short */
324 memcpy(name + no, buf + (*off), len);
327 *(name + (no++)) = '.';
330 *(name + no - 1) = '\0';
331 /* make sure we didn't allow someone to overflow the name buffer */
337 handle_dns(const char *buf, int len,
338 const struct in6_addr *s_addr,
339 const struct in6_addr *d_addr)
347 /* The DNS header is 12 bytes long */
351 memcpy(&us, buf + 0, 2);
354 memcpy(&us, buf + 2, 2);
356 fprintf (stderr, "Bytes 0, 1: 0x%04hx\n", us);
357 qh.qr = (us >> 15) & 0x01;
358 qh.opcode = (us >> 11) & 0x0F;
359 qh.aa = (us >> 10) & 0x01;
360 qh.tc = (us >> 9) & 0x01;
361 qh.rd = (us >> 8) & 0x01;
362 qh.ra = (us >> 7) & 0x01;
363 qh.z = (us >> 6) & 0x01;
364 qh.ad = (us >> 5) & 0x01;
365 qh.cd = (us >> 4) & 0x01;
366 qh.rcode = us & 0x0F;
368 memcpy(&us, buf + 4, 2);
369 qh.qdcount = ntohs(us);
371 memcpy(&us, buf + 6, 2);
372 qh.ancount = ntohs(us);
374 memcpy(&us, buf + 8, 2);
375 qh.nscount = ntohs(us);
377 memcpy(&us, buf + 10, 2);
378 qh.arcount = ntohs(us);
381 memset(qh.qname, '\0', MAX_QNAME_SZ);
382 x = rfc1035NameUnpack(buf, len, &offset, qh.qname, MAX_QNAME_SZ);
385 if ('\0' == qh.qname[0])
386 strcpy(qh.qname, ".");
387 while ((t = strchr(qh.qname, '\n')))
389 while ((t = strchr(qh.qname, '\r')))
391 for (t = qh.qname; *t; t++)
394 memcpy(&us, buf + offset, 2);
395 qh.qtype = ntohs(us);
396 memcpy(&us, buf + offset + 2, 2);
397 qh.qclass = ntohs(us);
399 qh.length = (uint16_t) len;
402 qtype_counts[qh.qtype]++;
403 qclass_counts[qh.qclass]++;
404 opcode_counts[qh.opcode]++;
406 if (Callback != NULL)
413 handle_udp(const struct udphdr *udp, int len,
414 const struct in6_addr *s_addr,
415 const struct in6_addr *d_addr)
417 char buf[PCAP_SNAPLEN];
418 if ((ntohs (udp->UDP_DEST) != 53)
419 && (ntohs (udp->UDP_SRC) != 53))
421 memcpy(buf, udp + 1, len - sizeof(*udp));
422 if (0 == handle_dns(buf, len - sizeof(*udp), s_addr, d_addr))
428 handle_ipv6 (struct ip6_hdr *ipv6, int len)
430 char buf[PCAP_SNAPLEN];
434 struct in6_addr s_addr;
435 struct in6_addr d_addr;
436 uint16_t payload_len;
438 offset = sizeof (struct ip6_hdr);
439 nexthdr = ipv6->ip6_nxt;
440 s_addr = ipv6->ip6_src;
441 d_addr = ipv6->ip6_dst;
442 payload_len = ntohs (ipv6->ip6_plen);
444 if (ignore_list_match (&s_addr))
447 /* Parse extension headers. This only handles the standard headers, as
448 * defined in RFC 2460, correctly. Fragments are discarded. */
449 while ((IPPROTO_ROUTING == nexthdr) /* routing header */
450 || (IPPROTO_HOPOPTS == nexthdr) /* Hop-by-Hop options. */
451 || (IPPROTO_FRAGMENT == nexthdr) /* fragmentation header. */
452 || (IPPROTO_DSTOPTS == nexthdr) /* destination options. */
453 || (IPPROTO_DSTOPTS == nexthdr) /* destination options. */
454 || (IPPROTO_AH == nexthdr) /* destination options. */
455 || (IPPROTO_ESP == nexthdr)) /* encapsulating security payload. */
457 struct ip6_ext ext_hdr;
458 uint16_t ext_hdr_len;
460 /* Catch broken packets */
461 if ((offset + sizeof (struct ip6_ext)) > len)
464 /* Cannot handle fragments. */
465 if (IPPROTO_FRAGMENT == nexthdr)
468 memcpy (&ext_hdr, (char *) ipv6 + offset, sizeof (struct ip6_ext));
469 nexthdr = ext_hdr.ip6e_nxt;
470 ext_hdr_len = (8 * (ntohs (ext_hdr.ip6e_len) + 1));
472 /* This header is longer than the packets payload.. WTF? */
473 if (ext_hdr_len > payload_len)
476 offset += ext_hdr_len;
477 payload_len -= ext_hdr_len;
480 /* Catch broken and empty packets */
481 if (((offset + payload_len) > len)
482 || (payload_len == 0)
483 || (payload_len > PCAP_SNAPLEN))
486 if (IPPROTO_UDP != nexthdr)
489 memcpy (buf, (char *) ipv6 + offset, payload_len);
490 if (handle_udp ((struct udphdr *) buf, payload_len, &s_addr, &d_addr) == 0)
493 return (1); /* Success */
494 } /* int handle_ipv6 */
497 handle_ip(const struct ip *ip, int len)
499 char buf[PCAP_SNAPLEN];
500 int offset = ip->ip_hl << 2;
501 struct in6_addr s_addr;
502 struct in6_addr d_addr;
505 return (handle_ipv6 ((struct ip6_hdr *) ip, len));
507 in6_addr_from_buffer (&s_addr, &ip->ip_src.s_addr, sizeof (ip->ip_src.s_addr), AF_INET);
508 in6_addr_from_buffer (&d_addr, &ip->ip_dst.s_addr, sizeof (ip->ip_dst.s_addr), AF_INET);
509 if (ignore_list_match (&s_addr))
511 if (IPPROTO_UDP != ip->ip_p)
513 memcpy(buf, (void *) ip + offset, len - offset);
514 if (0 == handle_udp((struct udphdr *) buf, len - offset, &s_addr, &d_addr))
519 #if HAVE_NET_IF_PPP_H
521 handle_ppp(const u_char * pkt, int len)
523 char buf[PCAP_SNAPLEN];
525 unsigned short proto;
528 if (*pkt == PPP_ADDRESS_VAL && *(pkt + 1) == PPP_CONTROL_VAL) {
529 pkt += 2; /* ACFC not used */
535 proto = *pkt; /* PFC is used */
539 memcpy(&us, pkt, sizeof(us));
544 if (ETHERTYPE_IP != proto && PPP_IP != proto)
546 memcpy(buf, pkt, len);
547 return handle_ip((struct ip *) buf, len);
549 #endif /* HAVE_NET_IF_PPP_H */
552 handle_null(const u_char * pkt, int len)
555 memcpy(&family, pkt, sizeof(family));
556 if (AF_INET != family)
558 return handle_ip((struct ip *) (pkt + 4), len - 4);
563 handle_loop(const u_char * pkt, int len)
566 memcpy(&family, pkt, sizeof(family));
567 if (AF_INET != ntohl(family))
569 return handle_ip((struct ip *) (pkt + 4), len - 4);
576 handle_raw(const u_char * pkt, int len)
578 return handle_ip((struct ip *) pkt, len);
584 handle_ether(const u_char * pkt, int len)
586 char buf[PCAP_SNAPLEN];
587 struct ether_header *e = (void *) pkt;
588 unsigned short etype = ntohs(e->ether_type);
589 if (len < ETHER_HDR_LEN)
591 pkt += ETHER_HDR_LEN;
592 len -= ETHER_HDR_LEN;
593 if (ETHERTYPE_8021Q == etype) {
594 etype = ntohs(*(unsigned short *) (pkt + 2));
598 if ((ETHERTYPE_IP != etype)
599 && (ETHERTYPE_IPV6 != etype))
601 memcpy(buf, pkt, len);
602 if (ETHERTYPE_IPV6 == etype)
603 return (handle_ipv6 ((struct ip6_hdr *) buf, len));
605 return handle_ip((struct ip *) buf, len);
610 handle_linux_sll (const u_char *pkt, int len)
622 if (len < sizeof (struct sll_header))
625 hdr = (struct sll_header *) pkt;
626 pkt = (u_char *) (hdr + 1);
627 len -= sizeof (struct sll_header);
629 etype = ntohs (hdr->proto_type);
631 if ((ETHERTYPE_IP != etype)
632 && (ETHERTYPE_IPV6 != etype))
635 if (ETHERTYPE_IPV6 == etype)
636 return (handle_ipv6 ((struct ip6_hdr *) pkt, len));
638 return handle_ip((struct ip *) pkt, len);
640 #endif /* DLT_LINUX_SLL */
642 /* public function */
643 void handle_pcap(u_char *udata, const struct pcap_pkthdr *hdr, const u_char *pkt)
647 fprintf (stderr, "handle_pcap (udata = %p, hdr = %p, pkt = %p): hdr->caplen = %i\n",
648 (void *) udata, (void *) hdr, (void *) pkt,
651 if (hdr->caplen < ETHER_HDR_LEN)
654 switch (pcap_datalink (pcap_obj))
657 status = handle_ether (pkt, hdr->caplen);
659 #if HAVE_NET_IF_PPP_H
661 status = handle_ppp (pkt, hdr->caplen);
666 status = handle_loop (pkt, hdr->caplen);
671 status = handle_raw (pkt, hdr->caplen);
676 status = handle_linux_sll (pkt, hdr->caplen);
680 status = handle_null (pkt, hdr->caplen);
684 fprintf (stderr, "unsupported data link type %d\n",
685 pcap_datalink(pcap_obj));
688 } /* switch (pcap_datalink(pcap_obj)) */
697 #endif /* HAVE_PCAP_H */
699 const char *qtype_str(int t)
703 case ns_t_a: return ("A");
704 case ns_t_ns: return ("NS");
705 case ns_t_md: return ("MD");
706 case ns_t_mf: return ("MF");
707 case ns_t_cname: return ("CNAME");
708 case ns_t_soa: return ("SOA");
709 case ns_t_mb: return ("MB");
710 case ns_t_mg: return ("MG");
711 case ns_t_mr: return ("MR");
712 case ns_t_null: return ("NULL");
713 case ns_t_wks: return ("WKS");
714 case ns_t_ptr: return ("PTR");
715 case ns_t_hinfo: return ("HINFO");
716 case ns_t_minfo: return ("MINFO");
717 case ns_t_mx: return ("MX");
718 case ns_t_txt: return ("TXT");
719 case ns_t_rp: return ("RP");
720 case ns_t_afsdb: return ("AFSDB");
721 case ns_t_x25: return ("X25");
722 case ns_t_isdn: return ("ISDN");
723 case ns_t_rt: return ("RT");
724 case ns_t_nsap: return ("NSAP");
725 case ns_t_nsap_ptr: return ("NSAP-PTR");
726 case ns_t_sig: return ("SIG");
727 case ns_t_key: return ("KEY");
728 case ns_t_px: return ("PX");
729 case ns_t_gpos: return ("GPOS");
730 case ns_t_aaaa: return ("AAAA");
731 case ns_t_loc: return ("LOC");
732 case ns_t_nxt: return ("NXT");
733 case ns_t_eid: return ("EID");
734 case ns_t_nimloc: return ("NIMLOC");
735 case ns_t_srv: return ("SRV");
736 case ns_t_atma: return ("ATMA");
737 case ns_t_naptr: return ("NAPTR");
738 case ns_t_kx: return ("KX");
739 case ns_t_cert: return ("CERT");
740 case ns_t_a6: return ("A6");
741 case ns_t_dname: return ("DNAME");
742 case ns_t_sink: return ("SINK");
743 case ns_t_opt: return ("OPT");
744 case ns_t_tsig: return ("TSIG");
745 case ns_t_ixfr: return ("IXFR");
746 case ns_t_axfr: return ("AXFR");
747 case ns_t_mailb: return ("MAILB");
748 case ns_t_maila: return ("MAILA");
749 case ns_t_any: return ("ANY");
750 case ns_t_zxfr: return ("ZXFR");
752 snprintf (buf, 32, "#%i", t);
760 const char *opcode_str (int o)
780 snprintf(buf, 30, "Opcode%d", o);
786 const char *rcode_str (int rcode)
791 case ns_r_noerror: return ("NOERROR");
792 case ns_r_formerr: return ("FORMERR");
793 case ns_r_servfail: return ("SERVFAIL");
794 case ns_r_nxdomain: return ("NXDOMAIN");
795 case ns_r_notimpl: return ("NOTIMPL");
796 case ns_r_refused: return ("REFUSED");
797 case ns_r_yxdomain: return ("YXDOMAIN");
798 case ns_r_yxrrset: return ("YXRRSET");
799 case ns_r_nxrrset: return ("NXRRSET");
800 case ns_r_notauth: return ("NOTAUTH");
801 case ns_r_notzone: return ("NOTZONE");
802 case ns_r_max: return ("MAX");
803 case ns_r_badsig: return ("BADSIG");
804 case ns_r_badkey: return ("BADKEY");
805 case ns_r_badtime: return ("BADTIME");
807 snprintf (buf, 32, "RCode%i", rcode);
813 } /* const char *rcode_str (int rcode) */
817 main(int argc, char *argv[])
819 char errbuf[PCAP_ERRBUF_SIZE];
822 int readfile_state = 0;
823 struct bpf_program fp;
826 SubReport = Sources_report;
827 ignore_addr.s_addr = 0;
828 progname = strdup(strrchr(argv[0], '/') ? strchr(argv[0], '/') + 1 : argv[0]);
832 while ((x = getopt(argc, argv, "ab:f:i:pst")) != -1) {
847 bpf_program_str = strdup(optarg);
850 ignore_addr.s_addr = inet_addr(optarg);
865 device = strdup(argv[0]);
867 if (0 == stat(device, &sb))
869 if (readfile_state) {
870 pcap_obj = pcap_open_offline(device, errbuf);
872 pcap_obj = pcap_open_live(device, PCAP_SNAPLEN, promisc_flag, 1000, errbuf);
874 if (NULL == pcap_obj) {
875 fprintf(stderr, "pcap_open_*: %s\n", errbuf);
879 if (0 == isatty(1)) {
880 if (0 == readfile_state) {
881 fprintf(stderr, "Non-interactive mode requires savefile argument\n");
888 memset(&fp, '\0', sizeof(fp));
889 x = pcap_compile(pcap_obj, &fp, bpf_program_str, 1, 0);
891 fprintf(stderr, "pcap_compile failed\n");
894 x = pcap_setfilter(pcap_obj, &fp);
896 fprintf(stderr, "pcap_setfilter failed\n");
901 * non-blocking call added for Mac OS X bugfix. Sent by Max Horn.
902 * ref http://www.tcpdump.org/lists/workers/2002/09/msg00033.html
904 x = pcap_setnonblock(pcap_obj, 1, errbuf);
906 fprintf(stderr, "pcap_setnonblock failed: %s\n", errbuf);
910 switch (pcap_datalink(pcap_obj)) {
912 handle_datalink = handle_ether;
914 #if HAVE_NET_IF_PPP_H
916 handle_datalink = handle_ppp;
921 handle_datalink = handle_loop;
926 handle_datalink = handle_raw;
930 handle_datalink = handle_null;
933 fprintf(stderr, "unsupported data link type %d\n",
934 pcap_datalink(pcap_obj));
941 if (readfile_state < 2) {
943 * On some OSes select() might return 0 even when
944 * there are packets to process. Thus, we always
945 * ignore its return value and just call pcap_dispatch()
948 if (0 == readfile_state) /* interactive */
949 pcap_select(pcap_obj, 1, 0);
950 x = pcap_dispatch(pcap_obj, 50, handle_pcap, NULL);
952 if (0 == x && 1 == readfile_state) {
953 /* block on keyboard until user quits */
962 endwin(); /* klin, Thu Nov 28 08:56:51 2002 */
964 while (pcap_dispatch(pcap_obj, 50, handle_pcap, NULL))
967 Sources_report(); print_func("\n");
968 Destinatioreport(); print_func("\n");
969 Qtypes_report(); print_func("\n");
970 Opcodes_report(); print_func("\n");
971 Tld_report(); print_func("\n");
972 Sld_report(); print_func("\n");
973 Nld_report(); print_func("\n");
974 SldBySource_report();
977 pcap_close(pcap_obj);
979 } /* static int main(int argc, char *argv[]) */
982 * vim:shiftwidth=4:tabstop=8:softtabstop=4