genreport.c 81 KB
Newer Older
Mark Andrews's avatar
Mark Andrews committed
1
/*
Mark Andrews's avatar
Mark Andrews committed
2
 * Copyright (C) 2016, 2017  Internet Systems Consortium, Inc. ("ISC")
Mark Andrews's avatar
Mark Andrews committed
3 4 5 6 7 8
 *
 * 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

11 12
#include <config.h>

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

Mark Andrews's avatar
Mark Andrews committed
20 21 22 23 24 25 26
#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>
27 28
#include <sys/ioctl.h>
#include <sys/uio.h>
29 30 31 32 33 34
#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
35 36

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

41 42 43 44
#ifndef FD_COPY
#define FD_COPY(x, y) memmove(y, x, sizeof(*x))
#endif

Mark Andrews's avatar
Mark Andrews committed
45 46 47 48 49 50
#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
51
#define ns_t_rrsig 46
Mark Andrews's avatar
Mark Andrews committed
52
#define ns_t_nsec 47
Mark Andrews's avatar
Mark Andrews committed
53
#define ns_t_dnskey 48
Mark Andrews's avatar
Mark Andrews committed
54 55 56 57
#define ns_t_dhcid 49
#define ns_t_nsec3 50
#define ns_t_nsec3param 51
#define ns_t_tlsa 52
58
#define ns_t_smimea 53
Mark Andrews's avatar
Mark Andrews committed
59 60 61 62 63
#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
64
#define ns_t_openpgpkey 61
Mark Andrews's avatar
Mark Andrews committed
65 66 67 68 69 70 71 72 73 74
#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
75 76
#define ns_t_avc 258
#define ns_t_doa 259
Mark Andrews's avatar
Mark Andrews committed
77 78
#define ns_t_ta 32768
#define ns_t_dlv 32769
Mark Andrews's avatar
Mark Andrews committed
79

80 81
#define ns_r_badcookie 23

82
static int eof = 0;
Mark Andrews's avatar
Mark Andrews committed
83 84
static int maxfd = -1;
static fd_set rfds, wfds;
Mark Andrews's avatar
Mark Andrews committed
85
static int outstanding = 0;
86
static int maxoutstanding = 100;
Mark Andrews's avatar
Mark Andrews committed
87 88 89 90 91 92

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

static int udp4 = -1;
static int udp6 = -1;
93 94
static int icmp4 = -1;
static int icmp6 = -1;
95

Mark Andrews's avatar
Mark Andrews committed
96 97 98
static int ipv4only = 0;
static int ipv6only = 0;

Mark Andrews's avatar
Mark Andrews committed
99
static int allok  = 0;
100
static int bad  = 0;
Mark Andrews's avatar
fixups  
Mark Andrews committed
101
static int badtag  = 0;
102
static int ednsonly  = 0;
103
static int debug = 0;
Mark Andrews's avatar
Mark Andrews committed
104
static int inorder = 0;
105 106
static int serial = 0;
static long long sent;
107

108 109

#ifdef HAVE_RES_GETSERVERS
110
static union res_sockaddr_union servers[10];
111 112 113 114 115 116 117
#else
static union {
        struct sockaddr_in sin;
        struct sockaddr_in6 sin6;
} servers[10];
#endif

118
static int nservers = 0;
119
int ident = 0;
Mark Andrews's avatar
Mark Andrews committed
120

Mark Andrews's avatar
Mark Andrews committed
121 122 123
/*
 * Doubly linked list macros.
 */
Mark Andrews's avatar
Mark Andrews committed
124 125 126 127 128 129 130 131
#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
132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
	(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
155
		(item)->link.linked = 1; \
Mark Andrews's avatar
Mark Andrews committed
156
	} \
Mark Andrews's avatar
Mark Andrews committed
157 158 159 160 161 162 163 164 165 166 167 168
} 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; \
169
	(item)->link.linked = 0; \
Mark Andrews's avatar
Mark Andrews committed
170 171 172
} while (0)

#define NEXT(item, link) (item)->link.next
Mark Andrews's avatar
Mark Andrews committed
173
#define PREV(item, link) (item)->link.prev
174
#define LINKED(item, link) (item)->link.linked
Mark Andrews's avatar
Mark Andrews committed
175 176 177 178

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

Mark Andrews's avatar
Mark Andrews committed
179 180 181
/*
 * Test groupings
 */
Mark Andrews's avatar
Mark Andrews committed
182 183 184 185 186 187 188
#define EDNS 0x01
#define COMM 0x02
#define FULL 0x04
#define TYPE 0x08
#define EXPL 0x10

static int what = EDNS;
189

Mark Andrews's avatar
Mark Andrews committed
190
static struct {
Mark Andrews's avatar
Mark Andrews committed
191 192 193 194 195 196 197 198
	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 */
199
	unsigned int cookie;		/* opt record has cookie */
Mark Andrews's avatar
Mark Andrews committed
200 201 202 203 204 205 206 207 208 209
	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 */
210
} opts[] = {
211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228
	/*                           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
229
	{ "ednsopt",   EDNS,  4, "\x00\x64\x00\x00",	/* 100 */
230
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
231
	{ "edns1opt",  EDNS,  4, "\x00\x64\x00\x00",	/* 100 */
232 233 234 235 236
				     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 },
237
	{ "optlist",   EDNS,  4 + 8 + 4 + 12,
238 239 240 241
	  "\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 */
242 243 244
				     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
245
	{ "ednsnsid", FULL,  4, "\x00\x03\x00\x00",	/* NSID */
246
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
247
	{ "ednscookie", FULL, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
248
				     4096, 0x0000, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
249
	{ "ednsexpire", FULL, 4, "\x00\x09\x00\x00",	/* EXPIRE */
250
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
251
	{ "ednssubnet", FULL,  8, "\x00\x08\x00\x04\x00\x01\x00\x00",	/* ECS */
252
				     4096, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
253
	{ "edns1nsid", FULL,  4, "\x00\x03\x00\x00",	/* NSID */
254
				     4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
255
	{ "edns1cookie", FULL, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
256
				     4096, 0x0000, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
257
	{ "edns1expire", FULL, 4, "\x00\x09\x00\x00",	/* EXPIRE */
258
				     4096, 0x0000, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
259
	{ "edns1subnet", FULL,  8, "\x00\x08\x00\x04\x00\x01\x00\x00",	/* ECS */
260 261 262 263
				     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
264
	{ "bind11",    COMM, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
265
				     4096, 0x8000, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_soa },
Mark Andrews's avatar
Mark Andrews committed
266
	{ "dig11",     COMM, 12, "\x00\x0a\x00\x08\x01\x02\x03\x04\x05\x06\x07\x08", /* COOKIE */
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 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338
				     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 },
339 340
	{ "AVC",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_avc },
	{ "DOA",       TYPE,  0, "",    0, 0x0000, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  0, ns_t_doa },
341 342
	{ "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 }
343 344
};

Mark Andrews's avatar
Mark Andrews committed
345 346 347
/*
 * Summary structure where results from multiple lookups are recorded.
 */
348
struct summary {
Mark Andrews's avatar
Mark Andrews committed
349 350 351 352 353
	struct {
		struct summary *prev;
		struct summary *next;
		int linked;
	} link;
Mark Andrews's avatar
Mark Andrews committed
354 355 356
	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
357
	int tests;			/* number of outstanding tests */
Mark Andrews's avatar
Mark Andrews committed
358
	int last;			/* last test sent */
Mark Andrews's avatar
Mark Andrews committed
359
	int deferred;			/* was the printing deferred */
Mark Andrews's avatar
Mark Andrews committed
360
	int done;			/* we are done */
Mark Andrews's avatar
Mark Andrews committed
361 362 363 364 365 366
	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 */
367 368
	int faileda;
	int failedaaaa;
369 370
	int cnamea;			/* Nameserver is CNAME */
	int cnameaaaa;			/* Nameserver is CNAME */
Mark Andrews's avatar
Mark Andrews committed
371
	int seenrrsig;			/* a rrsig was seen in "do" test */
372 373 374 375 376 377
	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
378
	struct summary *xlink;		/* cross link of recursive A/AAAA */
379 380 381
	char results[sizeof(opts)/sizeof(opts[0])][100];
};

Mark Andrews's avatar
Mark Andrews committed
382 383 384 385 386
static struct {
	struct summary *head;
	struct summary *tail;
} summaries;

Mark Andrews's avatar
Mark Andrews committed
387 388 389 390
struct workitem {
	struct {
		struct workitem *next;
		struct workitem *prev;
391
		int linked;
392
	} link, clink, plink, rlink, idlink, seqlink;
Mark Andrews's avatar
Mark Andrews committed
393 394 395
	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
396 397 398
	int test;			/* test number / server number */
	int sends;			/* number of times this UDP request
					 * has been sent */
Mark Andrews's avatar
Mark Andrews committed
399 400
	int buflen;			/* the size of the request to be sent */
	int tcpfd;			/* the tcp file descriptor */
Mark Andrews's avatar
Mark Andrews committed
401 402
	int outstanding;		/* outstanding has been set */
	int havelen;			/* readlen is tcp message length */
Mark Andrews's avatar
Mark Andrews committed
403 404 405 406 407
	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
408 409
};

410
/*
Mark Andrews's avatar
Mark Andrews committed
411
 * Work queues:
412 413 414
 *	'work' udp qeries;
 *	'connecting' tcp qeries;
 *	'reading' tcp qeries;
Mark Andrews's avatar
Mark Andrews committed
415
 *	'pending' deferred work items;
Mark Andrews's avatar
Mark Andrews committed
416 417 418
 *
 * Outstanding queries by qid.
 *	'ids'
419 420 421
 *
 * Outstanding icmp by seq.
 *	'ids'
422
 */
Mark Andrews's avatar
Mark Andrews committed
423 424 425
static struct {
	struct workitem *head;
	struct workitem *tail;
426
} work, connecting, reading, pending, ids[0x10000], seq[0x10000];
Mark Andrews's avatar
Mark Andrews committed
427

Mark Andrews's avatar
Mark Andrews committed
428 429 430
static void
dotest(struct workitem *item);

431 432 433
static void
nextserver(struct workitem *item);

434 435
static void
connecttoserver(struct workitem *item);
Mark Andrews's avatar
Mark Andrews committed
436

Mark Andrews's avatar
Mark Andrews committed
437 438 439
static void
report(struct summary *summary);

Mark Andrews's avatar
Mark Andrews committed
440 441 442 443 444
static int
storage_equal(struct sockaddr_storage *s1, struct sockaddr_storage *s2) {
	struct sockaddr_in *sin1, *sin2;
	struct sockaddr_in6 *sin61, *sin62;

445
	if (s1->ss_family != s2->ss_family)
Mark Andrews's avatar
Mark Andrews committed
446 447 448 449 450 451
		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
452

Mark Andrews's avatar
Mark Andrews committed
453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468
		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
469 470 471
/*
 * Check if it is safe to use this id to this address.
 */
Mark Andrews's avatar
Mark Andrews committed
472 473 474 475 476 477
static int
checkid(struct sockaddr_storage *storage, int id) {
	struct workitem *item;

	item = HEAD(ids[id]);
	while (item != NULL &&
478
	       !storage_equal(storage, &item->summary->storage))
Mark Andrews's avatar
Mark Andrews committed
479 480 481 482
		item = NEXT(item, idlink);
	return ((item == NULL) ? 1 : 0);
}

483 484 485 486 487 488 489 490 491 492 493 494 495 496 497
/*
 * 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
498 499 500 501 502 503 504
static void
freesummary(struct summary *summary) {
	if (LINKED(summary, link))
		UNLINK(summaries, summary, link);
	free(summary);
}

Mark Andrews's avatar
Mark Andrews committed
505 506 507
/*
 * Generate a report line.
 */
508
static void
Mark Andrews's avatar
Mark Andrews committed
509
printandfree(struct summary *summary) {
510 511 512 513 514 515 516 517
	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
518
	    summary->nodataa && summary->nodataaaaa) {
519 520
		printf("%s. %s: no address records found\n",
		       summary->zone, summary->ns);
Mark Andrews's avatar
Mark Andrews committed
521
		freesummary(summary);
522 523
		return;
	}
524

525 526 527 528
	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
529
		freesummary(summary);
530 531
		return;
	}
532

533 534 535 536 537 538 539 540
	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;
	}

541 542 543 544 545 546 547 548
	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;
	}

549 550 551
	/*
	 * Do deferred xlink failure reports.
	 */
552 553
	if (summary->type == ns_t_a &&
	    summary->nodataa && summary->failedaaaa) {
554 555 556 557 558
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" AAAAA lookup failed\n");
		freesummary(summary);
		return;
	}
559 560
	if (summary->type == ns_t_aaaa &&
	    summary->nodataaaaa && summary->faileda) {
561 562 563 564 565
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" A lookup failed\n");
		freesummary(summary);
		return;
	}
566 567
	if (summary->type == ns_t_a &&
	    summary->faileda && summary->nxdomainaaaa) {
568 569 570 571 572
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" AAAAA nxdomain\n");
		freesummary(summary);
		return;
	}
573 574
	if (summary->type == ns_t_aaaa &&
	    summary->failedaaaa && summary->nxdomaina) {
575 576 577 578 579
		printf("%s. %s:", summary->zone, summary->ns);
		printf(" A nxdomain\n");
		freesummary(summary);
		return;
	}
580

Mark Andrews's avatar
Mark Andrews committed
581
	if (summary->done || summary->nodataa || summary->nodataaaaa) {
Mark Andrews's avatar
Mark Andrews committed
582
		freesummary(summary);
583 584
		return;
	}
Mark Andrews's avatar
Mark Andrews committed
585

586 587 588
	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
589
			freesummary(summary);
590 591 592 593 594 595
			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
596
		freesummary(summary);
597 598 599 600 601
		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
602
		freesummary(summary);
603 604 605 606 607
		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
608
		freesummary(summary);
609 610 611
		return;
	}
	if (summary->type == ns_t_ns) {
Mark Andrews's avatar
Mark Andrews committed
612
		printf("%s.:", summary->zone);
613
		printf(" NS lookup failed\n");
Mark Andrews's avatar
Mark Andrews committed
614
		freesummary(summary);
615 616
		return;
	}
617

618 619 620 621 622
	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
623
		freesummary(summary);
Mark Andrews's avatar
Mark Andrews committed
624 625 626
		return;
	}

627 628 629 630 631
	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
632

633
	if (addr == NULL)
634
		strncpy(addrbuf, "<unknown>", sizeof(addrbuf));
635 636 637 638
	else
		inet_ntop(summary->storage.ss_family, addr,
			  addrbuf, sizeof(addrbuf));

639 640
	x = -1;
	printf("%s. @%s (%s.):", summary->zone, addrbuf, summary->ns);
Mark Andrews's avatar
Mark Andrews committed
641 642 643 644 645
	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
fixups  
Mark Andrews committed
646
				continue;
Mark Andrews's avatar
Mark Andrews committed
647
			if (summary->results[i][0] == 0)
648
				strncpy(summary->results[i], "skipped", 100);
Mark Andrews's avatar
Mark Andrews committed
649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664
			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
fixups  
Mark Andrews committed
665
		}
666
	printf("\n");
Mark Andrews's avatar
Mark Andrews committed
667 668 669 670 671 672
	freesummary(summary);
}

static void
report(struct summary *summary) {

Mark Andrews's avatar
Mark Andrews committed
673 674 675 676
	/*
	 * Send the next test now that we have completed the last test.
	 */
	if (serial && summary->type == 0 && summary->tests == 1) {
677 678 679
		for (summary->last++;
		     summary->last < sizeof(opts)/sizeof(opts[0]);
		     summary->last++) {
Mark Andrews's avatar
Mark Andrews committed
680
			struct workitem *item;
681 682
			if (opts[summary->last].what != 0 &&
			    (opts[summary->last].what & what) == 0)
Mark Andrews's avatar
Mark Andrews committed
683 684 685 686 687
				continue;
			item = calloc(1, sizeof(*item));
			if (item == NULL)
				continue;
			item->summary = summary;
688
			item->test = item->summary->last;
689
			item->tcpfd = -1;
Mark Andrews's avatar
Mark Andrews committed
690 691 692 693 694
			dotest(item);
			return;
		}
	}

Mark Andrews's avatar
Mark Andrews committed
695 696 697 698 699 700 701 702 703 704 705 706
	/*
	 * 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) {
707 708 709 710 711 712 713 714
		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
715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730
		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;
		}
731 732 733 734 735 736 737 738 739 740
		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
741 742 743 744 745 746 747 748 749 750 751

		/*
		 * Remove the cross link.
		 */
		summary->xlink->xlink = NULL;
		summary->xlink = NULL;
		if (summary->done) {
			freesummary(summary);
			goto print_deferred;
		}
	}
752 753 754 755 756 757
	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
758 759 760 761 762 763 764 765 766 767

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

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

Mark Andrews's avatar
Mark Andrews committed
770 771 772
/*
 * Free a work item and unlink.
 */
773 774
static void
freeitem(struct workitem * item) {
Mark Andrews's avatar
Mark Andrews committed
775 776
	if (item->summary)
		report(item->summary);
777
	if (item->tcpfd != -1) {
778 779 780 781 782 783 784 785 786 787
		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
788 789
	if (LINKED(item, plink))
		UNLINK(pending, item, plink);
790 791 792 793 794 795
	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);
796 797
	if (LINKED(item, seqlink))
		UNLINK(seq[item->id], item, seqlink);
798 799 800
	free(item);
}

Mark Andrews's avatar
Mark Andrews committed
801 802 803
/*
 * Add a tag to the report.
 */
804
static void
805
addtag(struct workitem *item, const char *tag) {
806
	char *result = item->summary->results[item->test];
807 808
	if (result[0]) strncat(result, ",", 100);
	strncat(result, tag, 100);
809 810
}

Mark Andrews's avatar
Mark Andrews committed
811 812 813
/*
 * Resend a UDP request.
 */
814 815 816
static void
resend(struct workitem *item) {
	int n, fd = -1;
817
	socklen_t ss_len;
818 819 820 821

	switch (item->summary->storage.ss_family) {
	case AF_INET:
		fd = udp4;
822
		ss_len = sizeof(struct sockaddr_in);
823 824 825
		break;
	case AF_INET6:
		fd = udp6;
826
		ss_len = sizeof(struct sockaddr_in6);
827 828 829 830 831
		break;
	}

	if (fd == -1) {
		addtag(item, "skipped");
832
		item->summary->allok = 0;
833 834 835 836
		freeitem(item);
		return;
	}

837
	if (!item->outstanding && outstanding > maxoutstanding) {
838 839
		gettimeofday(&item->when, NULL);
		item->when.tv_sec += 1;
Mark Andrews's avatar
Mark Andrews committed
840 841 842
		if (LINKED(item, link))
			UNLINK(work, item, link);
		APPEND(pending, item, plink);
843 844 845 846
		return;
	}

	n = sendto(fd, item->buf, item->buflen, 0,
847
		   (struct sockaddr *)&item->summary->storage, ss_len);
848 849 850 851 852 853 854 855
	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);
856
		sent++;
857 858 859 860 861
		if (!item->outstanding++)
			outstanding++;
		gettimeofday(&item->when, NULL);
		item->when.tv_sec += 1;
		item->sends++;
Mark Andrews's avatar
Mark Andrews committed
862 863
		if (LINKED(item, link))
			UNLINK(work, item, link);
864
		APPEND(work, item, link);
865 866
	} else if (item->summary->type) {
		nextserver(item);
867 868
	} else {
		addtag(item, "failed");
869 870
		item->summary->allok = 0;
		item->summary->seenfailure = 1;
871 872 873 874
		freeitem(item);
	}
}

Mark Andrews's avatar
Mark Andrews committed
875 876 877
/*
 * Start a individual test.
 */
Mark Andrews's avatar
Mark Andrews committed
878 879 880 881
static void
dotest(struct workitem *item) {
	unsigned char *cp;
	unsigned int ttl;
Mark Andrews's avatar
Mark Andrews committed
882
	int n, fd, id, tries = 0, opcode;
883 884
	socklen_t ss_len;
	
Mark Andrews's avatar
Mark Andrews committed
885

886
	switch (item->summary->storage.ss_family) {
Mark Andrews's avatar
Mark Andrews committed
887 888
	case AF_INET:
		fd = udp4;
889
		ss_len = sizeof(struct sockaddr_in);
Mark Andrews's avatar
Mark Andrews committed
890 891 892
		break;
	case AF_INET6:
		fd = udp6;
893
		ss_len = sizeof(struct sockaddr_in6);
Mark Andrews's avatar
Mark Andrews committed
894 895 896 897
		break;
	}

	if (fd == -1) {
898
		addtag(item, "skipped");
899
		item->summary->allok = 0;
900
		freeitem(item);
Mark Andrews's avatar
Mark Andrews committed
901 902
		return;
	}
Mark Andrews's avatar
Mark Andrews committed
903

Mark Andrews's avatar
Mark Andrews committed
904 905 906 907
	/*
	 * 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
908 909 910 911 912 913 914 915
	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,
916 917
			opts[item->test].type, NULL, 0, NULL,
			item->buf, sizeof(item->buf));
Mark Andrews's avatar
Mark Andrews committed
918 919 920 921 922 923 924
	/* 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;
925
		n = 12;
Mark Andrews's avatar
Mark Andrews committed
926
	}
Mark Andrews's avatar
Mark Andrews committed
927 928 929 930

	/*
	 * Set DNS flags as specified by test.
	 */
931 932
	if (opts[item->test].tc)
		item->buf[2] |= 0x2;	/* set tc */
Mark Andrews's avatar
Mark Andrews committed
933 934 935 936
	if (opts[item->test].rd)
		item->buf[2] |= 0x1;	/* set rd */
	else
		item->buf[2] &= ~0x1;	/* clear rd */
937 938 939 940 941 942 943 944
	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
945

946
	if (n > 12) {
947
		char name[1024];
Mark Andrews's avatar
Mark Andrews committed
948 949 950
		/*
		 * Make zone canonical.
		 */
951 952
		dn_expand(item->buf, item->buf + n, item->buf + 12,
			  name, sizeof(name));
953
		strncpy(item->summary->zone, name,
954
			sizeof(item->summary->zone));
955
	}
Mark Andrews's avatar
Mark Andrews committed
956

Mark Andrews's avatar
Mark Andrews committed
957 958 959 960 961
	/*
	 * 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
962
		cp = item->buf + n;
963 964 965 966 967 968 969 970 971 972 973
		*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
974 975
		memcpy(cp, opts[item->test].rdata, opts[item->test].rdlen);
		cp += opts[item->test].rdlen;
976 977
		item->buf[11] = 1;			/* adcount */
		n = cp - item->buf;			/* total length */
Mark Andrews's avatar
Mark Andrews committed
978 979 980
	}

	if (n > 0) {
Mark Andrews's avatar
Mark Andrews committed
981 982 983
		/*
		 * Adjust id if it clashes with a outstanding request.
		 */
Mark Andrews's avatar
Mark Andrews committed
984 985
		id = item->buf[0] << 8 | item->buf[1];

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

Mark Andrews's avatar
Mark Andrews committed
990
		if (tries == 0xffff) {
991
			addtag(item, "skipped");
992
			item->summary->allok = 0;
993
			item->summary->seenfailure = 1;
994
			freeitem(item);
Mark Andrews's avatar
Mark Andrews committed
995 996
			return;
		}
Mark Andrews's avatar
Mark Andrews committed
997

Mark Andrews's avatar
Mark Andrews committed
998 999 1000 1001 1002
		item->buf[0] = id >> 8;
		item->buf[1] = id & 0xff;
		item->id = id;
		item->buflen = n;

1003 1004 1005 1006 1007
		if (opts[item->test].tcp) {
			connecttoserver(item);
			return;
		}

Mark Andrews's avatar
Mark Andrews committed
1008 1009 1010
		/*
		 * If there is too much outstanding work queue this item.
		 */
1011
		if (!item->outstanding && outstanding > maxoutstanding) {
1012 1013
			gettimeofday(&item->when, NULL);
			item->when.tv_sec += 1;
Mark Andrews's avatar
Mark Andrews committed
1014
			APPEND(pending, item, plink);
1015 1016 1017 1018
			APPEND(ids[item->id], item, idlink);
			return;
		}

Mark Andrews's avatar
Mark Andrews committed
1019
		n = sendto(fd, item->buf, item->buflen, 0,
1020
			   (struct sockaddr *)&item->summary->storage, ss_len);
Mark Andrews's avatar
Mark Andrews committed
1021 1022 1023
	}

	if (n > 0) {
1024 1025 1026 1027 1028 1029 1030 1031 1032
		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++;
1033
		sent++;
Mark Andrews's avatar
Mark Andrews committed
1034
		gettimeofday(&item->when, NULL);
1035
		item->when.tv_sec += 1;
Mark Andrews's avatar
Mark Andrews committed
1036 1037 1038
		item->sends = 1;
		APPEND(work, item, link);
		APPEND(ids[item->id], item, idlink);
1039 1040
	} else {
		addtag(item, "failed");
1041 1042