ldns-host.c

/*-
 * (c) Magerya Vitaly
 *
 * Copying and distribution of this file, with or without modification,
 * are permitted in any medium without royalty provided the copyright
 * notice and this notice are preserved. This file is offered as-is,
 * without any warranty.
 */

#include <ldns/ldns.h>
#include <limits.h>
#include <netdb.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

/* General utilities.
 */

static char *progname;

#define countof(array) (sizeof(array)/sizeof(*(array)))

static void
die(int code, const char *fmt, ...) {
    va_list args;

    va_start(args, fmt);
    fprintf(stderr, "%s: ", progname);
    vfprintf(stderr, fmt, args);
    fprintf(stderr, "\n");
    va_end(args);
    exit(code);
}

static int
ndots(const char *name) {
    int n;

    for (n = 0; (name = strchr(name, '.')); n++, name++);
    return n;
}

/* General LDNS-specific utilities.
 */

static ldns_status
ldns_resolver_new_default(ldns_resolver **res) {
    if (ldns_resolver_new_frm_file(res, NULL) == LDNS_STATUS_OK ||
        (*res = ldns_resolver_new()) != NULL)
        return LDNS_STATUS_OK;
    return LDNS_STATUS_MEM_ERR;
}

static ldns_status
ldns_resolver_push_default_servers(ldns_resolver *res) {
    ldns_status status;
    ldns_rdf *addr;

    if ((status = ldns_str2rdf_a(&addr, "127.0.0.1")) != LDNS_STATUS_OK ||
        (status = ldns_resolver_push_nameserver(res, addr)) != LDNS_STATUS_OK)
        return ldns_rdf_deep_free(addr), status;
    ldns_rdf_deep_free(addr);
    if ((status = ldns_str2rdf_aaaa(&addr, "::1")) != LDNS_STATUS_OK ||
        (status = ldns_resolver_push_nameserver(res, addr)) != LDNS_STATUS_OK)
        return ldns_rdf_deep_free(addr), status;
    ldns_rdf_deep_free(addr);
    return LDNS_STATUS_OK;
}

static ldns_rdf *
ldns_rdf_new_addr_frm_str(const char *str) {
    ldns_rdf *addr;

    if ((addr = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_A, str)) == NULL)
        addr = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_AAAA, str);
    return addr;
}

static void
ldns_resolver_remove_nameservers(ldns_resolver *res) {
    while (ldns_resolver_nameserver_count(res) > 0)
        ldns_rdf_deep_free(ldns_resolver_pop_nameserver(res));
}

static ldns_rdf *
ldns_rdf_reverse_a(ldns_rdf *addr, const char *base) {
    char *buf;
    int i, len;

    len = strlen(base);
    buf = alloca(LDNS_IP4ADDRLEN*4 + len + 1);
    for (len = i = 0; i < LDNS_IP4ADDRLEN; i++)
        len += sprintf(&buf[len], "%d.",
            (int)ldns_rdf_data(addr)[LDNS_IP4ADDRLEN - i - 1]);
    sprintf(&buf[len], "%s", base);
    return ldns_dname_new_frm_str(buf);
}

static ldns_rdf *
ldns_rdf_reverse_aaaa(ldns_rdf *addr, const char *base) {
    char *buf;
    int i, len;

    len = strlen(base);
    buf = alloca(LDNS_IP6ADDRLEN*4 + len + 1);
    for (i = 0; i < LDNS_IP6ADDRLEN; i++) {
        uint8_t byte = ldns_rdf_data(addr)[LDNS_IP6ADDRLEN - i - 1];
        sprintf(&buf[i*4], "%x.%x.", byte & 0x0F, byte >> 4);
    }
    sprintf(&buf[LDNS_IP6ADDRLEN*4], "%s", base);
    return ldns_dname_new_frm_str(buf);
}

static ldns_status
ldns_pkt_push_rr_soa(ldns_pkt *pkt, ldns_pkt_section sec,
    const ldns_rdf *name, ldns_rr_class c, uint32_t serial) {
    ldns_rdf *rdf;
    ldns_rr *rr;
    uint32_t n;

    if ((rr = ldns_rr_new_frm_type(LDNS_RR_TYPE_SOA)) == NULL)
        return LDNS_STATUS_MEM_ERR;
    ldns_rr_set_class(rr, c);
    ldns_rr_set_owner(rr, ldns_rdf_clone(name));
    ldns_rr_set_ttl(rr, 0);

    n = 0;
    if ((rdf = ldns_rdf_new_frm_data(LDNS_RDF_TYPE_DNAME, 1, &n)) == NULL)
        goto memerr;
    ldns_rr_set_rdf(rr, rdf, 0);
    ldns_rr_set_rdf(rr, ldns_rdf_clone(rdf), 1);

    n = htonl(serial);
    if ((rdf = ldns_rdf_new_frm_data(LDNS_RDF_TYPE_INT32, 4, &n)) == NULL)
        goto memerr;
    ldns_rr_set_rdf(rr, rdf, 2);

    n = 0;
    if ((rdf = ldns_rdf_new_frm_data(LDNS_RDF_TYPE_PERIOD, 4, &n)) == NULL)
        goto memerr;
    ldns_rr_set_rdf(rr, rdf, 3);
    ldns_rr_set_rdf(rr, ldns_rdf_clone(rdf), 4);
    ldns_rr_set_rdf(rr, ldns_rdf_clone(rdf), 5);
    ldns_rr_set_rdf(rr, ldns_rdf_clone(rdf), 6);

    if (ldns_rr_rdf(rr, 1) == NULL || ldns_rr_rdf(rr, 4) == NULL ||
        ldns_rr_rdf(rr, 5) == NULL || ldns_rr_rdf(rr, 6) == NULL ||
        !ldns_pkt_push_rr(pkt, sec, rr))
        goto memerr;
    return LDNS_STATUS_OK;

memerr:
    ldns_rr_free(rr);
    return LDNS_STATUS_MEM_ERR;
}

static uint32_t
ldns_rr_soa_get_serial(const ldns_rr *rr)
{
    const ldns_rdf *rdf;
  
    if (ldns_rr_get_type(rr) != LDNS_RR_TYPE_SOA) return 0;
    if (ldns_rr_rd_count(rr) != 7) return 0;
    rdf = ldns_rr_rdf(rr, 2);
    if (ldns_rdf_get_type(rdf) != LDNS_RDF_TYPE_INT32) return 0;
    if (ldns_rdf_size(rdf) != 4) return 0;
    return ldns_rdf2native_int32(rdf);
}

static ldns_status
ldns_tcp_start(ldns_resolver *res, ldns_pkt *qpkt, int nameserver) {
    /* This routine is based on ldns_axfr_start, with the major
     * difference in that it takes a query packet explicitly.
     */
    struct sockaddr_storage *ns = NULL;
    size_t ns_len = 0;
    ldns_buffer *qbuf = NULL;
    ldns_status status;

    ns = ldns_rdf2native_sockaddr_storage(
            res->_nameservers[nameserver], ldns_resolver_port(res), &ns_len);
    if (ns == NULL) {
        status = LDNS_STATUS_MEM_ERR;
        goto error;
    }

    res->_socket = ldns_tcp_connect(
            ns, (socklen_t)ns_len, ldns_resolver_timeout(res));
    if (res->_socket <= 0) {
        status = LDNS_STATUS_ADDRESS_ERR;
        goto error;
    }

    qbuf = ldns_buffer_new(LDNS_MAX_PACKETLEN);
    if (qbuf == NULL) {
        status = LDNS_STATUS_MEM_ERR;
        goto error;
    }

    status = ldns_pkt2buffer_wire(qbuf, qpkt);
    if (status != LDNS_STATUS_OK)
        goto error;

    if (ldns_tcp_send_query(qbuf, res->_socket, ns, (socklen_t)ns_len) == 0) {
        status = LDNS_STATUS_NETWORK_ERR;
        goto error;
    }

    ldns_buffer_free(qbuf);
    free(ns);
    return LDNS_STATUS_OK;
 
error:
    ldns_buffer_free(qbuf);
    free(ns);
    if (res->_socket > 0) {
        close(res->_socket);
        res->_socket = 0;
    }
    return status;
}

static ldns_status
ldns_tcp_read(ldns_pkt **answer, ldns_resolver *res) {
    ldns_status status;
    struct timeval t1, t2;
    uint8_t *data;
    size_t size;

    if (res->_socket <= 0)
        return LDNS_STATUS_ERR;

    gettimeofday(&t1, NULL);
    data = ldns_tcp_read_wire_timeout(
            res->_socket, &size, ldns_resolver_timeout(res));
    if (data == NULL)
        goto error;

    status = ldns_wire2pkt(answer, data, size);
    free(data);
    if (status != LDNS_STATUS_OK)
        goto error;

    gettimeofday(&t2, NULL);
    ldns_pkt_set_querytime(*answer,
            (uint32_t)((t2.tv_sec - t1.tv_sec)*1000) +
                (t2.tv_usec - t1.tv_usec)/1000);
    ldns_pkt_set_timestamp(*answer, t2);
    return status;

error:
    close(res->_socket);
    res->_socket = 0;
    return LDNS_STATUS_ERR;
}

static void
ldns_tcp_close(ldns_resolver *res) {
    if (res->_socket > 0) {
        close(res->_socket);
        res->_socket = 0;
    }
}

static ldns_status
ldns_resolver_send_to(ldns_pkt **answer, ldns_resolver *res,
    const ldns_rdf *name, ldns_rr_type t, ldns_rr_class c,
    uint16_t flags, uint32_t ixfr_serial, int nameserver,
    bool close_tcp) {
    ldns_status status = LDNS_STATUS_OK;
    ldns_pkt *qpkt;
    struct timeval now;

    int nscnt = ldns_resolver_nameserver_count(res);
    ldns_rdf **ns = ldns_resolver_nameservers(res);
    size_t *rtt = ldns_resolver_rtt(res);

    ldns_resolver_set_nameservers(res, &ns[nameserver]);
    ldns_resolver_set_rtt(res, &rtt[nameserver]);
    ldns_resolver_set_nameserver_count(res, 1);

    /* The next fragment should have been a call to
     * ldns_resolver_prepare_query_pkt(), but starting with ldns
     * version 1.6.17 that function tries to add it's own SOA
     * records when rr_type is LDNS_RR_TYPE_IXFR, and we don't
     * want that.
     */
    qpkt = ldns_pkt_query_new(ldns_rdf_clone(name), t, c, flags);
    if (qpkt == NULL) {
        status = LDNS_STATUS_ERR;
        goto done;
    }
    now.tv_sec = time(NULL);
    now.tv_usec = 0;
    ldns_pkt_set_timestamp(qpkt, now);
    ldns_pkt_set_random_id(qpkt);

    if (t == LDNS_RR_TYPE_IXFR) {
        status = ldns_pkt_push_rr_soa(qpkt,
            LDNS_SECTION_AUTHORITY, name, c, ixfr_serial);
        if (status != LDNS_STATUS_OK) goto done;
    }
    if (close_tcp) {
        status = ldns_resolver_send_pkt(answer, res, qpkt);
    } else {
        status = ldns_tcp_start(res, qpkt, 0);
        if (status != LDNS_STATUS_OK) goto done;
        status = ldns_tcp_read(answer, res);
        if (status != LDNS_STATUS_OK) goto done;
        ldns_pkt_set_answerfrom(*answer, ldns_rdf_clone(ns[0]));
    }

done:
    ldns_pkt_free(qpkt);

    ldns_resolver_set_nameservers(res, ns);
    ldns_resolver_set_rtt(res, rtt);
    ldns_resolver_set_nameserver_count(res, nscnt);
    return status;
}

static void
ldns_pkt_filter_answer(ldns_pkt *pkt, ldns_rr_type type) {
    int i, j, cnt;
    ldns_rr_list *rrlist;
    ldns_rr *rr;
    ldns_rr_type rrtype;

    rrlist = ldns_pkt_answer(pkt);
    cnt = ldns_rr_list_rr_count(rrlist);
    for (i = j = 0; i < cnt; i++) {
        rr = ldns_rr_list_rr(rrlist, i);
        rrtype = ldns_rr_get_type(rr);
        if (type == LDNS_RR_TYPE_ANY ||
            type == rrtype ||
            (type == LDNS_RR_TYPE_AXFR &&
                (rrtype == LDNS_RR_TYPE_A ||
                rrtype == LDNS_RR_TYPE_AAAA ||
                rrtype == LDNS_RR_TYPE_NS ||
                rrtype == LDNS_RR_TYPE_PTR)))
            ldns_rr_list_set_rr(rrlist, rr, j++);
    }
    ldns_rr_list_set_rr_count(rrlist, j);
}

/* Packet content printing.
 */

static struct {
    ldns_rr_type type;
    const char *text;
} rr_types[] = {
    {LDNS_RR_TYPE_A,        "has address"},
    {LDNS_RR_TYPE_NS,       "name server"},
    {LDNS_RR_TYPE_CNAME,    "is an alias for"},
    {LDNS_RR_TYPE_WKS,      "has well known services"},
    {LDNS_RR_TYPE_PTR,      "domain name pointer"},
    {LDNS_RR_TYPE_HINFO,    "host information"},
    {LDNS_RR_TYPE_MX,       "mail is handled by"},
    {LDNS_RR_TYPE_TXT,      "descriptive text"},
    {LDNS_RR_TYPE_X25,      "x25 address"},
    {LDNS_RR_TYPE_ISDN,     "ISDN address"},
    {LDNS_RR_TYPE_SIG,      "has signature"},
    {LDNS_RR_TYPE_KEY,      "has key"},
    {LDNS_RR_TYPE_AAAA,     "has IPv6 address"},
    {LDNS_RR_TYPE_LOC,      "location"},
};

static void
print_opcode(ldns_pkt_opcode opcode) {
    ldns_lookup_table *lt = ldns_lookup_by_id(ldns_opcodes, opcode);

    if (lt && lt->name)
        printf("%s", lt->name);
    else
        printf("RESERVED%d", opcode);
}

static void
print_rcode(uint8_t rcode) {
    ldns_lookup_table *lt = ldns_lookup_by_id(ldns_rcodes, rcode);

    if (lt && lt->name)
        printf("%s", lt->name);
    else
        printf("RESERVED%d", rcode);
}

static int
print_rr_type(ldns_rr_type type) {
    char *str;
    int n;
    
    str = ldns_rr_type2str(type);
    n = printf("%s", str);
    free(str);
    return n;
}

static int
print_rr_class(ldns_rr_class cls) {
    char *str;
    int n;

    str = ldns_rr_class2str(cls);
    n = printf("%s", str);
    free(str);
    return n;
}

static int
print_rdf(ldns_rdf *rdf) {
    char *str;
    int n;

    str = ldns_rdf2str(rdf);
    n = printf("%s", str);
    free(str);
    return n;
}

static int
print_rdf_nodot(ldns_rdf *rdf) {
    char *str;
    int len, n;

    str = ldns_rdf2str(rdf);
    len = strlen(str);
    n = printf("%.*s", str[len-1] == '.' ? len-1 : len, str);
    free(str);
    return n;
}

static int
print_padding(int fromcol, int tocol) {
    int col = fromcol, nextcol = fromcol + 8 - fromcol%8;

    if (fromcol + 1 > tocol) tocol = fromcol + 1;
    for (; nextcol <= tocol; col = nextcol, nextcol += 8)
        printf("\t");
    for (; col < tocol; col++)
        printf(" ");
    return col - fromcol;
}

static void
print_rr_verbose(ldns_rr *rr) {
    bool isq = ldns_rr_is_question(rr);
    int rdcnt = ldns_rr_rd_count(rr);
    int i, n;

    /* bind9-host does not count the initial ';' here */
    n = isq ? printf(";") : 0;
    n = print_rdf(ldns_rr_owner(rr));
    if (!isq) {
        n += print_padding(n, 24);
        n += printf("%d", ldns_rr_ttl(rr));
    }
    n += print_padding(n, 32);
    n += print_rr_class(ldns_rr_get_class(rr));
    n += print_padding(n, 40);
    n += print_rr_type(ldns_rr_get_type(rr));
    for (i = 0; i < rdcnt; i++) {
        if (i == 0) print_padding(n, 48);
        else printf(" ");
        print_rdf(ldns_rr_rdf(rr, i));
    }
    printf("\n");
}

static void
print_pkt_section_verbose(const char *name, ldns_rr_list *rrlist) {
    int i, cnt = ldns_rr_list_rr_count(rrlist);

    if (cnt == 0)
        return;
    printf(";; %s SECTION:\n", name);
    for (i = 0; i < cnt; i++)
        print_rr_verbose(ldns_rr_list_rr(rrlist, i));
    printf("\n");
}

static void
print_pkt_verbose(ldns_pkt *pkt) {
    int got_flags = 0;

    printf(";; ->>HEADER<<- opcode: ");
    print_opcode(ldns_pkt_get_opcode(pkt));
    printf(", status: ");
    print_rcode(ldns_pkt_get_rcode(pkt));
    printf(", id: %u\n", ldns_pkt_id(pkt));
    printf(";; flags:");
    if (ldns_pkt_qr(pkt)) printf(" qr"), got_flags = 1;
    if (ldns_pkt_aa(pkt)) printf(" aa"), got_flags = 1;
    if (ldns_pkt_tc(pkt)) printf(" tc"), got_flags = 1;
    if (ldns_pkt_rd(pkt)) printf(" rd"), got_flags = 1;
    if (ldns_pkt_ra(pkt)) printf(" ra"), got_flags = 1;
    if (ldns_pkt_ad(pkt)) printf(" ad"), got_flags = 1;
    if (ldns_pkt_cd(pkt)) printf(" cd"), got_flags = 1;
    if (!got_flags) printf(" ");
    printf("; QUERY: %u, ANSWER: %u, AUTHORITY: %u, ADDITIONAL: %u\n",
        ldns_pkt_qdcount(pkt), ldns_pkt_ancount(pkt),
        ldns_pkt_nscount(pkt), ldns_pkt_arcount(pkt));
    if (ldns_pkt_edns(pkt))
        printf(";; EDNS: version: %u, udp=%u\n",
            ldns_pkt_edns_version(pkt), ldns_pkt_edns_udp_size(pkt));
    printf("\n");
    print_pkt_section_verbose("QUESTION", ldns_pkt_question(pkt));
    print_pkt_section_verbose("ANSWER", ldns_pkt_answer(pkt));
    print_pkt_section_verbose("AUTHORITY", ldns_pkt_authority(pkt));
    print_pkt_section_verbose("ADDITIONAL", ldns_pkt_additional(pkt));
}

static void
print_rr_short(ldns_rr *rr) {
    ldns_rr_type type = ldns_rr_get_type(rr);
    size_t i, rdcnt = ldns_rr_rd_count(rr);

    print_rdf_nodot(ldns_rr_owner(rr));
    printf(" ");
    for (i = 0; i < countof(rr_types); i++) {
        if (rr_types[i].type == type) {
            printf("%s", rr_types[i].text);
            goto found;
        }
    }

    printf("has ");
    print_rr_type(type);
    printf(" record");

found:
    for (i = 0; i < rdcnt; i++) {
        printf(" ");
        print_rdf(ldns_rr_rdf(rr, i));
    }
    printf("\n");
}

static void
print_pkt_short(ldns_pkt *pkt, bool print_rr_server) {
    ldns_rr_list *rrlist = ldns_pkt_answer(pkt);
    size_t i;

    for (i = 0; i < ldns_rr_list_rr_count(rrlist); i++) {
        if (print_rr_server) {
            printf("Nameserver ");
            print_rdf(ldns_pkt_answerfrom(pkt));
            printf(":\n\t");
        }
        print_rr_short(ldns_rr_list_rr(rrlist, i));
    }
}

static void
print_received_line(ldns_resolver *res, ldns_pkt *pkt) {
    char *from = ldns_rdf2str(ldns_pkt_answerfrom(pkt));

    printf("Received %zu bytes from %s#%d in %d ms\n",
            ldns_pkt_size(pkt), from, ldns_resolver_port(res),
            ldns_pkt_querytime(pkt));
    free(from);
}

/* Main program.
 *
 * Note that no memory is freed below this line by intention.
 */

#define DEFAULT_TCP_TIMEOUT 10
#define DEFAULT_UDP_TIMEOUT 5

enum operation_mode { M_AXFR, M_IXFR, M_DEFAULT_Q, M_SINGLE_Q, M_SOA };

static enum operation_mode o_mode = M_DEFAULT_Q;
static bool o_ignore_servfail = true;
static bool o_ip6_int = false;
static bool o_print_pkt_server = false;
static bool o_print_rr_server = false;
static bool o_recursive = true;
static bool o_tcp = false;
static bool o_verbose = false;
static char *o_name = NULL;
static char *o_server = NULL;
static int o_ipversion = LDNS_RESOLV_INETANY;
static int o_ndots = 1;
static int o_retries = 1;
static ldns_rr_class o_rrclass = LDNS_RR_CLASS_IN;
static ldns_rr_type o_rrtype = (ldns_rr_type)-1;
static time_t o_timeout = 0;
static uint32_t o_ixfr_serial = 0;

static void
usage(void) {
    fprintf(stderr,
    "Usage: %s [-aCdilrsTvw46] [-c class] [-N ndots] [-R number]\n"
    "       %*c [-t type] [-W wait] name [server]\n"
    "\t-a same as -v -t ANY\n"
    "\t-C query SOA records from all authoritative name servers\n"
    "\t-c use this query class (IN, CH, HS, etc)\n"
    "\t-d produce verbose output, same as -v\n"
    "\t-i use IP6.INT for IPv6 reverse lookups\n"
    "\t-l list records in a zone via AXFR\n"
    "\t-N consider names with at least this many dots as absolute\n"
    "\t-R retry UDP queries this many times\n"
    "\t-r disable recursive query\n"
    "\t-s do not ignore SERVFAIL responses\n"
    "\t-T send query via TCP\n"
    "\t-t use this query type (A, AAAA, MX, etc)\n"
    "\t-v produce verbose output\n"
    "\t-w wait forever for a server reply\n"
    "\t-W wait this many seconds for a reply\n"
    "\t-4 use IPv4 only\n"
    "\t-6 use IPv6 only\n",
    progname, (int)strlen(progname), ' ');
    exit(1);
}

static void
parse_args(int argc, char *argv[]) {
    int ch;

    progname = argv[0];
    while ((ch = getopt(argc, argv, "aCdilrsTvw46c:N:R:t:W:")) != -1) {
        switch (ch) {
        case 'a':
            if (o_mode != M_AXFR)
                o_mode = M_SINGLE_Q;
            o_rrtype = LDNS_RR_TYPE_ANY;
            o_verbose = true;
            break;
        case 'C':
            o_mode = M_SOA;
            o_print_rr_server = true;
            o_rrclass = LDNS_RR_CLASS_IN;
            o_rrtype = LDNS_RR_TYPE_NS;
            break;
        case 'c':
            /* bind9-host sets o_mode to M_SINGLE_Q here */
            o_rrclass = ldns_get_rr_class_by_name(optarg);
            if (o_rrclass <= 0)
                die(2, "invalid class: %s\n", optarg);
            break;
        case 'd': o_verbose = true; break;
        case 'i': o_ip6_int = true; break;
        case 'l':
            o_mode = M_AXFR;
            if (o_rrtype == (ldns_rr_type)-1)
                o_rrtype = LDNS_RR_TYPE_AXFR;
            o_tcp = true;
            break;
        case 'N':
            o_ndots = atoi(optarg);
            if (o_ndots < 0) o_ndots = 0;
            break;
        case 'n':
            /* bind9-host accepts and ignores this option */
            break;
        case 'r': o_recursive = 0; break;
        case 'R':
            o_retries = atoi(optarg);
            if (o_retries <= 0) o_retries = 1;
            if (o_retries > 255) o_retries = 255;
            break;
        case 's': o_ignore_servfail = false; break;
        case 'T': o_tcp = true; break;
        case 't':
            if (o_mode != M_AXFR)
                o_mode = M_SINGLE_Q;
            if (strncasecmp(optarg, "ixfr=", 5) == 0) {
                o_rrtype = LDNS_RR_TYPE_IXFR;
                o_ixfr_serial = atol(optarg + 5);
            } else {
                o_rrtype = ldns_get_rr_type_by_name(optarg);
                if (o_rrtype <= 0)
                    die(2, "invalid type: %s\n", optarg);
            }
            if (o_rrtype == LDNS_RR_TYPE_AXFR) {
                o_mode = M_AXFR;
                o_rrtype = LDNS_RR_TYPE_ANY;
                o_verbose = true;
            }
            if (o_rrtype == LDNS_RR_TYPE_IXFR) {
                o_mode = M_IXFR;
                o_rrtype = LDNS_RR_TYPE_ANY;
            }
            break;
        case 'v': o_verbose = true; break;
        case 'w':
              o_timeout = (time_t)INT_MAX;
              break;
        case 'W':
            o_timeout = atol(optarg);
            if (o_timeout <= 0) o_timeout = 1;
            break;
        case '4': o_ipversion = LDNS_RESOLV_INET; break;
        case '6': o_ipversion = LDNS_RESOLV_INET6; break;
        default:
            usage();
        }
    }
    argc -= optind;
    argv += optind;
    /* bind9-host ignores arguments after the 2-nd one */
    if (argc < 1)
        usage();
    o_name = argv[0];
    if (argc > 1) {
        o_server = argv[1];
        o_print_pkt_server = true;
    }
    if (o_rrtype == (ldns_rr_type)-1)
        o_rrtype = LDNS_RR_TYPE_A;
}

static ldns_rdf*
safe_str2rdf_dname(const char *name) {
    ldns_rdf *dname;
    ldns_status status;

    if ((status = ldns_str2rdf_dname(&dname, name)) != LDNS_STATUS_OK) {
        die(1, "'%s' is not a legal name (%s)",
            name, ldns_get_errorstr_by_id(status));
    }
    return dname;
}

static ldns_rdf*
safe_dname_cat_clone(const ldns_rdf *rd1, const ldns_rdf *rd2) {
    ldns_rdf *result = ldns_dname_cat_clone(rd1, rd2);

    if (!result)
        die(1, "not enought memory for a domain name");
    /* Why doesn't ldns_dname_cat_clone check this condition? */
    if (ldns_rdf_size(result) > LDNS_MAX_DOMAINLEN)
        die(1, "'%s' is not a legal name (%s)\n", ldns_rdf2str(result),
            ldns_get_errorstr_by_id(LDNS_STATUS_DOMAINNAME_OVERFLOW));
    return result;
}

static bool
query(ldns_resolver *res, ldns_rdf *domain, ldns_pkt **pkt, bool close_tcp) {
    ldns_status status;
    ldns_pkt_rcode rcode;
    int i, cnt;

    if (o_verbose) {
        printf("Trying \"");
        print_rdf_nodot(domain);
        printf("\"\n");
    }
    for (cnt = ldns_resolver_nameserver_count(res), i = 0; i < cnt; i++) {
        status = ldns_resolver_send_to(pkt, res, domain, o_rrtype,
            o_rrclass, o_recursive ? LDNS_RD : 0, o_ixfr_serial, i,
            close_tcp);
        if (status != LDNS_STATUS_OK) {
            *pkt = NULL;
            continue;
        }
        if (ldns_pkt_tc(*pkt) && !ldns_resolver_usevc(res)) {
            if (o_verbose)
                printf(";; Truncated, retrying in TCP mode.\n");
            ldns_resolver_set_usevc(res, true);
            status = ldns_resolver_send_to(pkt, res, domain, o_rrtype,
                o_rrclass, o_recursive ? LDNS_RD : 0, o_ixfr_serial, i,
                close_tcp);
            ldns_resolver_set_usevc(res, false);
            if (status != LDNS_STATUS_OK)
                continue;
        }
        rcode = ldns_pkt_get_rcode(*pkt);
        if (o_ignore_servfail && rcode == LDNS_RCODE_SERVFAIL && cnt > 1)
            continue;
        return rcode == LDNS_RCODE_NOERROR;
    }
    if (*pkt == NULL) {
        printf(";; connection timed out; no servers could be reached\n");
        exit(1);
    }
    return false;
}

static ldns_rdf *
search(ldns_resolver *res, ldns_rdf *domain, ldns_pkt **pkt,
    bool absolute, bool close_tcp) {
    ldns_rdf *dname, **searchlist;
    int i, n;

    if (absolute && query(res, domain, pkt, close_tcp))
        return domain;

    if ((dname = ldns_resolver_domain(res)) != NULL) {
        dname = safe_dname_cat_clone(domain, dname);
        if (query(res, dname, pkt, close_tcp))
            return dname;
    }

    searchlist = ldns_resolver_searchlist(res);
    n = ldns_resolver_searchlist_count(res);
    for (i = 0; i < n; i++) {
        dname = safe_dname_cat_clone(domain, searchlist[i]);
        if (query(res, dname, pkt, close_tcp))
            return dname;
    }

    if (!absolute && query(res, domain, pkt, close_tcp))
        return domain;

    return NULL;
}

static void
report(ldns_resolver *res, ldns_rdf *domain, ldns_pkt *pkt) {
    ldns_pkt_rcode rcode;

    if (o_print_pkt_server) {
        printf("Using domain server:\nName: %s\nAddress: ", o_server);
        print_rdf(ldns_pkt_answerfrom(pkt));
        printf("#%d\nAliases: \n\n", ldns_resolver_port(res));
        o_print_pkt_server = false;
    }
    rcode = ldns_pkt_get_rcode(pkt);
    if (rcode != LDNS_RCODE_NOERROR) {
        printf("Host ");
        print_rdf_nodot(domain);
        printf(" not found: %d(", rcode);
        print_rcode(rcode);
        printf(")\n");
    } else {
        if (o_verbose) {
            print_pkt_verbose(pkt);
        } else {
            print_pkt_short(pkt, o_print_rr_server);
            if (o_mode == M_SINGLE_Q &&
                ldns_rr_list_rr_count(ldns_pkt_answer(pkt)) == 0) {
                print_rdf_nodot(domain);
                printf(" has no ");
                print_rr_type(o_rrtype);
                printf(" record\n");
            }
        }
    }
    if (o_verbose)
        print_received_line(res, pkt);
}

static bool
doquery(ldns_resolver *res, ldns_rdf *domain) {
    ldns_pkt *pkt;
    bool q;

    q = query(res, domain, &pkt, true);
    report(res, domain, pkt);
    return q;
}

static bool
doquery_filtered(ldns_resolver *res, ldns_rdf *domain) {
    ldns_pkt *pkt;
    bool q;

    q = query(res, domain, &pkt, true);
    ldns_pkt_filter_answer(pkt, o_rrtype);
    report(res, domain, pkt);
    return q;
}

static bool
dosearch(ldns_resolver *res, ldns_rdf *domain, bool absolute) {
    ldns_pkt *pkt;
    ldns_rdf *dname;

    dname = search(res, domain, &pkt, absolute, true);
    report(res, dname != NULL ? dname : domain, pkt);
    return o_mode != M_DEFAULT_Q ? (dname != NULL) :
        (dname != NULL) &&
        (o_rrtype = LDNS_RR_TYPE_AAAA, doquery_filtered(res, dname)) &&
        (o_rrtype = LDNS_RR_TYPE_MX, doquery_filtered(res, dname));
}

static bool
dozonetransfer(ldns_resolver *res, ldns_rdf *domain, bool absolute) {
    ldns_pkt *pkt, *nextpkt;
    ldns_rdf *dname;
    ldns_rr_type rrtype;
    ldns_rr_list *rrl;
    ldns_rr *rr;
    size_t i, nsoa = 0;
    uint32_t first_serial = 0;

    rrtype = o_rrtype;
    o_rrtype = (o_mode == M_AXFR) ? LDNS_RR_TYPE_AXFR : LDNS_RR_TYPE_IXFR;
    dname = search(res, domain, &pkt, absolute, false);

    for (;;) {
        rrl = ldns_rr_list_clone(ldns_pkt_answer(pkt));
        ldns_pkt_filter_answer(pkt, rrtype);
        report(res, dname != NULL ? dname : domain, pkt);
        if ((dname == NULL) ||
                (ldns_pkt_get_rcode(pkt) != LDNS_RCODE_NOERROR)) {
            printf("; Transfer failed.\n");
            ldns_tcp_close(res);
            return false;
        }
        for (i = 0; i < ldns_rr_list_rr_count(rrl); i++) {
            rr = ldns_rr_list_rr(rrl, i);
            if (nsoa == 0) {
                if (ldns_rr_get_type(rr) != LDNS_RR_TYPE_SOA) {
                    printf("; Transfer failed. "
                           "Didn't start with SOA answer.\n");
                    ldns_tcp_close(res);
                    return false;
                }
                first_serial = ldns_rr_soa_get_serial(rr);
                if ((o_mode == M_IXFR) && (first_serial <= o_ixfr_serial)) {
                    ldns_tcp_close(res);
                    return true;
                }
            }
            if (ldns_rr_get_type(rr) == LDNS_RR_TYPE_SOA) {
                nsoa = nsoa < 2 ? nsoa + 1 : 1;
                if ((nsoa == 2) &&
                        (ldns_rr_soa_get_serial(rr) == first_serial)) {
                    ldns_tcp_close(res);
                    return true;
                }
            }
        }
        if (ldns_tcp_read(&nextpkt, res) != LDNS_STATUS_OK) {
            printf("; Transfer failed.\n");
            return false;
        }
        ldns_pkt_set_answerfrom(nextpkt,
                ldns_rdf_clone(ldns_pkt_answerfrom(pkt)));
        ldns_pkt_free(pkt);
        ldns_rr_list_free(rrl);
        pkt = nextpkt;
    }
}

static bool
dosoa(ldns_resolver *res, ldns_rdf *domain, bool absolute) {
    ldns_rr_list *answer, **nsaddrs;
    ldns_rdf *dname, *addr;
    ldns_pkt *pkt;
    ldns_rr *rr;
    size_t i, j, n, cnt;

    if ((dname = search(res, domain, &pkt, absolute, true)) == NULL)
        return false;

    answer = ldns_pkt_answer(pkt);
    cnt = ldns_rr_list_rr_count(answer);
    nsaddrs = alloca(cnt*sizeof(*nsaddrs));
    for (n = 0, i = 0; i < cnt; i++)
        if ((addr = ldns_rr_ns_nsdname(ldns_rr_list_rr(answer, i))) != NULL)
            nsaddrs[n++] = ldns_get_rr_list_addr_by_name(res, 
                addr, LDNS_RR_CLASS_IN, 0); 

    o_print_pkt_server = false;
    o_recursive = false;
    o_rrtype = LDNS_RR_TYPE_SOA;
    for (i = 0; i < n; i++) {
        cnt = ldns_rr_list_rr_count(nsaddrs[i]);
        for (j = 0; j < cnt; j++) {
            ldns_resolver_remove_nameservers(res);
            rr = ldns_rr_list_rr(nsaddrs[i], j);
            if ((ldns_resolver_ip6(res) == LDNS_RESOLV_INET &&
                ldns_rr_get_type(rr) == LDNS_RR_TYPE_AAAA) ||
                (ldns_resolver_ip6(res) == LDNS_RESOLV_INET6 &&
                ldns_rr_get_type(rr) == LDNS_RR_TYPE_A))
                continue;
            if (ldns_resolver_push_nameserver_rr(res, rr) == LDNS_STATUS_OK)
                /* bind9-host queries for domain, not dname here */
                doquery(res, dname);
        }
    }
    return 0;
}

static void
resolver_set_nameserver_hostname(ldns_resolver *res, const char *server) {
    struct addrinfo hints, *ailist, *ai;
    ldns_status status;
    ldns_rdf *rdf;
    int err;

    memset(&hints, 0, sizeof hints);
    switch (ldns_resolver_ip6(res)) {
    case LDNS_RESOLV_INET: hints.ai_family = PF_INET; break;
    case LDNS_RESOLV_INET6: hints.ai_family = PF_INET6; break;
    default: hints.ai_family = PF_UNSPEC; break;
    }
    hints.ai_socktype = SOCK_STREAM;
    do err = getaddrinfo(server, NULL, &hints, &ailist);
    while (err == EAI_AGAIN);
    if (err != 0)
        die(1, "couldn't get address for '%s': %s", server, gai_strerror(err));
    for (ai = ailist; ai != NULL; ai = ai->ai_next) {
        if ((rdf = ldns_sockaddr_storage2rdf((void*)ai->ai_addr, NULL)) == NULL)
            die(1, "couldn't allocate an rdf: %s",
                ldns_get_errorstr_by_id(LDNS_STATUS_MEM_ERR));
        status = ldns_resolver_push_nameserver(res, rdf);
        if (status != LDNS_STATUS_OK)
            die(1, "couldn't push a nameserver address: %s",
                ldns_get_errorstr_by_id(status));
    }
}

static void
resolver_set_nameserver_str(ldns_resolver *res, const char *server) {
    ldns_rdf *addr;

    ldns_resolver_remove_nameservers(res);
    addr = ldns_rdf_new_addr_frm_str(server);
    if (addr) {
        if (ldns_resolver_push_nameserver(res, addr) != LDNS_STATUS_OK)
            die(1, "couldn't push a nameserver address");
    } else
        resolver_set_nameserver_hostname(res, server);
}

int
main(int argc, char *argv[]) {
    ldns_rdf *addr, *dname;
    ldns_resolver *res;
    ldns_status status;
    struct timeval restimeout;

    parse_args(argc, argv);

    status = ldns_resolver_new_default(&res);
    if (status != LDNS_STATUS_OK)
        die(1, "error creating resolver: %s", ldns_get_errorstr_by_id(status));
    if (ldns_resolver_nameserver_count(res) == 0)
        ldns_resolver_push_default_servers(res);

    ldns_resolver_set_usevc(res, o_tcp);
    restimeout.tv_sec = o_timeout > 0 ? o_timeout :
        o_tcp ? DEFAULT_TCP_TIMEOUT : DEFAULT_UDP_TIMEOUT;
    restimeout.tv_usec = 0;
    ldns_resolver_set_timeout(res, restimeout);
    ldns_resolver_set_retry(res, o_retries+1);
    ldns_resolver_set_ip6(res, o_ipversion);
    ldns_resolver_set_defnames(res, false);
    ldns_resolver_set_fallback(res, false);

    if (o_server)
        resolver_set_nameserver_str(res, o_server);

    if (ldns_str2rdf_a(&addr, o_name) == LDNS_STATUS_OK) {
        dname = ldns_rdf_reverse_a(addr, "in-addr.arpa");
        if (dname == NULL)
            die(1, "can't reverse '%s': %s", o_name,
                ldns_get_errorstr_by_id(LDNS_STATUS_MEM_ERR));
        o_mode = M_SINGLE_Q;
        o_rrtype = LDNS_RR_TYPE_PTR;
        return !doquery(res, dname);
    } else if (ldns_str2rdf_aaaa(&addr, o_name) == LDNS_STATUS_OK) {
        dname = ldns_rdf_reverse_aaaa(addr, o_ip6_int ? "ip6.int" : "ip6.arpa");
        if (dname == NULL)
            die(1, "can't reverse '%s': %s", o_name,
                ldns_get_errorstr_by_id(LDNS_STATUS_MEM_ERR));
        o_mode = M_SINGLE_Q;
        o_rrtype = LDNS_RR_TYPE_PTR;
        return !doquery(res, dname);
    }
    return !(o_mode == M_SOA ? dosoa :
             o_mode == M_AXFR ? dozonetransfer :
             o_mode == M_IXFR ? dozonetransfer :
             dosearch)
        (res, safe_str2rdf_dname(o_name), ndots(o_name) >= o_ndots);
}