genreport.c 78.2 KB
Newer Older
Mark Andrews's avatar
Mark Andrews committed
1 2 3 4 5 6 7 8
/*
 * Copyright (C) 2016  Internet Systems Consortium, Inc. ("ISC")
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

9 10
#define FD_SETSIZE 1600

Mark Andrews's avatar
Mark Andrews committed
11
#include <stdio.h>
Mark Andrews's avatar
Mark Andrews committed
12 13
#include <string.h>
#include <stdlib.h>
14
#include <stddef.h>
Mark Andrews's avatar
Mark Andrews committed
15
#include <unistd.h>
Mark Andrews's avatar
Mark Andrews committed
16
#include <assert.h>
Mark Andrews's avatar
Mark Andrews committed
17

Mark Andrews's avatar
Mark Andrews committed
18 19 20 21 22 23 24
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/nameser.h>
#include <sys/select.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/time.h>
25 26
#include <sys/ioctl.h>
#include <sys/uio.h>
27 28 29 30 31 32
#include <netinet/ip.h>
#include <netinet/ip_icmp.h>
#include <netinet/ip6.h>
#include <netinet/icmp6.h>
#include <netinet/udp.h>
#include <netinet/tcp.h>
Mark Andrews's avatar
Mark Andrews committed
33 34

#include <errno.h>
35
#include <netdb.h>
Mark Andrews's avatar
Mark Andrews committed
36
#include <resolv.h>
37
#include <signal.h>
Mark Andrews's avatar
Mark Andrews committed
38

Mark Andrews's avatar
Mark Andrews committed
39 40 41 42 43 44
#define ns_t_dname 39
#define ns_t_sink 40
#define ns_t_apl 42
#define ns_t_ds 43
#define ns_t_sshfp 44
#define ns_t_ipseckey 45
45
#define ns_t_rrsig 46
Mark Andrews's avatar
Mark Andrews committed
46
#define ns_t_nsec 47
Mark Andrews's avatar
Mark Andrews committed
47
#define ns_t_dnskey 48
Mark Andrews's avatar
Mark Andrews committed
48 49 50 51
#define ns_t_dhcid 49
#define ns_t_nsec3 50
#define ns_t_nsec3param 51
#define ns_t_tlsa 52
52
#define ns_t_smimea 53
Mark Andrews's avatar
Mark Andrews committed
53 54 55 56 57
#define ns_t_hip 55
#define ns_t_ninfo 56
#define ns_t_talink 58
#define ns_t_cds 59
#define ns_t_cdnskey 60
58
#define ns_t_openpgpkey 61
Mark Andrews's avatar
Mark Andrews committed
59 60 61 62 63 64 65 66 67 68 69 70
#define ns_t_csync 62
#define ns_t_spf 99
#define ns_t_nid 104
#define ns_t_l32 105
#define ns_t_l34 106
#define ns_t_lp 107
#define ns_t_eui48 108
#define ns_t_eui64 109
#define ns_t_uri 256
#define ns_t_caa 257
#define ns_t_ta 32768
#define ns_t_dlv 32769
Mark Andrews's avatar
Mark Andrews committed
71

72 73
#define ns_r_badcookie 23

74
static int eof = 0;
Mark Andrews's avatar
Mark Andrews committed
75 76
static int maxfd = -1;
static fd_set rfds, wfds;
Mark Andrews's avatar
Mark Andrews committed
77
static int outstanding = 0;
78
static int maxoutstanding = 100;
Mark Andrews's avatar
Mark Andrews committed
79 80 81 82 83 84

static void(*rhandlers[FD_SETSIZE])(int);
static void(*whandlers[FD_SETSIZE])(int);

static int udp4 = -1;
static int udp6 = -1;
85 86
static int icmp4 = -1;
static int icmp6 = -1;
87

Mark Andrews's avatar
Mark Andrews committed
88 89 90
static int ipv4only = 0;
static int ipv6only = 0;

Mark Andrews's avatar
Mark Andrews committed
91
static int allok  = 0;
92
static int bad  = 0;
Mark Andrews's avatar
Mark Andrews committed
93
static int badtag  = 0;
94
static int ednsonly  = 0;
95
static int debug = 0;
Mark Andrews's avatar
Mark Andrews committed
96
static int inorder = 0;
97 98
static int serial = 0;
static long long sent;
99 100 101

static union res_sockaddr_union servers[10];
static int nservers = 0;
102
int ident = 0;
Mark Andrews's avatar
Mark Andrews committed
103

Mark Andrews's avatar
Mark Andrews committed
104 105 106
/*
 * Doubly linked list macros.
 */
Mark Andrews's avatar
Mark Andrews committed
107 108 109 110 111 112 113 114
#define APPEND(list, item, link) do { \
	if ((list).tail) \
		(list).tail->link.next = (item); \
	else \
		(list).head = (item); \
	(item)->link.prev = list.tail; \
	(item)->link.next = NULL; \
	(list).tail = (item); \
Mark Andrews's avatar
Mark Andrews committed
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
	(item)->link.linked = 1; \
} while (0)

#define PREPEND(list, item, link) do { \
	if ((list).head) \
		(list).head->link.prev = (item); \
	else \
		(list).tail = (item); \
	(item)->link.prev = NULL; \
	(item)->link.next = list.head; \
	(list).head = (item); \
	(item)->link.linked = 1; \
} while (0)

#define INSERTBEFORE(list, before, item, link) do { \
	assert(LINKED(before, link)); \
	if ((before)->link.prev == NULL) \
		PREPEND(list, item, link); \
	else { \
		(item)->link.prev = (before)->link.prev; \
		(before)->link.prev = (item); \
		(item)->link.prev->link.next = (item); \
		(item)->link.next = (before); \
Mark Andrews's avatar
Mark Andrews committed
138
		(item)->link.linked = 1; \
Mark Andrews's avatar
Mark Andrews committed
139
	} \
Mark Andrews's avatar
Mark Andrews committed
140 141 142 143 144 145 146 147 148 149 150 151
} while (0)

#define UNLINK(list, item, link) do { \
	if ((item)->link.next) \
		(item)->link.next->link.prev = (item)->link.prev; \
	else \
		list.tail = (item)->link.prev; \
	if ((item)->link.prev) \
		(item)->link.prev->link.next = (item)->link.next; \
	else \
		list.head = (item)->link.next; \
	(item)->link.next = (item)->link.prev = NULL; \
152
	(item)->link.linked = 0; \
Mark Andrews's avatar
Mark Andrews committed
153 154 155
} while (0)

#define NEXT(item, link) (item)->link.next
Mark Andrews's avatar
Mark Andrews committed
156
#define PREV(item, link) (item)->link.prev
157
#define LINKED(item, link) (item)->link.linked
Mark Andrews's avatar
Mark Andrews committed
158 159 160 161

#define HEAD(list) (list).head
#define TAIL(list) (list).tail

Mark Andrews's avatar
Mark Andrews committed
162 163 164
/*
 * Test groupings
 */
Mark Andrews's avatar
Mark Andrews committed
165 166 167 168 169 170 171
#define EDNS 0x01
#define COMM 0x02
#define FULL 0x04
#define TYPE 0x08
#define EXPL 0x10

static int what = EDNS;
172

Mark Andrews's avatar
Mark Andrews committed
173
static struct {
Mark Andrews's avatar
Mark Andrews committed
174 175 176 177 178 179 180 181
	const char *name;		/* test nmemonic */
	unsigned int what;		/* select what test to make / report */
	unsigned short rdlen;		/* edns rdata length */
	const char *rdata;		/* edns rdata */
	unsigned short udpsize;		/* edns UDP size (0 == no EDNS) */
	unsigned short flags;		/* edns flags to be set */
	unsigned short version;		/* edns version */
	unsigned int tcp;		/* use tcp */
182
	unsigned int cookie;		/* opt record has cookie */
Mark Andrews's avatar
Mark Andrews committed
183 184 185 186 187 188 189 190 191 192
	unsigned int ignore;		/* ignore tc in response */
	unsigned int tc;		/* set tc in request */
	unsigned int rd;		/* set rd in request */
	unsigned int ra;		/* set ra in request */
	unsigned int cd;		/* set cd in request */
	unsigned int ad;		/* set ad  in request */
	unsigned int aa;		/* set aa in request */
	unsigned int z;			/* set z in request */
	unsigned int opcode;		/* use opcode for request */
	unsigned short type;		/* query type code */
193
} opts[] = {
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211
	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "dns",       EDNS,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "aa",        FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0,  0, ns_t_soa },
	{ "ad",        FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0,  0, ns_t_soa },
	{ "cd",        FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0,  0, ns_t_soa },
	{ "ra",        FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0,  0, ns_t_soa },
	{ "rd",        FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "tc",        FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "zflag",     FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,  0, ns_t_soa },
	{ "opcode",    FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 15, 0 },
	{ "opcodeflg", FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 15, 0 },
	{ "type666",   FULL,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, 666 },
	{ "tcp",       FULL,  0, "",    0, 0x0000, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "edns",      EDNS,  0, "", 4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "edns1",     EDNS,  0, "", 4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "edns@512",  EDNS,  0, "",  512, 0x0000, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_dnskey },
Mark Andrews's avatar
Mark Andrews committed
212
	{ "ednsopt",   EDNS,  4, "\x00\x64\x00\x00",	/* 100 */
213
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
214
	{ "edns1opt",  EDNS,  4, "\x00\x64\x00\x00",	/* 100 */
215 216 217 218 219
				     4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "do",        EDNS,  0, "",
				     4096, 0x8000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "edns1do",   FULL,  0, "", 4096, 0x8000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "ednsflags", EDNS,  0, "", 4096, 0x0080, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
220
	{ "optlist",   EDNS,  4 + 8 + 4 + 12,
221 222 223 224
	  "\x00\x03\x00\x00" 		     /* NSID */
	  "\x00\x08\x00\x04\x00\x01\x00\x00" /* ECS */
	  "\x00\x09\x00\x00" 		     /* EXPIRE */
	  "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08",	/* COOKIE */
225 226 227
				     4096, 0x0000, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
Mark Andrews's avatar
Mark Andrews committed
228
	{ "ednsnsid", FULL,  4, "\x00\x03\x00\x00",	/* NSID */
229
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
230
	{ "ednscookie", FULL, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
231
				     4096, 0x0000, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
232
	{ "ednsexpire", FULL, 4, "\x00\x09\x00\x00",	/* EXPIRE */
233
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
234
	{ "ednssubnet", FULL,  8, "\x00\x08\x00\x04\x00\x01\x00\x00",	/* ECS */
235
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
236
	{ "edns1nsid", FULL,  4, "\x00\x03\x00\x00",	/* NSID */
237
				     4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
238
	{ "edns1cookie", FULL, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
239
				     4096, 0x0000, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
240
	{ "edns1expire", FULL, 4, "\x00\x09\x00\x00",	/* EXPIRE */
241
				     4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
242
	{ "edns1subnet", FULL,  8, "\x00\x08\x00\x04\x00\x01\x00\x00",	/* ECS */
243 244 245 246
				     4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "ednstcp",   EDNS,  0, "",  512, 0x8000, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_dnskey },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
Mark Andrews's avatar
Mark Andrews committed
247
	{ "bind11",    COMM, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
248
				     4096, 0x8000, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
249
	{ "dig11",     COMM, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323
				     4096, 0x0000, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0,  0, ns_t_soa },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "A",         TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_a },
	{ "NS",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_ns },
	{ "MD",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_md },
	{ "MF",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_mf },
	{ "CNAME",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_cname },
	{ "SOA",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
	{ "MB",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_mb },
	{ "MG",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_mg },
	{ "MR",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_mr },
	{ "NULL",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_null },
	{ "WKS",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_wks },
	{ "PTR",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_ptr },
	{ "HINFO",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_hinfo },
	{ "MINFO",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_minfo },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "MX",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_mx },
	{ "TXT",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_txt },
	{ "RP",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_rp },
	{ "AFSDB",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_afsdb },
	{ "X25",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_x25 },
	{ "ISDN",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_isdn },
	{ "RT",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_rt },
	{ "NSAP",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nsap },
	{ "NSAP-PTR",  TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nsap_ptr },
	{ "SIG",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_sig },
	{ "KEY",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_key },
	{ "PX",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_px },
	{ "GPOS",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_gpos },
	{ "AAAA",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_aaaa },
	{ "LOC",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_loc },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "NXT",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nxt },
	{ "SRV",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_srv },
	{ "NAPTR",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_naptr },
	{ "KX",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_kx },
	{ "CERT",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_cert },
	{ "A6",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_a6 },
	{ "DNAME",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_dname },
	{ "APL",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_apl },
	{ "DS",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_ds },
	{ "SSHFP",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_sshfp },
	{ "IPSECKEY",  TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_ipseckey },
	{ "RRSIG",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_rrsig },
	{ "NSEC",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nsec },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "DNSKEY",    TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_dnskey },
	{ "DHCID",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_dhcid },
	{ "NSEC3",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nsec3 },
	{ "NSEC3PARAM",TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nsec3param },
	{ "TLSA",      TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_tlsa },
	{ "SMIMEA",    TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_smimea },
	{ "HIP",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_hip },
	{ "CDS",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_cds },
	{ "CDNSKEY",   TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_cdnskey },
	{ "OPENPGPKEY",TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_openpgpkey },
	{ "SPF",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_spf },
	{ "NID",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_nid },
	{ "L32",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_l32 },

	/*                           size   eflgs vr  T ck ig tc rd ra cd ad aa  z  op  type */
	{ "L64",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_l34 },
	{ "LP",        TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_lp },
	{ "EUI48",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_eui48 },
	{ "EUI64",     TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_eui64 },
	{ "URI",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_uri },
	{ "CAA",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_caa },
	{ "DLV",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_dlv },
	{ "TYPE1000",  TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, 1000 }
324 325
};

Mark Andrews's avatar
Mark Andrews committed
326 327 328
/*
 * Summary structure where results from multiple lookups are recorded.
 */
329
struct summary {
Mark Andrews's avatar
Mark Andrews committed
330 331 332 333 334
	struct {
		struct summary *prev;
		struct summary *next;
		int linked;
	} link;
Mark Andrews's avatar
Mark Andrews committed
335 336 337
	char zone[1024];		/* the zone's name */
	char ns[1024];			/* the server's name */
	struct sockaddr_storage storage;/* server we are talking to */
Mark Andrews's avatar
Mark Andrews committed
338
	int tests;			/* number of outstanding tests */
Mark Andrews's avatar
Mark Andrews committed
339
	int last;			/* last test sent */
Mark Andrews's avatar
Mark Andrews committed
340
	int deferred;			/* was the printing deferred */
Mark Andrews's avatar
Mark Andrews committed
341
	int done;			/* we are done */
Mark Andrews's avatar
Mark Andrews committed
342 343 344 345 346 347
	int type;			/* recursive query lookup type */
	int nodataa;			/* recursive query got nodata */
	int nodataaaaa;			/* recursive query got nodata */
	int nxdomain;			/* recursive query got nxdomain */
	int nxdomaina;			/* recursive query got nxdomain */
	int nxdomainaaaa;		/* recursive query got nxdomain */
348 349
	int faileda;
	int failedaaaa;
350 351
	int cnamea;			/* Nameserver is CNAME */
	int cnameaaaa;			/* Nameserver is CNAME */
Mark Andrews's avatar
Mark Andrews committed
352
	int seenrrsig;			/* a rrsig was seen in "do" test */
353 354 355 356 357 358
	int seenopt;			/* see a EDNS response */
	int seenedns;			/* see a EDNS response */
	int seenfailure;		/* see a lookup failure */
	int allok;			/* all answers are current ok */
	int allrefused;			/* all answers are current ok */
	int allservfail;		/* all answers are current ok */
Mark Andrews's avatar
Mark Andrews committed
359
	struct summary *xlink;		/* cross link of recursive A/AAAA */
360 361 362
	char results[sizeof(opts)/sizeof(opts[0])][100];
};

Mark Andrews's avatar
Mark Andrews committed
363 364 365 366 367
static struct {
	struct summary *head;
	struct summary *tail;
} summaries;

Mark Andrews's avatar
Mark Andrews committed
368 369 370 371
struct workitem {
	struct {
		struct workitem *next;
		struct workitem *prev;
372
		int linked;
373
	} link, clink, plink, rlink, idlink, seqlink;
Mark Andrews's avatar
Mark Andrews committed
374 375 376
	unsigned short id;		/* the query id we are waiting for */
	struct timeval when;		/* when we will timeout */
	int type;			/* the query type being looked up */
Mark Andrews's avatar
Mark Andrews committed
377 378 379
	int test;			/* test number / server number */
	int sends;			/* number of times this UDP request
					 * has been sent */
Mark Andrews's avatar
Mark Andrews committed
380 381
	int buflen;			/* the size of the request to be sent */
	int tcpfd;			/* the tcp file descriptor */
Mark Andrews's avatar
Mark Andrews committed
382 383
	int outstanding;		/* outstanding has been set */
	int havelen;			/* readlen is tcp message length */
Mark Andrews's avatar
Mark Andrews committed
384 385 386 387 388
	int readlen;			/* how much we need to read */
	int read;			/* how much has been read so far */
	unsigned char buf[512];		/* the question we sent */
	unsigned char tcpbuf[0x10000];	/* where to accumulate the tcp response */
	struct summary *summary;	/* where this test is summaried */
Mark Andrews's avatar
Mark Andrews committed
389 390
};

391
/*
Mark Andrews's avatar
Mark Andrews committed
392
 * Work queues:
393 394 395
 *	'work' udp qeries;
 *	'connecting' tcp qeries;
 *	'reading' tcp qeries;
Mark Andrews's avatar
Mark Andrews committed
396
 *	'pending' deferred work items;
Mark Andrews's avatar
Mark Andrews committed
397 398 399
 *
 * Outstanding queries by qid.
 *	'ids'
400 401 402
 *
 * Outstanding icmp by seq.
 *	'ids'
403
 */
Mark Andrews's avatar
Mark Andrews committed
404 405 406
static struct {
	struct workitem *head;
	struct workitem *tail;
407
} work, connecting, reading, pending, ids[0x10000], seq[0x10000];
Mark Andrews's avatar
Mark Andrews committed
408

Mark Andrews's avatar
Mark Andrews committed
409 410 411
static void
dotest(struct workitem *item);

412 413 414
static void
nextserver(struct workitem *item);

415 416
static void
connecttoserver(struct workitem *item);
Mark Andrews's avatar
Mark Andrews committed
417

Mark Andrews's avatar
Mark Andrews committed
418 419 420
static void
report(struct summary *summary);

Mark Andrews's avatar
Mark Andrews committed
421 422 423 424 425 426 427 428 429 430 431 432
static int
storage_equal(struct sockaddr_storage *s1, struct sockaddr_storage *s2) {
	struct sockaddr_in *sin1, *sin2;
	struct sockaddr_in6 *sin61, *sin62;

	if (s1->ss_len != s2->ss_len || s1->ss_family != s2->ss_family)
		return (0);

	switch (s1->ss_family) {
	case AF_INET:
		sin1 = (struct sockaddr_in *)s1;
		sin2 = (struct sockaddr_in *)s2;
Mark Andrews's avatar
Mark Andrews committed
433

Mark Andrews's avatar
Mark Andrews committed
434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449
		if (sin1->sin_port != sin2->sin_port ||
		    sin1->sin_addr.s_addr != sin2->sin_addr.s_addr)
			return (0);
		return (1);
	case AF_INET6:
		sin61 = (struct sockaddr_in6 *)s1;
		sin62 = (struct sockaddr_in6 *)s2;

		if (sin61->sin6_port != sin62->sin6_port ||
		    memcmp(&sin61->sin6_addr, &sin62->sin6_addr, 16) != 0)
			return (0);
		return (1);
	}
	return (0);
}

Mark Andrews's avatar
Mark Andrews committed
450 451 452
/*
 * Check if it is safe to use this id to this address.
 */
Mark Andrews's avatar
Mark Andrews committed
453 454 455 456 457 458
static int
checkid(struct sockaddr_storage *storage, int id) {
	struct workitem *item;

	item = HEAD(ids[id]);
	while (item != NULL &&
459
	       !storage_equal(storage, &item->summary->storage))
Mark Andrews's avatar
Mark Andrews committed
460 461 462 463
		item = NEXT(item, idlink);
	return ((item == NULL) ? 1 : 0);
}

464 465 466 467 468 469 470 471 472 473 474 475 476 477 478
/*
 * Check if we have a outstanding icmp with this sequence number
 * to this address.
 */
static int
checkseq(struct sockaddr_storage *storage, int id) {
	struct workitem *item;

	item = HEAD(seq[id]);
	while (item != NULL &&
	       !storage_equal(storage, &item->summary->storage))
		item = NEXT(item, seqlink);
	return ((item == NULL) ? 1 : 0);
}

Mark Andrews's avatar
Mark Andrews committed
479 480 481 482 483 484 485
static void
freesummary(struct summary *summary) {
	if (LINKED(summary, link))
		UNLINK(summaries, summary, link);
	free(summary);
}

Mark Andrews's avatar
Mark Andrews committed
486 487 488
/*
 * Generate a report line.
 */
489
static void
Mark Andrews's avatar
Mark Andrews committed
490
printandfree(struct summary *summary) {
491 492 493 494 495 496 497 498
	struct sockaddr_in *s = (struct sockaddr_in *)&summary->storage;
	struct sockaddr_in6 *s6 = (struct sockaddr_in6 *)&summary->storage;;
	char addrbuf[64];
	void *addr;
	unsigned int i;
	int x;

	if ((summary->type == ns_t_a || summary->type == ns_t_aaaa) &&
Mark Andrews's avatar
Mark Andrews committed
499
	    summary->nodataa && summary->nodataaaaa) {
500 501
		printf("%s. %s: no address records found\n",
		       summary->zone, summary->ns);
Mark Andrews's avatar
Mark Andrews committed
502
		freesummary(summary);
503 504
		return;
	}
505

506 507 508 509
	if ((summary->type == ns_t_a || summary->type == ns_t_aaaa) &&
	    summary->nxdomaina && summary->nxdomainaaaa) {
		printf("%s. %s: no address records found (NXDOMAIN)\n",
		       summary->zone, summary->ns);
Mark Andrews's avatar
Mark Andrews committed
510
		freesummary(summary);
511 512
		return;
	}
513

514 515 516 517 518 519 520 521
	if ((summary->type == ns_t_a || summary->type == ns_t_aaaa) &&
	    summary->faileda && summary->failedaaaa) {
		printf("%s. %s: address lookups failed\n",
		       summary->zone, summary->ns);
		freesummary(summary);
		return;
	}

522 523 524 525 526 527 528 529
	if ((summary->type == ns_t_a || summary->type == ns_t_aaaa) &&
	    (summary->cnamea || summary->cnameaaaa)) {
		printf("%s. %s: nameserver is a CNAME\n",
		       summary->zone, summary->ns);
		freesummary(summary);
		return;
	}

530 531 532
	/*
	 * Do deferred xlink failure reports.
	 */
533 534
	if (summary->type == ns_t_a &&
	    summary->nodataa && summary->failedaaaa) {
535 536 537 538 539
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" AAAAA lookup failed\n");
		freesummary(summary);
		return;
	}
540 541
	if (summary->type == ns_t_aaaa &&
	    summary->nodataaaaa && summary->faileda) {
542 543 544 545 546
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" A lookup failed\n");
		freesummary(summary);
		return;
	}
547 548
	if (summary->type == ns_t_a &&
	    summary->faileda && summary->nxdomainaaaa) {
549 550 551 552 553
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" AAAAA nxdomain\n");
		freesummary(summary);
		return;
	}
554 555
	if (summary->type == ns_t_aaaa &&
	    summary->failedaaaa && summary->nxdomaina) {
556 557 558 559 560
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" A nxdomain\n");
		freesummary(summary);
		return;
	}
561

Mark Andrews's avatar
Mark Andrews committed
562
	if (summary->done || summary->nodataa || summary->nodataaaaa) {
Mark Andrews's avatar
Mark Andrews committed
563
		freesummary(summary);
564 565
		return;
	}
Mark Andrews's avatar
Mark Andrews committed
566

567 568 569
	if (summary->type != 0 && summary->nxdomain) {
		if (summary->type == ns_t_ns) {
			printf("%s.: NS nxdomain\n", summary->zone);
Mark Andrews's avatar
Mark Andrews committed
570
			freesummary(summary);
571 572 573 574 575 576
			return;
		}
		printf("%s. %s:", summary->zone, summary->ns);
		if (summary->type == ns_t_a) printf(" A");
		if (summary->type == ns_t_aaaa) printf(" AAAA");
		printf(" nxdomain\n");
Mark Andrews's avatar
Mark Andrews committed
577
		freesummary(summary);
578 579 580 581 582
		return;
	}
	if (summary->type == ns_t_a) {
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" A lookup failed\n");
Mark Andrews's avatar
Mark Andrews committed
583
		freesummary(summary);
584 585 586 587 588
		return;
	}
	if (summary->type == ns_t_aaaa) {
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" AAAA lookup failed\n");
Mark Andrews's avatar
Mark Andrews committed
589
		freesummary(summary);
590 591 592
		return;
	}
	if (summary->type == ns_t_ns) {
Mark Andrews's avatar
Mark Andrews committed
593
		printf("%s.:", summary->zone);
594
		printf(" NS lookup failed\n");
Mark Andrews's avatar
Mark Andrews committed
595
		freesummary(summary);
596 597
		return;
	}
598

599 600 601 602 603
	if (summary->seenopt && (summary->allrefused && summary->allservfail))
		summary->seenedns = 1;

	if (summary->type != 0 || (summary->allok && bad) ||
	    (!summary->seenfailure && !summary->seenedns && ednsonly)) {
Mark Andrews's avatar
Mark Andrews committed
604
		freesummary(summary);
Mark Andrews's avatar
Mark Andrews committed
605 606 607
		return;
	}

608 609 610 611 612
	switch (summary->storage.ss_family) {
	case AF_INET: addr = &s->sin_addr; break;
	case AF_INET6: addr = &s6->sin6_addr; break;
	default: addr = NULL; break;
	}
Mark Andrews's avatar
Mark Andrews committed
613

614 615 616 617 618 619
	if (addr == NULL)
		strlcpy(addrbuf, "<unknown>", sizeof(addrbuf));
	else
		inet_ntop(summary->storage.ss_family, addr,
			  addrbuf, sizeof(addrbuf));

620 621
	x = -1;
	printf("%s. @%s (%s.):", summary->zone, addrbuf, summary->ns);
Mark Andrews's avatar
Mark Andrews committed
622 623 624 625 626
	if (allok && summary->allok)
		printf(" all ok");
	else
		for (i = 0; i < sizeof(opts)/sizeof(opts[0]); i++) {
			if (opts[i].what != 0 && (opts[i].what & what) == 0)
Mark Andrews's avatar
Mark Andrews committed
627
				continue;
Mark Andrews's avatar
Mark Andrews committed
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645
			if (summary->results[i][0] == 0)
				strlcpy(summary->results[i], "skipped", 100);
			if (strcmp(opts[i].name, "do") == 0)
				x = i;
			if (strcmp(opts[i].name, "ednstcp") == 0 && x != -1 &&
			    (!badtag || (strcmp(summary->results[x], "ok") != 0 &&
					 strncmp(summary->results[x], "ok,", 3) != 0)))
			{
				printf(" signed=%s", summary->results[x]);
				if (summary->seenrrsig)
					printf(",yes");
			}
			if (badtag) {
				if (strcmp(summary->results[i], "ok") == 0 ||
				    strncmp(summary->results[i], "ok,", 3) == 0)
					continue;
			}
			printf(" %s=%s", opts[i].name, summary->results[i]);
Mark Andrews's avatar
Mark Andrews committed
646
		}
647
	printf("\n");
Mark Andrews's avatar
Mark Andrews committed
648 649 650 651 652 653
	freesummary(summary);
}

static void
report(struct summary *summary) {

Mark Andrews's avatar
Mark Andrews committed
654 655 656 657
	/*
	 * Send the next test now that we have completed the last test.
	 */
	if (serial && summary->type == 0 && summary->tests == 1) {
658 659 660
		for (summary->last++;
		     summary->last < sizeof(opts)/sizeof(opts[0]);
		     summary->last++) {
Mark Andrews's avatar
Mark Andrews committed
661
			struct workitem *item;
662 663
			if (opts[summary->last].what != 0 &&
			    (opts[summary->last].what & what) == 0)
Mark Andrews's avatar
Mark Andrews committed
664 665 666 667 668
				continue;
			item = calloc(1, sizeof(*item));
			if (item == NULL)
				continue;
			item->summary = summary;
669
			item->test = item->summary->last;
670
			item->tcpfd = -1;
Mark Andrews's avatar
Mark Andrews committed
671 672 673 674 675
			dotest(item);
			return;
		}
	}

Mark Andrews's avatar
Mark Andrews committed
676 677 678 679 680 681 682 683 684 685 686 687
	/*
	 * Have all the tests completed?
	 */
	summary->tests--;
	if (summary->tests)
		return;

	/*
	 * If we are cross linked record the lookup details on the other
	 * structure.
	 */
	if (summary->xlink) {
688 689 690 691 692 693 694 695
		if (summary->cnamea) {
			summary->xlink->cnamea = 1;
			summary->done = 1;
		}
		if (summary->cnameaaaa) {
			summary->xlink->cnameaaaa = 1;
			summary->done = 1;
		}
Mark Andrews's avatar
Mark Andrews committed
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711
		if (summary->nodataa) {
			summary->xlink->nodataa = 1;
			summary->done = 1;
		}
		if (summary->nodataaaaa) {
			summary->xlink->nodataaaaa = 1;
			summary->done = 1;
		}
		if (summary->nxdomaina) {
			summary->xlink->nxdomaina = 1;
			summary->done = 1;
		}
		if (summary->nxdomainaaaa) {
			summary->xlink->nxdomainaaaa = 1;
			summary->done = 1;
		}
712 713 714 715 716 717 718 719 720 721
		if (!summary->done) {
			if (summary->type == ns_t_a) {
				summary->xlink->faileda = 1;
				summary->done = 1;
			}
			if (summary->type == ns_t_aaaa) {
				summary->xlink->failedaaaa = 1;
				summary->done = 1;
			}
		}
Mark Andrews's avatar
Mark Andrews committed
722 723 724 725 726 727 728 729 730 731 732

		/*
		 * Remove the cross link.
		 */
		summary->xlink->xlink = NULL;
		summary->xlink = NULL;
		if (summary->done) {
			freesummary(summary);
			goto print_deferred;
		}
	}
733 734 735 736 737 738
	if (!summary->done) {
		if (summary->type == ns_t_a)
			summary->faileda = 1;
		if (summary->type == ns_t_aaaa)
			summary->failedaaaa = 1;
	}
Mark Andrews's avatar
Mark Andrews committed
739 740 741 742 743 744 745 746 747 748

	if (inorder && PREV(summary, link)) {
		summary->deferred = 1;
		return;
	}
	printandfree(summary);

 print_deferred:
	while ((summary = HEAD(summaries)) && summary->deferred)
		printandfree(summary);
749 750
}

Mark Andrews's avatar
Mark Andrews committed
751 752 753
/*
 * Free a work item and unlink.
 */
754 755
static void
freeitem(struct workitem * item) {
Mark Andrews's avatar
Mark Andrews committed
756 757
	if (item->summary)
		report(item->summary);
758
	if (item->tcpfd != -1) {
759 760 761 762 763 764 765 766 767 768
		FD_CLR(item->tcpfd, &rfds);
		FD_CLR(item->tcpfd, &wfds);
		rhandlers[item->tcpfd] = NULL;
		whandlers[item->tcpfd] = NULL;
		close(item->tcpfd);
	}
	if (item->outstanding)
		outstanding--;
	if (LINKED(item, link))
		UNLINK(work, item, link);
Mark Andrews's avatar
Mark Andrews committed
769 770
	if (LINKED(item, plink))
		UNLINK(pending, item, plink);
771 772 773 774 775 776
	if (LINKED(item, rlink))
		UNLINK(reading, item, rlink);
	if (LINKED(item, clink))
		UNLINK(connecting, item, clink);
	if (LINKED(item, idlink))
		UNLINK(ids[item->id], item, idlink);
777 778
	if (LINKED(item, seqlink))
		UNLINK(seq[item->id], item, seqlink);
779 780 781
	free(item);
}

Mark Andrews's avatar
Mark Andrews committed
782 783 784
/*
 * Add a tag to the report.
 */
785
static void
786
addtag(struct workitem *item, const char *tag) {
787
	char *result = item->summary->results[item->test];
788 789
	if (result[0]) strlcat(result, ",", 100);
	strlcat(result, tag, 100);
790 791
}

Mark Andrews's avatar
Mark Andrews committed
792 793 794
/*
 * Resend a UDP request.
 */
795 796 797 798 799 800 801 802 803 804 805 806 807 808 809
static void
resend(struct workitem *item) {
	int n, fd = -1;

	switch (item->summary->storage.ss_family) {
	case AF_INET:
		fd = udp4;
		break;
	case AF_INET6:
		fd = udp6;
		break;
	}

	if (fd == -1) {
		addtag(item, "skipped");
810
		item->summary->allok = 0;
811 812 813 814
		freeitem(item);
		return;
	}

815
	if (!item->outstanding && outstanding > maxoutstanding) {
816 817
		gettimeofday(&item->when, NULL);
		item->when.tv_sec += 1;
Mark Andrews's avatar
Mark Andrews committed
818 819 820
		if (LINKED(item, link))
			UNLINK(work, item, link);
		APPEND(pending, item, plink);
821 822 823 824 825 826 827 828 829 830 831 832 833 834
		return;
	}

	n = sendto(fd, item->buf, item->buflen, 0,
		   (struct sockaddr *)&item->summary->storage,
		   item->summary->storage.ss_len);
	if (n > 0) {
		if (debug)
			printf("resend %s rdlen=%u udpsize=%u flags=%04x "
			       "version=%u tcp=%u ignore=%u id=%u\n",
			       opts[item->test].name, opts[item->test].rdlen,
			       opts[item->test].udpsize, opts[item->test].flags,
			       opts[item->test].version, opts[item->test].tcp,
			       opts[item->test].ignore, item->id);
835
		sent++;
836 837 838 839 840
		if (!item->outstanding++)
			outstanding++;
		gettimeofday(&item->when, NULL);
		item->when.tv_sec += 1;
		item->sends++;
Mark Andrews's avatar
Mark Andrews committed
841 842
		if (LINKED(item, link))
			UNLINK(work, item, link);
843
		APPEND(work, item, link);
844 845
	} else if (item->summary->type) {
		nextserver(item);
846 847
	} else {
		addtag(item, "failed");
848 849
		item->summary->allok = 0;
		item->summary->seenfailure = 1;
850 851 852 853
		freeitem(item);
	}
}

Mark Andrews's avatar
Mark Andrews committed
854 855 856
/*
 * Start a individual test.
 */
Mark Andrews's avatar
Mark Andrews committed
857 858 859 860
static void
dotest(struct workitem *item) {
	unsigned char *cp;
	unsigned int ttl;
Mark Andrews's avatar
Mark Andrews committed
861
	int n, fd, id, tries = 0, opcode;
Mark Andrews's avatar
Mark Andrews committed
862

863
	switch (item->summary->storage.ss_family) {
Mark Andrews's avatar
Mark Andrews committed
864 865 866 867 868 869 870 871 872
	case AF_INET:
		fd = udp4;
		break;
	case AF_INET6:
		fd = udp6;
		break;
	}

	if (fd == -1) {
873
		addtag(item, "skipped");
874
		item->summary->allok = 0;
875
		freeitem(item);
Mark Andrews's avatar
Mark Andrews committed
876 877
		return;
	}
Mark Andrews's avatar
Mark Andrews committed
878

Mark Andrews's avatar
Mark Andrews committed
879 880 881 882
	/*
	 * res_mkquery only really knows about QUERY but it is useful
	 * for initalising the header when the opcode isn't QUERY.
	 */
Mark Andrews's avatar
Mark Andrews committed
883 884 885 886 887 888 889 890
	opcode = opts[item->test].opcode;
	switch (opcode) {
	case 0: break;
	default:
		opcode = ns_o_query;
	}

	n = res_mkquery(opcode, item->summary->zone, ns_c_in,
891 892
			opts[item->test].type, NULL, 0, NULL,
			item->buf, sizeof(item->buf));
Mark Andrews's avatar
Mark Andrews committed
893 894 895 896 897 898 899
	/* fixup opcode? */
	if (n > 0 && opts[item->test].opcode != opcode) {
		item->buf[2] &= 0x17;
		item->buf[2] |= (opts[item->test].opcode & 0x0f) << 3;
		/* Zero question section. */
		if (opts[item->test].opcode == 15)
			item->buf[4] = item->buf[5] = 0;
900
		n = 12;
Mark Andrews's avatar
Mark Andrews committed
901
	}
Mark Andrews's avatar
Mark Andrews committed
902 903 904 905

	/*
	 * Set DNS flags as specified by test.
	 */
906 907
	if (opts[item->test].tc)
		item->buf[2] |= 0x2;	/* set tc */
Mark Andrews's avatar
Mark Andrews committed
908 909 910 911
	if (opts[item->test].rd)
		item->buf[2] |= 0x1;	/* set rd */
	else
		item->buf[2] &= ~0x1;	/* clear rd */
912 913 914 915 916 917 918 919
	if (opts[item->test].ra)
		item->buf[3] |= 0x80;	/* set ra */
	if (opts[item->test].z)
		item->buf[3] |= 0x40;	/* set z */
	if (opts[item->test].ad)
		item->buf[3] |= 0x20;	/* set ad */
	if (opts[item->test].cd)
		item->buf[3] |= 0x10;	/* set cd */
Mark Andrews's avatar
Mark Andrews committed
920

921
	if (n > 12) {
922
		char name[1024];
Mark Andrews's avatar
Mark Andrews committed
923 924 925
		/*
		 * Make zone canonical.
		 */
926 927
		dn_expand(item->buf, item->buf + n, item->buf + 12,
			  name, sizeof(name));
928 929
		strlcpy(item->summary->zone, name,
			sizeof(item->summary->zone));
930
	}
Mark Andrews's avatar
Mark Andrews committed
931

Mark Andrews's avatar
Mark Andrews committed
932 933 934 935 936
	/*
	 * Add OPT record if required by test.
	 */
	if (n > 0 && opts[item->test].udpsize > 0 &&
	    11 + opts[item->test].rdlen < 512 - n) {
Mark Andrews's avatar
Mark Andrews committed
937
		cp = item->buf + n;
938 939 940 941 942 943 944 945 946 947 948
		*cp++ = 0;				/* name */
		ns_put16(ns_t_opt, cp);			/* type */
		cp += 2;
		ns_put16(opts[item->test].udpsize, cp);	/* class */
		cp += 2;
		ttl = (opts[item->test].version << 16) |
		      opts[item->test].flags;
		ns_put32(ttl, cp);			/* ttl */
		cp += 4;
		ns_put16(opts[item->test].rdlen, cp);	/* rdlen */
		cp += 2;
Mark Andrews's avatar
Mark Andrews committed
949 950
		memcpy(cp, opts[item->test].rdata, opts[item->test].rdlen);
		cp += opts[item->test].rdlen;
951 952
		item->buf[11] = 1;			/* adcount */
		n = cp - item->buf;			/* total length */
Mark Andrews's avatar
Mark Andrews committed
953 954 955
	}

	if (n > 0) {
Mark Andrews's avatar
Mark Andrews committed
956 957 958
		/*
		 * Adjust id if it clashes with a outstanding request.
		 */
Mark Andrews's avatar
Mark Andrews committed
959 960
		id = item->buf[0] << 8 | item->buf[1];

961 962
		while (!checkid(&item->summary->storage, id) &&
		       tries++ < 0xffff)
Mark Andrews's avatar
Mark Andrews committed
963
			id = (id + 1) & 0xffff;
Mark Andrews's avatar
Mark Andrews committed
964

Mark Andrews's avatar
Mark Andrews committed
965
		if (tries == 0xffff) {
966
			addtag(item, "skipped");
967
			item->summary->allok = 0;
968
			item->summary->seenfailure = 1;
969
			freeitem(item);
Mark Andrews's avatar
Mark Andrews committed
970 971
			return;
		}
Mark Andrews's avatar
Mark Andrews committed
972

Mark Andrews's avatar
Mark Andrews committed
973 974 975 976 977
		item->buf[0] = id >> 8;
		item->buf[1] = id & 0xff;
		item->id = id;
		item->buflen = n;

978 979 980 981 982
		if (opts[item->test].tcp) {
			connecttoserver(item);
			return;
		}

Mark Andrews's avatar
Mark Andrews committed
983 984 985
		/*
		 * If there is too much outstanding work queue this item.
		 */
986
		if (!item->outstanding && outstanding > maxoutstanding) {
987 988
			gettimeofday(&item->when, NULL);
			item->when.tv_sec += 1;
Mark Andrews's avatar
Mark Andrews committed
989
			APPEND(pending, item, plink);
990 991 992 993
			APPEND(ids[item->id], item, idlink);
			return;
		}

Mark Andrews's avatar
Mark Andrews committed
994
		n = sendto(fd, item->buf, item->buflen, 0,
995 996
			   (struct sockaddr *)&item->summary->storage,
			   item->summary->storage.ss_len);
Mark Andrews's avatar
Mark Andrews committed
997 998 999
	}

	if (n > 0) {
1000 1001 1002 1003 1004 1005 1006 1007 1008
		if (debug)
			printf("%s rdlen=%u udpsize=%u flags=%04x version=%u "
			       "tcp=%u ignore=%u id=%u\n",
			       opts[item->test].name, opts[item->test].rdlen,
			       opts[item->test].udpsize, opts[item->test].flags,
			       opts[item->test].version, opts[item->test].tcp,
			       opts[item->test].ignore, item->id);
		if (!item->outstanding++)
			outstanding++;
1009
		sent++;
Mark Andrews's avatar
Mark Andrews committed
1010
		gettimeofday(&item->when, NULL);
1011
		item->when.tv_sec += 1;
Mark Andrews's avatar
Mark Andrews committed
1012 1013 1014
		item->sends = 1;
		APPEND(work, item, link);
		APPEND(ids[item->id], item, idlink);
1015 1016
	} else {
		addtag(item, "failed");
1017 1018
		item->summary->allok = 0;
		item->summary->seenfailure = 1;
1019 1020
		freeitem(item);
	}
Mark Andrews's avatar
Mark Andrews committed
1021 1022
}

Mark Andrews's avatar
Mark Andrews committed
1023 1024 1025
/*
 * Start a series of tests.
 */
Mark Andrews's avatar
Mark Andrews committed
1026
static void
Mark Andrews's avatar
Mark Andrews committed
1027
check(char *zone, char *ns, char *address, struct summary *parent) {
Mark Andrews's avatar
Mark Andrews committed
1028
	size_t i;
Mark Andrews's avatar
Mark Andrews committed
1029
	int fd;
Mark Andrews's avatar
Mark Andrews committed
1030 1031 1032
	struct in_addr addr;
	struct in6_addr addr6;
	struct sockaddr_storage storage;
1033
	struct summary *summary;
Mark Andrews's avatar
Mark Andrews committed
1034

Mark Andrews's avatar
Mark Andrews committed
1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055
	memset(&storage, 0, sizeof(storage));
	if (inet_pton(AF_INET6, address, &addr6) == 1) {
		struct sockaddr_in6 *s = (struct sockaddr_in6 *)&storage;
		s->sin6_len = sizeof(struct sockaddr_in6);
		s->sin6_family = AF_INET6;
		s->sin6_port = htons(53);
		s->sin6_addr = addr6;
		fd = udp6;
	} else if (inet_pton(AF_INET, address, &addr) == 1) {
		struct sockaddr_in *s = (struct sockaddr_in *)&storage;
		s->sin_len = sizeof(struct sockaddr_in);
		s->sin_family = AF_INET;
		s->sin_port = htons(53);
		s->sin_addr = addr;
		fd = udp4;
	} else
		return;

	if (fd == -1)
		return;

1056 1057 1058
	summary = calloc(1, sizeof(*summary));
	if (summary == NULL)
		return;
Mark Andrews's avatar
Mark Andrews committed
1059

Mark Andrews's avatar
Mark Andrews committed
1060 1061 1062
	/*
	 * Hold a reference until all the tests have been initiated.
	 */
Mark Andrews's avatar
Mark Andrews committed
1063 1064 1065 1066 1067
	summary->tests++;
	if (parent)
		INSERTBEFORE(summaries, parent, summary, link);
	else
		APPEND(summaries, summary, link);
Mark Andrews's avatar
Mark Andrews committed
1068

1069
	summary->storage = storage;
1070 1071 1072
	summary->allok = 1;
	summary->allrefused = 1;
	summary->allservfail = 1;
Mark Andrews's avatar
Mark Andrews committed
1073

1074 1075 1076
	ns_makecanon(zone, summary->zone, sizeof(summary->zone));
	i = strlen(summary->zone);
	if (i) summary->zone[i-1] = 0;
Mark Andrews's avatar
Mark Andrews committed
1077

1078 1079 1080
	ns_makecanon(ns, summary->ns, sizeof(summary->ns));
	i = strlen(summary->ns);
	if (i) summary->ns[i-1] = 0;
Mark Andrews's avatar
Mark Andrews committed
1081

Mark Andrews's avatar
Mark Andrews committed
1082
	for (i = 0; i < sizeof(opts)/sizeof(opts[0]); i++) {
Mark Andrews's avatar
style  
Mark Andrews committed
1083
		struct workitem *item;
1084 1085
		if (opts[i].what != 0 && (opts[i].what & what) == 0)
			continue;
Mark Andrews's avatar
Mark Andrews committed
1086

Mark Andrews's avatar
style  
Mark Andrews committed
1087
		item = calloc(1, sizeof(*item));
Mark Andrews's avatar
Mark Andrews committed
1088
		if (item == NULL)
Mark Andrews's avatar
Mark Andrews committed
1089
			break;
1090
		item->summary = summary;
Mark Andrews's avatar
Mark Andrews committed
1091
		item->summary->tests++;
Mark Andrews's avatar
Mark Andrews committed
1092
		item->summary->last = item->test = i;
1093
		item->tcpfd = -1;
Mark Andrews's avatar
Mark Andrews committed
1094
		dotest(item);
Mark Andrews's avatar
Mark Andrews committed
1095 1096
		if (serial)
			break;
Mark Andrews's avatar
Mark Andrews committed
1097
	}
Mark Andrews's avatar
Mark Andrews committed
1098
	report(summary);	/* Release reference. */
Mark Andrews's avatar
Mark Andrews committed
1099 1100
}

1101 1102 1103
static char *
rcodetext(int code) {
	static char buf[64];
Mark Andrews's avatar
Mark Andrews committed
1104

1105
	switch(code) {
1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118
	case ns_r_noerror: return("noerror");
	case ns_r_formerr: return("formerr");
	case ns_r_servfail: return("servfail");
	case ns_r_nxdomain: return("nxdomain");
	case ns_r_notimpl: return("notimpl");
	case ns_r_refused: return("refused");
	case ns_r_yxdomain: return("yxdomain");
	case ns_r_yxrrset: return("yxrrset");
	case ns_r_nxrrset: return("nxrrset");
	case ns_r_notauth: return("notauth");
	case ns_r_notzone: return("notzone");
	case ns_r_badvers: return("badvers");
	case ns_r_badcookie: return("badcookie");
1119
	default:
Mark Andrews's avatar
Mark Andrews committed
1120
		snprintf(buf, sizeof(buf), "rcode%u", code);
1121 1122 1123 1124
		return (buf);
	}
}

Mark Andrews's avatar
Mark Andrews committed
1125 1126 1127
/*
 * Start a lookup using the recursive server(s).
 */
1128 1129 1130 1131 1132 1133 1134 1135
static void
dolookup(struct workitem *item, int type) {
	char name[1024];
	int n, fd = -1;

	item->summary->tests++;
	item->summary->type = item->type = type;

1136
 again:
1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
	if (servers[item->test].sin.sin_family == AF_INET)
		memcpy(&item->summary->storage, &servers[item->test].sin,
		       sizeof(servers[item->test].sin));
	else
		memcpy(&item->summary->storage, &servers[0].sin6,
		       sizeof(servers[item->test].sin6));

	switch (item->summary->storage.ss_family) {
	case AF_INET:
		fd = udp4;
		break;
	case AF_INET6:
		fd = udp6;
		break;
	}

	if (fd == -1) {
1154 1155
		if (++item->test < nservers)
			goto again;
1156
		addtag(item, "skipped");
1157 1158
		item->summary->allok = 0;
		item->summary->seenfailure = 1;
1159 1160 1161
		freeitem(item);
		return;
	}
Mark Andrews's avatar
Mark Andrews committed
1162

1163 1164 1165 1166 1167 1168 1169 1170 1171
	if (type == ns_t_ns)
		n = res_mkquery(ns_o_query, item->summary->zone, ns_c_in,
				type, NULL, 0, NULL,
				item->buf, sizeof(item->buf));
	else
		n = res_mkquery(ns_o_query, item->summary->ns, ns_c_in,
				type, NULL, 0, NULL,
				item->buf, sizeof(item->buf));
	if (n > 0) {
Mark Andrews's avatar
Mark Andrews committed
1172
		int id, tries = 0;
1173

Mark Andrews's avatar
Mark Andrews committed
1174 1175 1176
		/*
		 * Make name canonical.
		 */
1177
		dn_expand(item->buf, item->buf + n, item->buf + 12,
Mark Andrews's avatar
Mark Andrews committed
1178 1179 1180 1181
			  name, sizeof(name));

		switch (type) {
		case ns_t_ns:
1182 1183
			strlcpy(item->summary->zone, name,
				sizeof(item->summary->zone));
Mark Andrews's avatar
Mark Andrews committed
1184 1185 1186
			break;
		case ns_t_a:
		case ns_t_aaaa:
1187 1188
			strlcpy(item->summary->ns, name,
				sizeof(item->summary->zone));
Mark Andrews's avatar
Mark Andrews committed
1189 1190 1191 1192 1193
			break;
		}

		item->buf[2] |= 0x1;	/* set rd */
		id = item->buf[0] << 8 | item->buf[1];
1194 1195 1196 1197

		while (!checkid(&item->summary->storage, id) &&
		       tries++ < 0xffff)
			id = (id + 1) & 0xffff;
Mark Andrews's avatar
Mark Andrews committed
1198

1199
		if (tries == 0xffff) {
1200
			nextserver(item);
1201 1202 1203 1204 1205 1206 1207
			return;
		}

		item->buf[0] = id >> 8;
		item->buf[1] = id & 0xff;
		item->id = id;
		item->buflen = n;
Mark Andrews's avatar