#include <config.h>

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <limits.h>

#include <dns/byaddr.h>

#include <dns/fixedname.h>

#include <dns/message.h>

#include <dns/name.h>

#ifdef DNS_OPT_NEWCODES

#include <dns/opt.h>

#endif /* DNS_OPT_NEWCODES */

#include <dns/rdata.h>

#include <dns/rdataclass.h>

#include <dns/rdatalist.h>

#include <dns/rdataset.h>

#include <dns/rdatastruct.h>

#include <dns/rdatatype.h>

#include <dns/result.h>

#include <dns/tsig.h>

#include <dst/dst.h>

#include <isc/app.h>

#include <isc/base64.h>

#include <isc/entropy.h>

#include <isc/lang.h>

#include <isc/netaddr.h>

#include <isc/netdb.h>

#include <isc/print.h>

#include <isc/result.h>

#include <isc/string.h>

#include <isc/task.h>

#include <isc/timer.h>

#include <isc/types.h>

#include <isc/util.h>

#include <dig/dig.h>

#ifdef HAVE_ADDRINFO

#ifdef HAVE_GETADDRINFO

#ifdef HAVE_GAISTRERROR

#define USE_GETADDRINFO

#endif

#endif

#endif

#ifndef USE_GETADDRINFO

#ifndef ISC_PLATFORM_NONSTDHERRNO

extern int h_errno;

#endif

#endif

ISC_LIST(dig_lookup_t) lookup_list;

dig_serverlist_t server_list;

ISC_LIST(dig_searchlist_t) search_list;

have_ipv4 = ISC_FALSE,

have_ipv6 = ISC_FALSE,

specified_source = ISC_FALSE,

free_now = ISC_FALSE,

cancel_now = ISC_FALSE,

usesearch = ISC_FALSE,

qr = ISC_FALSE,

is_dst_up = ISC_FALSE;

in_port_t port = 53;

unsigned int timeout = 0;

isc_mem_t *mctx = NULL;

isc_taskmgr_t *taskmgr = NULL;

isc_task_t *global_task = NULL;

isc_timermgr_t *timermgr = NULL;

isc_socketmgr_t *socketmgr = NULL;

isc_sockaddr_t bind_address;

isc_sockaddr_t bind_any;

isc_buffer_t rootbuf;

int sendcount = 0;

int recvcount = 0;

int sockcount = 0;

int ndots = -1;

int tries = 2;

int lookup_counter = 0;

int exitcode = 0;

char keynametext[MXNAME];

char keyfile[MXNAME] = "";

char keysecret[MXNAME] = "";

dns_name_t keyname;

isc_buffer_t *namebuf = NULL;

dns_tsigkey_t *key = NULL;

isc_boolean_t validated = ISC_TRUE;

isc_entropy_t *entp = NULL;

isc_mempool_t *commctx = NULL;

isc_boolean_t debugging = ISC_FALSE;

isc_boolean_t memdebugging = ISC_FALSE;

char *progname = NULL;

isc_mutex_t lookup_lock;

dig_lookup_t *current_lookup = NULL;

#define LOCK_LOOKUP {\

debug("lock_lookup %s:%d", __FILE__, __LINE__);\

check_result(isc_mutex_lock((&lookup_lock)), "isc_mutex_lock");\

debug("success");\

}

#define UNLOCK_LOOKUP {\

debug("unlock_lookup %s:%d", __FILE__, __LINE__);\

check_result(isc_mutex_unlock((&lookup_lock)),\

"isc_mutex_unlock");\

}

static void

cancel_lookup(dig_lookup_t *lookup);

static void

recv_done(isc_task_t *task, isc_event_t *event);

static void

connect_timeout(isc_task_t *task, isc_event_t *event);

static void

launch_next_query(dig_query_t *query, isc_boolean_t include_question);

char *

next_token(char **stringp, const char *delim) {

char *res;

do {

res = strsep(stringp, delim);

if (res == NULL)

break;

} while (*res == '\0');

return (res);

}

static int

count_dots(char *string) {

char *s;

int i = 0;

s = string;

while (*s != '\0') {

if (*s == '.')

i++;

s++;

}

return (i);

}

static void

hex_dump(isc_buffer_t *b) {

unsigned int len;

isc_region_t r;

isc_buffer_usedregion(b, &r);

printf("%d bytes\n", r.length);

for (len = 0; len < r.length; len++) {

printf("%02x ", r.base[len]);

if (len % 16 == 15)

printf("\n");

}

if (len % 16 != 0)

printf("\n");

}

get_reverse(char *reverse, char *value, isc_boolean_t nibble) {

int adrs[4];

char working[MXNAME];

int remaining;

int i, n;

isc_result_t result;

result = DNS_R_BADDOTTEDQUAD;

reverse[0] = 0;

debug("get_reverse(%s)", value);

if (strspn(value, "0123456789.") == strlen(value)) {

n = sscanf(value, "%d.%d.%d.%d",

&adrs[0], &adrs[1],

&adrs[2], &adrs[3]);

if (n == 0) {

return (DNS_R_BADDOTTEDQUAD);

}

reverse[MXNAME - 1] = 0;

for (i = n - 1; i >= 0; i--) {

snprintf(working, sizeof(working), "%d.",

adrs[i]);

remaining = MXNAME - strlen(reverse) - 1;

strncat(reverse, working, remaining);

}

remaining = MXNAME - strlen(reverse) - 1;

strncat(reverse, "in-addr.arpa.", remaining);

result = ISC_R_SUCCESS;

} else if (strspn(value, "0123456789abcdefABCDEF:")

== strlen(value)) {

isc_netaddr_t addr;

dns_fixedname_t fname;

dns_name_t *name;

addr.family = AF_INET6;

n = inet_pton(AF_INET6, value, &addr.type.in6);

if (n <= 0)

return (DNS_R_BADDOTTEDQUAD);

dns_fixedname_init(&fname);

name = dns_fixedname_name(&fname);

result = dns_byaddr_createptrname(&addr, nibble, name);

if (result != ISC_R_SUCCESS)

return (result);

dns_name_format(name, reverse, MXNAME);

}

return (result);

}

fatal(const char *format, ...) {

va_list args;

fprintf(stderr, "%s: ", progname);

va_start(args, format);

vfprintf(stderr, format, args);

va_end(args);

fprintf(stderr, "\n");

if (exitcode < 10)

exitcode = 10;

exit(exitcode);

}

debug(const char *format, ...) {

va_list args;

if (debugging) {

va_start(args, format);

vfprintf(stderr, format, args);

va_end(args);

fprintf(stderr, "\n");

}

}

check_result(isc_result_t result, const char *msg) {

if (result != ISC_R_SUCCESS) {

fatal("%s: %s", msg, isc_result_totext(result));

}

}

dig_server_t *

make_server(const char *servname) {

dig_server_t *srv;

REQUIRE(servname != NULL);

debug("make_server(%s)", servname);

srv = isc_mem_allocate(mctx, sizeof(struct dig_server));

if (srv == NULL)

fatal("Memory allocation failure in %s:%d",

__FILE__, __LINE__);

strncpy(srv->servername, servname, MXNAME);

srv->servername[MXNAME-1] = 0;

ISC_LINK_INIT(srv, link);

return (srv);

}

clone_server_list(dig_serverlist_t src,

dig_serverlist_t *dest)

{

dig_server_t *srv, *newsrv;

debug("clone_server_list()");

srv = ISC_LIST_HEAD(src);

while (srv != NULL) {

newsrv = make_server(srv->servername);

ISC_LINK_INIT(newsrv, link);

ISC_LIST_ENQUEUE(*dest, newsrv, link);

srv = ISC_LIST_NEXT(srv, link);

}

}

dig_lookup_t *

make_empty_lookup(void) {

dig_lookup_t *looknew;

debug("make_empty_lookup()");

INSIST(!free_now);

looknew = isc_mem_allocate(mctx, sizeof(struct dig_lookup));

if (looknew == NULL)

fatal("Memory allocation failure in %s:%d",

__FILE__, __LINE__);

looknew->pending = ISC_TRUE;

looknew->textname[0] = 0;

looknew->cmdline[0] = 0;

looknew->rdtype = dns_rdatatype_a;

looknew->rdclass = dns_rdataclass_in;

looknew->rdtypeset = ISC_FALSE;

looknew->rdclassset = ISC_FALSE;

looknew->sendspace = NULL;

looknew->sendmsg = NULL;

looknew->name = NULL;

looknew->oname = NULL;

looknew->timer = NULL;

looknew->xfr_q = NULL;

looknew->current_query = NULL;

looknew->doing_xfr = ISC_FALSE;

looknew->ixfr_serial = ISC_FALSE;

looknew->trace = ISC_FALSE;

looknew->trace_root = ISC_FALSE;

looknew->identify = ISC_FALSE;

looknew->identify_previous_line = ISC_FALSE;

looknew->ignore = ISC_FALSE;

looknew->servfail_stops = ISC_FALSE;

looknew->besteffort = ISC_TRUE;

looknew->dnssec = ISC_FALSE;

looknew->udpsize = 0;

looknew->recurse = ISC_TRUE;

looknew->aaonly = ISC_FALSE;

looknew->adflag = ISC_FALSE;

looknew->cdflag = ISC_FALSE;

looknew->ns_search_only = ISC_FALSE;

looknew->ns_search_only_leafnode = ISC_FALSE;

looknew->origin = NULL;

looknew->querysig = NULL;

looknew->retries = tries;

looknew->nsfound = 0;

looknew->tcp_mode = ISC_FALSE;

looknew->nibble = ISC_FALSE;

looknew->comments = ISC_TRUE;

looknew->stats = ISC_TRUE;

looknew->section_question = ISC_TRUE;

looknew->section_answer = ISC_TRUE;

looknew->section_authority = ISC_TRUE;

looknew->section_additional = ISC_TRUE;

looknew->new_search = ISC_FALSE;

#ifdef DNS_OPT_NEWCODES_LIVE

looknew->zonename[0] = 0;

looknew->viewname[0] = 0;

#endif /* DNS_OPT_NEWCODES_LIVE */

ISC_LINK_INIT(looknew, link);

ISC_LIST_INIT(looknew->q);

ISC_LIST_INIT(looknew->my_server_list);

return (looknew);

}

dig_lookup_t *

clone_lookup(dig_lookup_t *lookold, isc_boolean_t servers) {

dig_lookup_t *looknew;

debug("clone_lookup()");

INSIST(!free_now);

looknew = make_empty_lookup();

INSIST(looknew != NULL);

strncpy(looknew->textname, lookold->textname, MXNAME);

strncpy(looknew->cmdline, lookold->cmdline, MXNAME);

looknew->textname[MXNAME-1] = 0;

looknew->rdtype = lookold->rdtype;

looknew->rdclass = lookold->rdclass;

looknew->rdtypeset = lookold->rdtypeset;

looknew->rdclassset = lookold->rdclassset;

looknew->doing_xfr = lookold->doing_xfr;

looknew->ixfr_serial = lookold->ixfr_serial;

looknew->trace = lookold->trace;

looknew->trace_root = lookold->trace_root;

looknew->identify = lookold->identify;

looknew->identify_previous_line = lookold->identify_previous_line;

looknew->ignore = lookold->ignore;

looknew->servfail_stops = lookold->servfail_stops;

looknew->besteffort = lookold->besteffort;

looknew->dnssec = lookold->dnssec;

looknew->udpsize = lookold->udpsize;

looknew->recurse = lookold->recurse;

looknew->aaonly = lookold->aaonly;

looknew->adflag = lookold->adflag;

looknew->cdflag = lookold->cdflag;

looknew->ns_search_only = lookold->ns_search_only;

looknew->ns_search_only_leafnode = lookold->ns_search_only_leafnode;

looknew->tcp_mode = lookold->tcp_mode;

looknew->comments = lookold->comments;

looknew->stats = lookold->stats;

looknew->section_question = lookold->section_question;

looknew->section_answer = lookold->section_answer;

looknew->section_authority = lookold->section_authority;

looknew->section_additional = lookold->section_additional;

looknew->retries = lookold->retries;

#ifdef DNS_OPT_NEWCODES_LIVE

strncpy(looknew->viewname, lookold->viewname, MXNAME);

strncpy(looknew->zonename, lookold->zonename, MXNAME);

#endif /* DNS_OPT_NEWCODES_LIVE */

if (servers)

clone_server_list(lookold->my_server_list,

&looknew->my_server_list);

return (looknew);

}

dig_lookup_t *

requeue_lookup(dig_lookup_t *lookold, isc_boolean_t servers) {

dig_lookup_t *looknew;

debug("requeue_lookup()");

lookup_counter++;

if (lookup_counter > LOOKUP_LIMIT)

fatal("Too many lookups");

looknew = clone_lookup(lookold, servers);

INSIST(looknew != NULL);

debug("before insertion, init@%p -> %p, new@%p -> %p",

lookold, lookold->link.next, looknew, looknew->link.next);

ISC_LIST_PREPEND(lookup_list, looknew, link);

debug("after insertion, init -> %p, new = %p, new -> %p",

lookold, looknew, looknew->link.next);

return (looknew);

}

static void

setup_text_key(void) {

isc_result_t result;

isc_buffer_t secretbuf;

int secretsize;

unsigned char *secretstore;

isc_stdtime_t now;

debug("setup_text_key()");

result = isc_buffer_allocate(mctx, &namebuf, MXNAME);

check_result(result, "isc_buffer_allocate");

dns_name_init(&keyname, NULL);

check_result(result, "dns_name_init");

isc_buffer_putstr(namebuf, keynametext);

secretsize = strlen(keysecret) * 3 / 4;

secretstore = isc_mem_allocate(mctx, secretsize);

if (secretstore == NULL)

fatal("Memory allocation failure in %s:%d",

__FILE__, __LINE__);

isc_buffer_init(&secretbuf, secretstore, secretsize);

result = isc_base64_decodestring(keysecret, &secretbuf);

if (result != ISC_R_SUCCESS) {

printf(";; Couldn't create key %s: %s\n",

keynametext, isc_result_totext(result));

goto failure;

}

secretsize = isc_buffer_usedlength(&secretbuf);

isc_stdtime_get(&now);

result = dns_name_fromtext(&keyname, namebuf,

dns_rootname, ISC_FALSE,

namebuf);

if (result != ISC_R_SUCCESS) {

printf(";; Couldn't create key %s: %s\n",

keynametext, isc_result_totext(result));

goto failure;

}

result = dns_tsigkey_create(&keyname, dns_tsig_hmacmd5_name,

secretstore, secretsize,

ISC_TRUE, NULL, now, now, mctx,

NULL, &key);

if (result != ISC_R_SUCCESS) {

printf(";; Couldn't create key %s: %s\n",

keynametext, isc_result_totext(result));

}

failure:

isc_mem_free(mctx, secretstore);

dns_name_invalidate(&keyname);

isc_buffer_free(&namebuf);

}

static void

setup_file_key(void) {

isc_result_t result;

isc_buffer_t secretbuf;

unsigned char *secretstore = NULL;

int secretlen;

dst_key_t *dstkey = NULL;

isc_stdtime_t now;

debug("setup_file_key()");

result = dst_key_fromnamedfile(keyfile, DST_TYPE_PRIVATE,

mctx, &dstkey);

if (result != ISC_R_SUCCESS) {

fprintf(stderr, "Couldn't read key from %s: %s\n",

keyfile, isc_result_totext(result));

goto failure;

}

/*

secretlen = (dst_key_size(dstkey) + 7) >> 3;

secretstore = isc_mem_allocate(mctx, secretlen);

if (secretstore == NULL)

fatal("out of memory");

isc_buffer_init(&secretbuf, secretstore, secretlen);

result = dst_key_tobuffer(dstkey, &secretbuf);

if (result != ISC_R_SUCCESS) {

fprintf(stderr, "Couldn't read key from %s: %s\n",

keyfile, isc_result_totext(result));

goto failure;

}

isc_stdtime_get(&now);

dns_name_init(&keyname, NULL);

dns_name_clone(dst_key_name(dstkey), &keyname);

result = dns_tsigkey_create(&keyname, dns_tsig_hmacmd5_name,

secretstore, secretlen,

ISC_TRUE, NULL, now, now, mctx,

NULL, &key);

if (result != ISC_R_SUCCESS) {

printf(";; Couldn't create key %s: %s\n",

keynametext, isc_result_totext(result));

}

failure:

if (dstkey != NULL)

dst_key_free(&dstkey);

if (secretstore != NULL)

isc_mem_free(mctx, secretstore);

}

static dig_searchlist_t *

make_searchlist_entry(char *domain) {

dig_searchlist_t *search;

search = isc_mem_allocate(mctx, sizeof(*search));

if (search == NULL)

fatal("Memory allocation failure in %s:%d",

__FILE__, __LINE__);

strncpy(search->origin, domain, MXNAME);

search->origin[MXNAME-1] = 0;

ISC_LINK_INIT(search, link);

return (search);

}

setup_system(void) {

char rcinput[MXNAME];

FILE *fp;

char *ptr;

dig_server_t *srv;

dig_searchlist_t *search, *domain = NULL;

isc_boolean_t get_servers;

char *input;

debug("setup_system()");

free_now = ISC_FALSE;

get_servers = ISC_TF(server_list.head == NULL);

fp = fopen(RESOLVCONF, "r");

/* XXX Use lwres resolv.conf reader */

if (fp == NULL)

goto no_file;

while (fgets(rcinput, MXNAME, fp) != 0) {

input = rcinput;

ptr = next_token(&input, " \t\r\n");

if (ptr != NULL) {

if (get_servers &&

strcasecmp(ptr, "nameserver") == 0) {

debug("got a nameserver line");

ptr = next_token(&input, " \t\r\n");

if (ptr != NULL) {

srv = make_server(ptr);

ISC_LIST_APPEND(server_list, srv, link);

}

} else if (strcasecmp(ptr, "options") == 0) {

ptr = next_token(&input, " \t\r\n");

if (ptr != NULL) {

if((strncasecmp(ptr, "ndots:",

6) == 0) &&

(ndots == -1)) {

ndots = atoi(

&ptr[6]);

debug("ndots is %d.",

ndots);

}

}

} else if (strcasecmp(ptr, "search") == 0){

while ((ptr = next_token(&input, " \t\r\n"))

!= NULL) {

debug("adding search %s", ptr);

search = make_searchlist_entry(ptr);

ISC_LIST_INITANDAPPEND(search_list,

search, link);

}

} else if (strcasecmp(ptr, "domain") == 0) {

while ((ptr = next_token(&input, " \t\r\n"))

!= NULL) {

if (domain != NULL)

isc_mem_free(mctx, domain);

domain = make_searchlist_entry(ptr);

}

}

}

}

fclose(fp);

no_file:

if (ISC_LIST_EMPTY(search_list) && domain != NULL) {

ISC_LIST_INITANDAPPEND(search_list, domain, link);

domain = NULL;

}

if (domain != NULL)

isc_mem_free(mctx, domain);

if (ndots == -1)

ndots = 1;

if (server_list.head == NULL) {

srv = make_server("127.0.0.1");

ISC_LIST_APPEND(server_list, srv, link);

}

if (keyfile[0] != 0)

setup_file_key();

else if (keysecret[0] != 0)

setup_text_key();

}

static void

clear_searchlist(void) {

dig_searchlist_t *search;

while ((search = ISC_LIST_HEAD(search_list)) != NULL) {

ISC_LIST_UNLINK(search_list, search, link);

isc_mem_free(mctx, search);

}

}

set_search_domain(char *domain) {

dig_searchlist_t *search;

clear_searchlist();

search = make_searchlist_entry(domain);

ISC_LIST_APPEND(search_list, search, link);

}

setup_libs(void) {

isc_result_t result;

debug("setup_libs()");

/*

srandom(getpid() + (int)&setup_libs);

result = isc_net_probeipv4();

if (result == ISC_R_SUCCESS)

have_ipv4 = ISC_TRUE;

result = isc_net_probeipv6();

if (result == ISC_R_SUCCESS)

have_ipv6 = ISC_TRUE;

if (!have_ipv6 && !have_ipv4)

fatal("can't find either v4 or v6 networking");

result = isc_mem_create(0, 0, &mctx);

check_result(result, "isc_mem_create");

result = isc_taskmgr_create(mctx, 1, 0, &taskmgr);

check_result(result, "isc_taskmgr_create");

result = isc_task_create(taskmgr, 0, &global_task);

check_result(result, "isc_task_create");

result = isc_timermgr_create(mctx, &timermgr);

check_result(result, "isc_timermgr_create");

result = isc_socketmgr_create(mctx, &socketmgr);

check_result(result, "isc_socketmgr_create");

result = isc_entropy_create(mctx, &entp);

check_result(result, "isc_entropy_create");

result = dst_lib_init(mctx, entp, 0);

check_result(result, "dst_lib_init");

is_dst_up = ISC_TRUE;

result = isc_mempool_create(mctx, COMMSIZE, &commctx);

check_result(result, "isc_mempool_create");

isc_mempool_setname(commctx, "COMMPOOL");

/*

isc_mempool_setfreemax(commctx, 6);

isc_mempool_setfillcount(commctx, 2);

result = isc_mutex_init(&lookup_lock);

check_result(result, "isc_mutex_init");

dns_result_register();

}

static void

add_opt(dns_message_t *msg, isc_uint16_t udpsize, isc_boolean_t dnssec

#ifdef DNS_OPT_NEWCODES_LIVE

, dns_optlist_t optlist

#endif /* DNS_OPT_NEWCODES_LIVE */

)

{

dns_rdataset_t *rdataset = NULL;

dns_rdatalist_t *rdatalist = NULL;

dns_rdata_t *rdata = NULL;

isc_result_t result;

#ifdef DNS_OPT_NEWCODES_LIVE

isc_buffer_t *rdatabuf = NULL;

unsigned int i, optsize = 0;

#endif /* DNS_OPT_NEWCODES_LIVE */

debug("add_opt()");

result = dns_message_gettemprdataset(msg, &rdataset);

check_result(result, "dns_message_gettemprdataset");

dns_rdataset_init(rdataset);

result = dns_message_gettemprdatalist(msg, &rdatalist);

check_result(result, "dns_message_gettemprdatalist");

result = dns_message_gettemprdata(msg, &rdata);

check_result(result, "dns_message_gettemprdata");

debug("setting udp size of %d", udpsize);

rdatalist->type = dns_rdatatype_opt;

rdatalist->covers = 0;

rdatalist->rdclass = udpsize;

rdatalist->ttl = 0;

if (dnssec)

rdatalist->ttl = DNS_MESSAGEEXTFLAG_DO;

rdata->data = NULL;

rdata->length = 0;

#ifdef DNS_OPT_NEWCODES_LIVE

for (i = 0; i < optlist.used; i++)

optsize += optlist.attrs[i].value.length + 4;

result = isc_buffer_allocate(mctx, &rdatabuf, optsize);

check_result(result, "isc_buffer_allocate");

result = dns_opt_add(rdata, &optlist, rdatabuf);

check_result(result, "dns_opt_add");

dns_message_takebuffer(msg, &rdatabuf);

#endif /* DNS_OPT_NEWCODES_LIVE */

ISC_LIST_INIT(rdatalist->rdata);

ISC_LIST_APPEND(rdatalist->rdata, rdata, link);

dns_rdatalist_tordataset(rdatalist, rdataset);

result = dns_message_setopt(msg, rdataset);

check_result(result, "dns_message_setopt");

}

static void

add_question(dns_message_t *message, dns_name_t *name,

dns_rdataclass_t rdclass, dns_rdatatype_t rdtype)

{

dns_rdataset_t *rdataset;

isc_result_t result;

debug("add_question()");

rdataset = NULL;

result = dns_message_gettemprdataset(message, &rdataset);

check_result(result, "dns_message_gettemprdataset()");

dns_rdataset_init(rdataset);

dns_rdataset_makequestion(rdataset, rdclass, rdtype);

ISC_LIST_APPEND(name->list, rdataset, link);

}

static void

check_if_done(void) {

debug("check_if_done()");

debug("list %s", ISC_LIST_EMPTY(lookup_list) ? "empty" : "full");

if (ISC_LIST_EMPTY(lookup_list) && current_lookup == NULL &&

sendcount == 0) {

INSIST(sockcount == 0);

INSIST(recvcount == 0);

debug("shutting down");

dighost_shutdown();

}

}

static void

clear_query(dig_query_t *query) {

dig_lookup_t *lookup;

REQUIRE(query != NULL);

debug("clear_query(%p)", query);

lookup = query->lookup;

if (lookup->current_query == query)

lookup->current_query = NULL;

ISC_LIST_UNLINK(lookup->q, query, link);

if (ISC_LINK_LINKED(&query->recvbuf, link))

ISC_LIST_DEQUEUE(query->recvlist, &query->recvbuf,

link);

if (ISC_LINK_LINKED(&query->lengthbuf, link))

ISC_LIST_DEQUEUE(query->lengthlist, &query->lengthbuf,

link);

INSIST(query->recvspace != NULL);

if (query->sock != NULL) {

isc_socket_detach(&query->sock);

sockcount--;

debug("sockcount=%d", sockcount);

}

isc_mempool_put(commctx, query->recvspace);

isc_buffer_invalidate(&query->recvbuf);

isc_buffer_invalidate(&query->lengthbuf);

isc_mem_free(mctx, query);

}

static isc_boolean_t

try_clear_lookup(dig_lookup_t *lookup) {

dig_server_t *s;

dig_query_t *q;

void *ptr;

REQUIRE(lookup != NULL);

debug("try_clear_lookup(%p)", lookup);

if (ISC_LIST_HEAD(lookup->q) != NULL) {

if (debugging) {

q = ISC_LIST_HEAD(lookup->q);

while (q != NULL) {

debug("query to %s still pending",

q->servname);

q = ISC_LIST_NEXT(q, link);

}

return (ISC_FALSE);

}

}

/*

debug("cleared");

s = ISC_LIST_HEAD(lookup->my_server_list);

while (s != NULL) {

debug("freeing server %p belonging to %p",

s, lookup);

ptr = s;

s = ISC_LIST_NEXT(s, link);

ISC_LIST_DEQUEUE(lookup->my_server_list,

(dig_server_t *)ptr, link);

isc_mem_free(mctx, ptr);

}

if (lookup->sendmsg != NULL)

dns_message_destroy(&lookup->sendmsg);

if (lookup->querysig != NULL) {

debug("freeing buffer %p", lookup->querysig);

isc_buffer_free(&lookup->querysig);

}

if (lookup->timer != NULL)

isc_timer_detach(&lookup->timer);

if (lookup->sendspace != NULL)

isc_mempool_put(commctx, lookup->sendspace);

isc_mem_free(mctx, lookup);

return (ISC_TRUE);

}

start_lookup(void) {

debug("start_lookup()");

if (cancel_now)

return;

/*

INSIST(current_lookup == NULL);

current_lookup = ISC_LIST_HEAD(lookup_list);

/*

if (current_lookup != NULL) {

ISC_LIST_DEQUEUE(lookup_list, current_lookup, link);

setup_lookup(current_lookup);

do_lookup(current_lookup);

} else {

check_if_done();

}

}

static void

check_next_lookup(dig_lookup_t *lookup) {

INSIST(!free_now);

debug("check_next_lookup(%p)", lookup);

if (ISC_LIST_HEAD(lookup->q) != NULL) {

debug("still have a worker");

return;

}

if (try_clear_lookup(lookup)) {

current_lookup = NULL;

start_lookup();

}

}

static int

followup_lookup(dns_message_t *msg, dig_query_t *query, dns_section_t section)

{

dig_lookup_t *lookup = NULL;

dig_server_t *srv = NULL;

dns_rdataset_t *rdataset = NULL;

dns_rdata_t rdata = DNS_RDATA_INIT;

dns_name_t *name = NULL;

isc_result_t result;

isc_boolean_t success = ISC_FALSE;

int numLookups = 0;

INSIST(!free_now);

debug("followup_lookup()");

result = dns_message_firstname(msg, section);

if (result != ISC_R_SUCCESS) {

debug("firstname returned %s", isc_result_totext(result));

if ((section == DNS_SECTION_ANSWER) &&

(query->lookup->trace || query->lookup->ns_search_only))

numLookups +=

followup_lookup(msg, query, DNS_SECTION_AUTHORITY);

return (numLookups);

}

debug("following up %s", query->lookup->textname);

for (;

result == ISC_R_SUCCESS;

result = dns_message_nextname(msg, section))

{

name = NULL;

dns_message_currentname(msg, section, &name);

for (rdataset = ISC_LIST_HEAD(name->list);

rdataset != NULL;

rdataset = ISC_LIST_NEXT(rdataset, link))

{

debug("got rdataset with type %d", rdataset->type);

if (rdataset->type != dns_rdatatype_ns)

continue;

for (result = dns_rdataset_first(rdataset);

result == ISC_R_SUCCESS;

result = dns_rdataset_next(rdataset))

{

char namestr[DNS_NAME_FORMATSIZE];

dns_rdata_ns_t ns;

if (query->lookup->trace_root &&

query->lookup->nsfound >= MXSERV)

break;

dns_rdataset_current(rdataset, &rdata);

query->lookup->nsfound++;

(void)dns_rdata_tostruct(&rdata, &ns, NULL);

dns_name_format(&ns.name, namestr,

sizeof(namestr));

dns_rdata_freestruct(&ns);

/* Initialize lookup if we've not yet */

debug("found NS %d %s", namestr);

numLookups++;

if (!success) {

success = ISC_TRUE;

lookup_counter++;

cancel_lookup(query->lookup);

lookup = requeue_lookup(query->lookup,

ISC_FALSE);

lookup->doing_xfr = ISC_FALSE;

if (section == DNS_SECTION_ANSWER) {

lookup->trace = ISC_FALSE;

lookup->ns_search_only =

ISC_FALSE;

} else {

lookup->trace =

query->lookup->trace;

lookup->ns_search_only =

query->lookup->ns_search_only;

lookup->ns_search_only_leafnode =

query->lookup->ns_search_only_leafnode;

}

lookup->trace_root = ISC_FALSE;

}

srv = make_server(namestr);

debug("adding server %s", srv->servername);

ISC_LIST_APPEND(lookup->my_server_list,

srv, link);

dns_rdata_reset(&rdata);

}

}

}

if ((lookup == NULL) && (section == DNS_SECTION_ANSWER) &&

(query->lookup->trace || query->lookup->ns_search_only))

numLookups +=

followup_lookup(msg, query, DNS_SECTION_AUTHORITY);

return numLookups;

}

static isc_boolean_t

next_origin(dns_message_t *msg, dig_query_t *query) {

dig_lookup_t *lookup;

UNUSED(msg);

INSIST(!free_now);

debug("next_origin()");

debug("following up %s", query->lookup->textname);

if (!usesearch)

/*

return (ISC_FALSE);

if (query->lookup->origin == NULL)

/*

return (ISC_FALSE);

cancel_lookup(query->lookup);

lookup = requeue_lookup(query->lookup, ISC_TRUE);

lookup->origin = ISC_LIST_NEXT(query->lookup->origin, link);

return (ISC_TRUE);

}

static void

insert_soa(dig_lookup_t *lookup) {

isc_result_t result;

dns_rdata_soa_t soa;

dns_rdata_t *rdata = NULL;

dns_rdatalist_t *rdatalist = NULL;

dns_rdataset_t *rdataset = NULL;

dns_name_t *soaname = NULL;

debug("insert_soa()");

soa.mctx = mctx;

soa.serial = lookup->ixfr_serial;

soa.refresh = 1;

soa.retry = 1;

soa.expire = 1;

soa.minimum = 1;

soa.common.rdclass = lookup->rdclass;

soa.common.rdtype = dns_rdatatype_soa;

dns_name_init(&soa.origin, NULL);

dns_name_init(&soa.mname, NULL);

dns_name_clone(lookup->name, &soa.origin);

dns_name_clone(lookup->name, &soa.mname);

isc_buffer_init(&lookup->rdatabuf, lookup->rdatastore,

sizeof(lookup->rdatastore));

result = dns_message_gettemprdata(lookup->sendmsg, &rdata);

check_result(result, "dns_message_gettemprdata");

result = dns_rdata_fromstruct(rdata, lookup->rdclass,

dns_rdatatype_soa, &soa,

&lookup->rdatabuf);

check_result(result, "isc_rdata_fromstruct");

result = dns_message_gettemprdatalist(lookup->sendmsg, &rdatalist);

check_result(result, "dns_message_gettemprdatalist");

result = dns_message_gettemprdataset(lookup->sendmsg, &rdataset);

check_result(result, "dns_message_gettemprdataset");

dns_rdatalist_init(rdatalist);

rdatalist->type = dns_rdatatype_soa;

rdatalist->rdclass = lookup->rdclass;

rdatalist->covers = dns_rdatatype_soa;

rdatalist->ttl = 1;

ISC_LIST_INIT(rdatalist->rdata);

ISC_LIST_APPEND(rdatalist->rdata, rdata, link);

dns_rdataset_init(rdataset);

dns_rdatalist_tordataset(rdatalist, rdataset);

result = dns_message_gettempname(lookup->sendmsg, &soaname);

check_result(result, "dns_message_gettempname");

dns_name_init(soaname, NULL);

dns_name_clone(lookup->name, soaname);

ISC_LIST_INIT(soaname->list);

ISC_LIST_APPEND(soaname->list, rdataset, link);

dns_message_addname(lookup->sendmsg, soaname, DNS_SECTION_AUTHORITY);

}

setup_lookup(dig_lookup_t *lookup) {

isc_result_t result;

int len;

dig_server_t *serv;

dig_query_t *query;

isc_buffer_t b;

dns_compress_t cctx;

char store[MXNAME];

REQUIRE(lookup != NULL);

INSIST(!free_now);

debug("setup_lookup(%p)", lookup);

result = dns_message_create(mctx, DNS_MESSAGE_INTENTRENDER,

&lookup->sendmsg);

check_result(result, "dns_message_create");

if (lookup->new_search) {

debug("resetting lookup counter.");

lookup_counter = 0;

}

if (ISC_LIST_EMPTY(lookup->my_server_list)) {

debug("cloning server list");

clone_server_list(server_list, &lookup->my_server_list);

}

result = dns_message_gettempname(lookup->sendmsg, &lookup->name);

check_result(result, "dns_message_gettempname");

dns_name_init(lookup->name, NULL);

isc_buffer_init(&lookup->namebuf, lookup->namespace,

sizeof(lookup->namespace));

isc_buffer_init(&lookup->onamebuf, lookup->onamespace,

sizeof(lookup->onamespace));

/*

/* XXX New search here? */

if ((count_dots(lookup->textname) >= ndots) || !usesearch)

lookup->origin = NULL; /* Force abs lookup */

else if (lookup->origin == NULL && lookup->new_search && usesearch) {

lookup->origin = ISC_LIST_HEAD(search_list);

}

if (lookup->origin != NULL) {

debug("trying origin %s", lookup->origin->origin);

result = dns_message_gettempname(lookup->sendmsg,

&lookup->oname);

check_result(result, "dns_message_gettempname");

dns_name_init(lookup->oname, NULL);

/* XXX Helper funct to conv char* to name? */

len = strlen(lookup->origin->origin);

isc_buffer_init(&b, lookup->origin->origin, len);

isc_buffer_add(&b, len);

result = dns_name_fromtext(lookup->oname, &b, dns_rootname,

ISC_FALSE, &lookup->onamebuf);

if (result != ISC_R_SUCCESS) {

dns_message_puttempname(lookup->sendmsg,

&lookup->name);

dns_message_puttempname(lookup->sendmsg,

&lookup->oname);

fatal("'%s' is not in legal name syntax (%s)",

lookup->origin->origin,

isc_result_totext(result));

}

if (lookup->trace_root) {

dns_name_clone(dns_rootname, lookup->name);

} else {

len = strlen(lookup->textname);

isc_buffer_init(&b, lookup->textname, len);

isc_buffer_add(&b, len);

result = dns_name_fromtext(lookup->name, &b,

lookup->oname, ISC_FALSE,

&lookup->namebuf);

}

if (result != ISC_R_SUCCESS) {

dns_message_puttempname(lookup->sendmsg,

&lookup->name);

dns_message_puttempname(lookup->sendmsg,

&lookup->oname);

fatal("'%s' is not in legal name syntax (%s)",

lookup->textname, isc_result_totext(result));

}

dns_message_puttempname(lookup->sendmsg, &lookup->oname);

} else {

debug("using root origin");

if (!lookup->trace_root) {

len = strlen(lookup->textname);

isc_buffer_init(&b, lookup->textname, len);

isc_buffer_add(&b, len);

result = dns_name_fromtext(lookup->name, &b,

dns_rootname,

ISC_FALSE,

&lookup->namebuf);

} else {

dns_name_clone(dns_rootname, lookup->name);

}

if (result != ISC_R_SUCCESS) {

dns_message_puttempname(lookup->sendmsg,

&lookup->name);

isc_buffer_init(&b, store, MXNAME);

fatal("'%s' is not a legal name "

"(%s)", lookup->textname,

isc_result_totext(result));

}

}

dns_name_format(lookup->name, store, sizeof(store));

trying(store, lookup);

INSIST(dns_name_isabsolute(lookup->name));

lookup->sendmsg->id = (unsigned short)(random() & 0xFFFF);

lookup->sendmsg->opcode = dns_opcode_query;

lookup->msgcounter = 0;

/*

if (lookup->recurse && !lookup->trace && !lookup->ns_search_only) {

debug("recursive query");

lookup->sendmsg->flags |= DNS_MESSAGEFLAG_RD;

}

/* XXX aaflag */

if (lookup->aaonly) {

debug("AA query");

lookup->sendmsg->flags |= DNS_MESSAGEFLAG_AA;

}

if (lookup->adflag) {

debug("AD query");

lookup->sendmsg->flags |= DNS_MESSAGEFLAG_AD;

}

if (lookup->cdflag) {

debug("CD query");

lookup->sendmsg->flags |= DNS_MESSAGEFLAG_CD;

}

dns_message_addname(lookup->sendmsg, lookup->name,

DNS_SECTION_QUESTION);

if (lookup->trace_root)

lookup->rdtype = dns_rdatatype_soa;

if ((lookup->rdtype == dns_rdatatype_axfr) ||

(lookup->rdtype == dns_rdatatype_ixfr)) {

lookup->doing_xfr = ISC_TRUE;

/*

lookup->tcp_mode = ISC_TRUE;

}

add_question(lookup->sendmsg, lookup->name, lookup->rdclass,

lookup->rdtype);

/* XXX add_soa */

if (lookup->rdtype == dns_rdatatype_ixfr)

insert_soa(lookup);

/* XXX Insist this? */

lookup->tsigctx = NULL;

lookup->querysig = NULL;

if (key != NULL) {

debug("initializing keys");

result = dns_message_settsigkey(lookup->sendmsg, key);

check_result(result, "dns_message_settsigkey");

}

lookup->sendspace = isc_mempool_get(commctx);

if (lookup->sendspace == NULL)

fatal("memory allocation failure");

result = dns_compress_init(&cctx, -1, mctx);

check_result(result, "dns_compress_init");

debug("starting to render the message");

isc_buffer_init(&lookup->sendbuf, lookup->sendspace, COMMSIZE);

result = dns_message_renderbegin(lookup->sendmsg, &cctx,

&lookup->sendbuf);

check_result(result, "dns_message_renderbegin");

#ifndef DNS_OPT_NEWCODES_LIVE

if (lookup->udpsize > 0 || lookup->dnssec) {

#else /* DNS_OPT_NEWCODES_LIVE */

if (lookup->udpsize > 0 || || lookup->dnssec ||

lookup->zonename[0] != 0 || lookup->viewname[0] != 0) {

dns_fixedname_t fname;

isc_buffer_t namebuf, *wirebuf = NULL;

dns_compress_t zcctx;

dns_optlist_t optlist;

dns_optattr_t optattr[2];

#endif /* DNS_OPT_NEWCODES_LIVE */

if (lookup->udpsize == 0)

lookup->udpsize = 2048;

#ifdef DNS_OPT_NEWCODES_LIVE

optlist.size = 2;

optlist.used = 0;

optlist.next = 0;

optlist.attrs = optattr;

if (lookup->zonename[0] != 0) {

optattr[optlist.used].code = DNS_OPTCODE_ZONE;

dns_fixedname_init(&fname);

isc_buffer_init(&namebuf, lookup->zonename,

strlen(lookup->zonename));

isc_buffer_add(&namebuf, strlen(lookup->zonename));

result = dns_name_fromtext(&(fname.name), &namebuf,

dns_rootname, ISC_FALSE,

NULL);

check_result(result, "; illegal zone option");

result = dns_compress_init(&zcctx, 0, mctx);

check_result(result, "dns_compress_init");

result = isc_buffer_allocate(mctx, &wirebuf,

MXNAME);

check_result(result, "isc_buffer_allocate");

result = dns_name_towire(&(fname.name), &zcctx,

wirebuf);

check_result(result, "dns_name_towire");

optattr[optlist.used].value.base =

isc_buffer_base(wirebuf);

optattr[optlist.used].value.length =

isc_buffer_usedlength(wirebuf);

optlist.used++;

dns_compress_invalidate(&zcctx);

}

if (lookup->viewname[0] != 0) {

optattr[optlist.used].code = DNS_OPTCODE_VIEW;

optattr[optlist.used].value.base =

lookup->viewname;

optattr[optlist.used].value.length =

strlen(lookup->viewname);

optlist.used++;

}

add_opt(lookup->sendmsg, lookup->udpsize, lookup->dnssec,

optlist);

if (wirebuf != NULL)

isc_buffer_free(&wirebuf);

#else /* DNS_OPT_NEWCODES_LIVE */

add_opt(lookup->sendmsg, lookup->udpsize, lookup->dnssec);

#endif /* DNS_OPT_NEWCODES_LIVE */

}

result = dns_message_rendersection(lookup->sendmsg,

DNS_SECTION_QUESTION, 0);

check_result(result, "dns_message_rendersection");

result = dns_message_rendersection(lookup->sendmsg,

DNS_SECTION_AUTHORITY, 0);

check_result(result, "dns_message_rendersection");

result = dns_message_renderend(lookup->sendmsg);

check_result(result, "dns_message_renderend");

debug("done rendering");

dns_compress_invalidate(&cctx);

/*

if (isc_buffer_usedlength(&lookup->sendbuf) > 512)

lookup->tcp_mode = ISC_TRUE;

lookup->pending = ISC_FALSE;

for (serv = ISC_LIST_HEAD(lookup->my_server_list);

serv != NULL;

serv = ISC_LIST_NEXT(serv, link)) {

query = isc_mem_allocate(mctx, sizeof(dig_query_t));

if (query == NULL)

fatal("Memory allocation failure in %s:%d",

__FILE__, __LINE__);

debug("create query %p linked to lookup %p",

query, lookup);

query->lookup = lookup;

query->waiting_connect = ISC_FALSE;

query->recv_made = ISC_FALSE;

query->first_pass = ISC_TRUE;

query->first_soa_rcvd = ISC_FALSE;

query->second_rr_rcvd = ISC_FALSE;

query->first_repeat_rcvd = ISC_FALSE;

query->first_rr_serial = 0;

query->second_rr_serial = 0;

query->servname = serv->servername;

query->rr_count = 0;

ISC_LINK_INIT(query, link);

ISC_LIST_INIT(query->recvlist);

ISC_LIST_INIT(query->lengthlist);

query->sock = NULL;

query->recvspace = isc_mempool_get(commctx);

if (query->recvspace == NULL)

fatal("memory allocation failure");

isc_buffer_init(&query->recvbuf, query->recvspace, COMMSIZE);

isc_buffer_init(&query->lengthbuf, query->lengthspace, 2);

isc_buffer_init(&query->slbuf, query->slspace, 2);

ISC_LINK_INIT(query, link);

ISC_LIST_ENQUEUE(lookup->q, query, link);

}

/* XXX qrflag, print_query, etc... */

if (!ISC_LIST_EMPTY(lookup->q) && qr) {

printmessage(ISC_LIST_HEAD(lookup->q), lookup->sendmsg,

ISC_TRUE);

}

}

static void

send_done(isc_task_t *_task, isc_event_t *event) {

REQUIRE(event->ev_type == ISC_SOCKEVENT_SENDDONE);

UNUSED(_task);

LOCK_LOOKUP;

isc_event_free(&event);

debug("send_done()");

sendcount--;

debug("sendcount=%d", sendcount);

INSIST(sendcount >= 0);

check_if_done();

UNLOCK_LOOKUP;

}

static void

cancel_lookup(dig_lookup_t *lookup) {

dig_query_t *query, *next;

debug("cancel_lookup()");

query = ISC_LIST_HEAD(lookup->q);

while (query != NULL) {

next = ISC_LIST_NEXT(query, link);

if (query->sock != NULL) {

isc_socket_cancel(query->sock, global_task,

ISC_SOCKCANCEL_ALL);

check_if_done();

} else {

clear_query(query);

}

query = next;

}

if (lookup->timer != NULL)

isc_timer_detach(&lookup->timer);

lookup->pending = ISC_FALSE;

lookup->retries = 0;

}

static void

bringup_timer(dig_query_t *query, unsigned int default_timeout) {

dig_lookup_t *l;

unsigned int local_timeout;

isc_result_t result;

debug("bringup_timer()");

/*

l = query->lookup;

if (ISC_LIST_NEXT(query, link) != NULL)

local_timeout = SERVER_TIMEOUT;

else {

if (timeout == 0) {

local_timeout = default_timeout;

} else

local_timeout = timeout;

}

debug("have local timeout of %d", local_timeout);

isc_interval_set(&l->interval, local_timeout, 0);

if (l->timer != NULL)

isc_timer_detach(&l->timer);

result = isc_timer_create(timermgr,

isc_timertype_once,

NULL,

&l->interval,

global_task,

connect_timeout,

l, &l->timer);

check_result(result, "isc_timer_create");

}

static void

connect_done(isc_task_t *task, isc_event_t *event);

static void

send_tcp_connect(dig_query_t *query) {

isc_result_t result;

dig_query_t *next;

dig_lookup_t *l;

debug("send_tcp_connect(%lx)", query);

l = query->lookup;

query->waiting_connect = ISC_TRUE;

query->lookup->current_query = query;

get_address(query->servname, port, &query->sockaddr);

if (specified_source &&

(isc_sockaddr_pf(&query->sockaddr) !=

isc_sockaddr_pf(&bind_address))) {

printf(";; Skipping server %s, incompatible "

"address family\n", query->servname);

query->waiting_connect = ISC_FALSE;

next = ISC_LIST_NEXT(query, link);

l = query->lookup;

clear_query(query);

if (next == NULL) {

printf(";; No acceptable nameservers\n");

check_next_lookup(l);

return;

}

send_tcp_connect(next);

return;

}

INSIST(query->sock == NULL);

result = isc_socket_create(socketmgr,

isc_sockaddr_pf(&query->sockaddr),

isc_sockettype_tcp, &query->sock) ;

check_result(result, "isc_socket_create");

sockcount++;

debug("sockcount=%d", sockcount);

if (specified_source)

result = isc_socket_bind(query->sock, &bind_address);

else {

if ((isc_sockaddr_pf(&query->sockaddr) == AF_INET) &&

have_ipv4)

isc_sockaddr_any(&bind_any);

else

isc_sockaddr_any6(&bind_any);

result = isc_socket_bind(query->sock, &bind_any);

}

check_result(result, "isc_socket_bind");

bringup_timer(query, TCP_TIMEOUT);

result = isc_socket_connect(query->sock, &query->sockaddr,

global_task, connect_done, query);

check_result(result, "isc_socket_connect");

/*

if (l->ns_search_only_leafnode) {

debug("sending next, since searching");

next = ISC_LIST_NEXT(query, link);

if (next != NULL)

send_tcp_connect(next);

}

}

static void

send_udp(dig_query_t *query) {

dig_lookup_t *l = NULL;

dig_query_t *next;

isc_result_t result;

debug("send_udp(%lx)", query);

l = query->lookup;

bringup_timer(query, UDP_TIMEOUT);

l->current_query = query;

debug("working on lookup %p, query %p",

query->lookup, query);

if (!query->recv_made) {

/* XXX Check the sense of this, need assertion? */

query->waiting_connect = ISC_FALSE;

get_address(query->servname, port, &query->sockaddr);

result = isc_socket_create(socketmgr,

isc_sockaddr_pf(&query->sockaddr),

isc_sockettype_udp, &query->sock);

check_result(result, "isc_socket_create");

sockcount++;

debug("sockcount=%d", sockcount);

if (specified_source) {

result = isc_socket_bind(query->sock, &bind_address);

} else {

isc_sockaddr_anyofpf(&bind_any,

isc_sockaddr_pf(&query->sockaddr));

result = isc_socket_bind(query->sock, &bind_any);

}

check_result(result, "isc_socket_bind");

query->recv_made = ISC_TRUE;

ISC_LINK_INIT(&query->recvbuf, link);

ISC_LIST_ENQUEUE(query->recvlist, &query->recvbuf,

link);

debug("recving with lookup=%p, query=%p, sock=%p",

query->lookup, query,

query->sock);

result = isc_socket_recvv(query->sock,

&query->recvlist, 1,

global_task, recv_done,

query);

check_result(result, "isc_socket_recvv");

recvcount++;

debug("recvcount=%d", recvcount);

}

ISC_LIST_INIT(query->sendlist);

ISC_LINK_INIT(&l->sendbuf, link);

ISC_LIST_ENQUEUE(query->sendlist, &l->sendbuf,

link);

debug("sending a request");

result = isc_time_now(&query->time_sent);

check_result(result, "isc_time_now");

INSIST(query->sock != NULL);

result = isc_socket_sendtov(query->sock, &query->sendlist,

global_task, send_done, query,

&query->sockaddr, NULL);

check_result(result, "isc_socket_sendtov");

sendcount++;

/*

if (l->ns_search_only_leafnode) {

debug("sending next, since searching");

next = ISC_LIST_NEXT(query, link);

if (next != NULL)

send_udp(next);

}

}

static void

connect_timeout(isc_task_t *task, isc_event_t *event) {

dig_lookup_t *l=NULL;

dig_query_t *query=NULL, *cq;

UNUSED(task);

REQUIRE(event->ev_type == ISC_TIMEREVENT_IDLE);

debug("connect_timeout()");

LOCK_LOOKUP;

l = event->ev_arg;

query = l->current_query;

isc_event_free(&event);

INSIST(!free_now);

if ((query != NULL) && (query->lookup->current_query != NULL) &&

(ISC_LIST_NEXT(query->lookup->current_query, link) != NULL)) {

debug("trying next server...");

cq = query->lookup->current_query;

if (!l->tcp_mode)

send_udp(ISC_LIST_NEXT(cq, link));

else

send_tcp_connect(ISC_LIST_NEXT(cq, link));

UNLOCK_LOOKUP;

return;

}

if (l->retries > 1) {

if (!l->tcp_mode) {

l->retries--;

debug("resending UDP request to first server");

send_udp(ISC_LIST_HEAD(l->q));

} else {

debug("making new TCP request, %d tries left",

l->retries);

cancel_lookup(l);

l->retries--;

requeue_lookup(l, ISC_TRUE);

}

} else {

fputs(l->cmdline, stdout);

printf(";; connection timed out; no servers could be "

"reached\n");

cancel_lookup(l);

if (exitcode < 9)

exitcode = 9;

}

UNLOCK_LOOKUP;

}

static void

tcp_length_done(isc_task_t *task, isc_event_t *event) {

isc_socketevent_t *sevent;

isc_buffer_t *b = NULL;

isc_result_t result;

dig_query_t *query = NULL;

dig_lookup_t *l;

isc_uint16_t length;

REQUIRE(event->ev_type == ISC_SOCKEVENT_RECVDONE);

INSIST(!free_now);

UNUSED(task);

debug("tcp_length_done()");

LOCK_LOOKUP;

sevent = (isc_socketevent_t *)event;

query = event->ev_arg;

recvcount--;

INSIST(recvcount >= 0);

if (sevent->result == ISC_R_CANCELED) {

isc_event_free(&event);

l = query->lookup;

clear_query(query);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

if (sevent->result != ISC_R_SUCCESS) {

char sockstr[ISC_SOCKADDR_FORMATSIZE];

isc_sockaddr_format(&query->sockaddr, sockstr,

sizeof(sockstr));

printf(";; communications error to %s: %s\n",

sockstr, isc_result_totext(sevent->result));

l = query->lookup;

isc_socket_detach(&query->sock);

sockcount--;

debug("sockcount=%d", sockcount);

INSIST(sockcount >= 0);

isc_event_free(&event);

clear_query(query);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

b = ISC_LIST_HEAD(sevent->bufferlist);

ISC_LIST_DEQUEUE(sevent->bufferlist, &query->lengthbuf, link);

length = isc_buffer_getuint16(b);

if (length == 0) {

isc_event_free(&event);

launch_next_query(query, ISC_FALSE);

UNLOCK_LOOKUP;

return;

}

/*

isc_buffer_invalidate(&query->recvbuf);

isc_buffer_init(&query->recvbuf, query->recvspace, length);

ENSURE(ISC_LIST_EMPTY(query->recvlist));

ISC_LINK_INIT(&query->recvbuf, link);

ISC_LIST_ENQUEUE(query->recvlist, &query->recvbuf, link);

debug("recving with lookup=%p, query=%p",

query->lookup, query);

result = isc_socket_recvv(query->sock, &query->recvlist, length, task,

recv_done, query);

check_result(result, "isc_socket_recvv");

recvcount++;

debug("resubmitted recv request with length %d, recvcount=%d",

length, recvcount);

isc_event_free(&event);

UNLOCK_LOOKUP;

}

static void

launch_next_query(dig_query_t *query, isc_boolean_t include_question) {

isc_result_t result;

dig_lookup_t *l;

INSIST(!free_now);

debug("launch_next_query()");

if (!query->lookup->pending) {

debug("ignoring launch_next_query because !pending");

isc_socket_detach(&query->sock);

sockcount--;

debug("sockcount=%d", sockcount);

INSIST(sockcount >= 0);

query->waiting_connect = ISC_FALSE;

l = query->lookup;

clear_query(query);

check_next_lookup(l);

return;

}

isc_buffer_clear(&query->slbuf);

isc_buffer_clear(&query->lengthbuf);

isc_buffer_putuint16(&query->slbuf, (isc_uint16_t) query->lookup->sendbuf.used);

ISC_LIST_INIT(query->sendlist);

ISC_LINK_INIT(&query->slbuf, link);

ISC_LIST_ENQUEUE(query->sendlist, &query->slbuf, link);

if (include_question) {

ISC_LINK_INIT(&query->lookup->sendbuf, link);

ISC_LIST_ENQUEUE(query->sendlist, &query->lookup->sendbuf,

link);

}

ISC_LINK_INIT(&query->lengthbuf, link);

ISC_LIST_ENQUEUE(query->lengthlist, &query->lengthbuf, link);

result = isc_socket_recvv(query->sock, &query->lengthlist, 0,

global_task, tcp_length_done, query);

check_result(result, "isc_socket_recvv");

recvcount++;

debug("recvcount=%d",recvcount);

if (!query->first_soa_rcvd) {

debug("sending a request in launch_next_query");

result = isc_time_now(&query->time_sent);

check_result(result, "isc_time_now");

result = isc_socket_sendv(query->sock, &query->sendlist,

global_task, send_done, query);

check_result(result, "isc_socket_sendv");

sendcount++;

debug("sendcount=%d", sendcount);

}

query->waiting_connect = ISC_FALSE;

#if 0

check_next_lookup(query->lookup);

#endif

return;

}

static void

connect_done(isc_task_t *task, isc_event_t *event) {

isc_socketevent_t *sevent = NULL;

dig_query_t *query = NULL, *next;

dig_lookup_t *l;

UNUSED(task);

REQUIRE(event->ev_type == ISC_SOCKEVENT_CONNECT);

INSIST(!free_now);

debug("connect_done()");

LOCK_LOOKUP;

sevent = (isc_socketevent_t *)event;

query = sevent->ev_arg;

INSIST(query->waiting_connect);

query->waiting_connect = ISC_FALSE;

if (sevent->result == ISC_R_CANCELED) {

debug("in cancel handler");

isc_socket_detach(&query->sock);

sockcount--;

INSIST(sockcount >= 0);

debug("sockcount=%d", sockcount);

query->waiting_connect = ISC_FALSE;

isc_event_free(&event);

l = query->lookup;

clear_query(query);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

if (sevent->result != ISC_R_SUCCESS) {

char sockstr[ISC_SOCKADDR_FORMATSIZE];

debug("unsuccessful connection: %s",

isc_result_totext(sevent->result));

isc_sockaddr_format(&query->sockaddr, sockstr,

sizeof(sockstr));

if (sevent->result != ISC_R_CANCELED)

printf(";; Connection to %s(%s) for %s failed: "

"%s.\n", sockstr,

query->servname, query->lookup->textname,

isc_result_totext(sevent->result));

isc_socket_detach(&query->sock);

sockcount--;

INSIST(sockcount >= 0);

/* XXX Clean up exitcodes */

if (exitcode < 9)

exitcode = 9;

debug("sockcount=%d", sockcount);

query->waiting_connect = ISC_FALSE;

isc_event_free(&event);

l = query->lookup;

if (l->current_query != NULL)

next = ISC_LIST_NEXT(l->current_query, link);

else

next = NULL;

clear_query(query);

if (next != NULL) {

bringup_timer(next, TCP_TIMEOUT);

send_tcp_connect(next);

} else {

check_next_lookup(l);

}

UNLOCK_LOOKUP;

return;

}

launch_next_query(query, ISC_TRUE);

isc_event_free(&event);

UNLOCK_LOOKUP;

}

static isc_boolean_t

check_for_more_data(dig_query_t *query, dns_message_t *msg,

isc_socketevent_t *sevent)

{

dns_rdataset_t *rdataset = NULL;

dns_rdata_t rdata = DNS_RDATA_INIT;

dns_rdata_soa_t soa;

isc_uint32_t serial;

isc_result_t result;

debug("check_for_more_data()");

/*

result = dns_message_firstname(msg, DNS_SECTION_ANSWER);

if (result != ISC_R_SUCCESS) {

puts("; Transfer failed.");

return (ISC_TRUE);

}

do {

dns_name_t *name;

name = NULL;

dns_message_currentname(msg, DNS_SECTION_ANSWER,

&name);

for (rdataset = ISC_LIST_HEAD(name->list);

rdataset != NULL;

rdataset = ISC_LIST_NEXT(rdataset, link)) {

result = dns_rdataset_first(rdataset);

if (result != ISC_R_SUCCESS)

continue;

do {

query->rr_count++;

dns_rdata_reset(&rdata);

dns_rdataset_current(rdataset, &rdata);

/*

if ((!query->first_soa_rcvd) &&

(rdata.type != dns_rdatatype_soa)) {

puts("; Transfer failed. "

"Didn't start with "

"SOA answer.");

return (ISC_TRUE);

}

if ((!query->second_rr_rcvd) &&

(rdata.type != dns_rdatatype_soa)) {

query->second_rr_rcvd = ISC_TRUE;

query->second_rr_serial = 0;

debug("got the second rr as nonsoa");

continue;

}

/*

if (rdata.type != dns_rdatatype_soa)

goto next_rdata;

/* Now we have an SOA. Work with it. */

debug("got an SOA");

(void)dns_rdata_tostruct(&rdata, &soa, NULL);

serial = soa.serial;

dns_rdata_freestruct(&soa);

if (!query->first_soa_rcvd) {

query->first_soa_rcvd = ISC_TRUE;

query->first_rr_serial = serial;

debug("this is the first %d",

query->lookup->ixfr_serial);

if (query->lookup->ixfr_serial >=

serial)

goto doexit;

goto next_rdata;

}

if (query->lookup->rdtype ==

dns_rdatatype_axfr) {

debug("doing axfr, got second SOA");

goto doexit;

}

if (!query->second_rr_rcvd) {

if (query->first_rr_serial == serial) {

debug("doing ixfr, got "

"empty zone");

goto doexit;

}

debug("this is the second %d",

query->lookup->ixfr_serial);

query->second_rr_rcvd = ISC_TRUE;

query->second_rr_serial = serial;

goto next_rdata;

}

if (query->second_rr_serial == 0) {

/*

debug("done, since axfr");

goto doexit;

}

/*

if (query->first_rr_serial == serial) {

debug("got a match for ixfr");

if (!query->first_repeat_rcvd) {

query->first_repeat_rcvd =

ISC_TRUE;

goto next_rdata;

}

debug("done with ixfr");

goto doexit;

}

debug("meaningless soa %d", serial);

next_rdata:

result = dns_rdataset_next(rdataset);

} while (result == ISC_R_SUCCESS);

}

result = dns_message_nextname(msg, DNS_SECTION_ANSWER);

} while (result == ISC_R_SUCCESS);

launch_next_query(query, ISC_FALSE);

return (ISC_FALSE);

doexit:

received(sevent->n, &sevent->address, query);

return (ISC_TRUE);

}

static void

recv_done(isc_task_t *task, isc_event_t *event) {

isc_socketevent_t *sevent = NULL;

dig_query_t *query = NULL;

isc_buffer_t *b = NULL;

dns_message_t *msg = NULL;

isc_result_t result;

dig_lookup_t *n, *l;

isc_boolean_t docancel = ISC_FALSE;

unsigned int local_timeout;

UNUSED(task);

INSIST(!free_now);

debug("recv_done()");

LOCK_LOOKUP;

recvcount--;

debug("recvcount=%d", recvcount);

INSIST(recvcount >= 0);

query = event->ev_arg;

debug("lookup=%p, query=%p", query->lookup, query);

l = query->lookup;

REQUIRE(event->ev_type == ISC_SOCKEVENT_RECVDONE);

sevent = (isc_socketevent_t *)event;

if ((l->tcp_mode) && (l->timer != NULL))

isc_timer_touch(l->timer);

if ((!l->pending && !l->ns_search_only && !l->ns_search_only_leafnode)

|| cancel_now) {

debug("no longer pending. Got %s",

isc_result_totext(sevent->result));

query->waiting_connect = ISC_FALSE;

isc_event_free(&event);

clear_query(query);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

if (sevent->result == ISC_R_SUCCESS) {

b = ISC_LIST_HEAD(sevent->bufferlist);

ISC_LIST_DEQUEUE(sevent->bufferlist, &query->recvbuf, link);

result = dns_message_create(mctx, DNS_MESSAGE_INTENTPARSE,

&msg);

check_result(result, "dns_message_create");

if (key != NULL) {

if (l->querysig == NULL) {

debug("getting initial querysig");

result = dns_message_getquerytsig(

l->sendmsg,

mctx, &l->querysig);

check_result(result,

"dns_message_getquerytsig");

}

result = dns_message_setquerytsig(msg,

l->querysig);

check_result(result, "dns_message_setquerytsig");

result = dns_message_settsigkey(msg, key);

check_result(result, "dns_message_settsigkey");

msg->tsigctx = l->tsigctx;

if (l->msgcounter != 0)

msg->tcp_continuation = 1;

l->msgcounter++;

}

debug("before parse starts");

if (l->besteffort)

result = dns_message_parse(msg, b,

DNS_MESSAGEPARSE_PRESERVEORDER

|DNS_MESSAGEPARSE_BESTEFFORT);

else

result = dns_message_parse(msg, b,

DNS_MESSAGEPARSE_PRESERVEORDER);

if (result != ISC_R_SUCCESS &&

result != DNS_R_RECOVERABLE ) {

printf(";; Got bad packet: %s\n",

isc_result_totext(result));

hex_dump(b);

query->waiting_connect = ISC_FALSE;

dns_message_destroy(&msg);

isc_event_free(&event);

clear_query(query);

cancel_lookup(l);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

if (result == DNS_R_RECOVERABLE)

printf(";; Warning: Message parser reports malformed "

"message packet.\n");

if (((msg->flags & DNS_MESSAGEFLAG_TC) != 0)

&& ! l->ignore && !l->tcp_mode) {

printf(";; Truncated, retrying in TCP mode.\n");

n = requeue_lookup(l, ISC_TRUE);

n->tcp_mode = ISC_TRUE;

dns_message_destroy(&msg);

isc_event_free(&event);

clear_query(query);

cancel_lookup(l);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

if ((msg->rcode == dns_rcode_servfail) &&

l->servfail_stops) {

dig_query_t *next = ISC_LIST_NEXT(query, link);

if (l->current_query == query)

l->current_query = NULL;

if (next != NULL) {

debug("sending query %lx\n", next);

if (l->tcp_mode)

send_tcp_connect(next);

else

send_udp(next);

}

/*

if ((ISC_LIST_HEAD(l->q) != query) ||

(ISC_LIST_NEXT(query, link) != NULL)) {

printf(";; Got SERVFAIL reply from %s, "

"trying next server\n",

query->servname);

clear_query(query);

check_next_lookup(l);

dns_message_destroy(&msg);

isc_event_free(&event);

UNLOCK_LOOKUP;

return;

}

}

if (key != NULL) {

result = dns_tsig_verify(&query->recvbuf, msg,

NULL, NULL);

if (result != ISC_R_SUCCESS) {

printf(";; Couldn't verify signature: %s\n",

isc_result_totext(result));

validated = ISC_FALSE;

}

l->tsigctx = msg->tsigctx;

if (l->querysig != NULL) {

debug("freeing querysig buffer %p",

l->querysig);

isc_buffer_free(&l->querysig);

}

result = dns_message_getquerytsig(msg, mctx,

&l->querysig);

check_result(result,"dns_message_getquerytsig");

debug("querysig 3 is %p", l->querysig);

}

debug("after parse");

if (l->xfr_q == NULL) {

l->xfr_q = query;

/*

if ((timeout != INT_MAX) &&

(l->timer != NULL) &&

l->doing_xfr ) {

if (timeout == 0) {

if (l->tcp_mode)

local_timeout = TCP_TIMEOUT;

else

local_timeout = UDP_TIMEOUT;

} else {

if (timeout < (INT_MAX / 4))

local_timeout = timeout * 4;

else

local_timeout = INT_MAX;

}

debug("have local timeout of %d",

local_timeout);

isc_interval_set(&l->interval,

local_timeout, 0);

result = isc_timer_reset(l->timer,

isc_timertype_once,

NULL,

&l->interval,

ISC_FALSE);

check_result(result, "isc_timer_reset");

}

}

if (l->xfr_q == query) {

if ((l->trace)||

(l->ns_search_only)) {

debug("in TRACE code");

printmessage(query, msg, ISC_TRUE);

if ((msg->rcode != 0) &&

(l->origin != NULL)) {

if (!next_origin(msg, query)) {

printmessage(query, msg,

ISC_TRUE);

received(b->used,

&sevent->address,

query);

}

} else {

result = dns_message_firstname

(msg,DNS_SECTION_ANSWER);

if (l->ns_search_only)

{

if ((result != ISC_R_SUCCESS) || l->trace_root)

{

/*

l->rdtype = dns_rdatatype_ns;

} else {

/*

l->rdtype = dns_rdatatype_soa;

l->ns_search_only_leafnode = ISC_TRUE;

if (followup_lookup(msg, query,

DNS_SECTION_ANSWER) == 0)

{

docancel = ISC_TRUE;

}

}

}

if ((result != ISC_R_SUCCESS) ||

l->trace_root)

{

/*

if (followup_lookup(msg, query,

DNS_SECTION_AUTHORITY)

== 0)

{

docancel = ISC_TRUE;

}

}

}

} else if ((msg->rcode != 0) &&

(l->origin != NULL)) {

if (!next_origin(msg, query)) {

printmessage(query, msg,

ISC_TRUE);

received(b->used,

&sevent->address,

query);

}

} else {

printmessage(query, msg, ISC_TRUE);

}

} else if ((dns_message_firstname(msg, DNS_SECTION_ANSWER)

== ISC_R_SUCCESS) &&

(l->ns_search_only || l->ns_search_only_leafnode) &&

!l->trace_root ) {

printmessage(query, msg, ISC_TRUE);

}

if (l->pending)

debug("still pending.");

if (l->doing_xfr) {

if (query != l->xfr_q) {

dns_message_destroy(&msg);

isc_event_free (&event);

query->waiting_connect = ISC_FALSE;

UNLOCK_LOOKUP;

return;

}

if (! docancel)

docancel = check_for_more_data(query, msg, sevent);

if (docancel) {

dns_message_destroy(&msg);

clear_query(query);

cancel_lookup(l);

check_next_lookup(l);

}

if (msg != NULL)

dns_message_destroy(&msg);

isc_event_free(&event);

} else {

if ((msg->rcode == 0) ||

(l->origin == NULL)) {

received(b->used,

&sevent->address,

query);

}

if (!(query->lookup->ns_search_only ||

query->lookup->ns_search_only_leafnode))

query->lookup->pending = ISC_FALSE;

if (!(query->lookup->ns_search_only ||

query->lookup->ns_search_only_leafnode) ||

query->lookup->trace_root || docancel) {

dns_message_destroy(&msg);

cancel_lookup(l);

}

if (msg != NULL)

dns_message_destroy(&msg);

isc_event_free(&event);

clear_query(query);

check_next_lookup(l);

}

UNLOCK_LOOKUP;

return;

}

/*

if (sevent->result == ISC_R_CANCELED) {

debug("in recv cancel handler");

query->waiting_connect = ISC_FALSE;

isc_event_free(&event);

clear_query(query);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

printf(";; communications error: %s\n",

isc_result_totext(sevent->result));

isc_socket_detach(&query->sock);

sockcount--;

debug("sockcount=%d", sockcount);

INSIST(sockcount >= 0);

isc_event_free(&event);

clear_query(query);

check_next_lookup(l);

UNLOCK_LOOKUP;

return;

}

get_address(char *host, in_port_t port, isc_sockaddr_t *sockaddr) {

struct in_addr in4;

struct in6_addr in6;

#ifdef USE_GETADDRINFO

struct addrinfo *res = NULL, hints;

int result;

#else

struct hostent *he;

#endif

debug("get_address()");