busybox/networking/nslookup.c
<<
>>
Prefs
   1/* vi: set sw=4 ts=4: */
   2
   3//config:config NSLOOKUP
   4//config:       bool "nslookup (9.7 kb)"
   5//config:       default y
   6//config:       help
   7//config:       nslookup is a tool to query Internet name servers.
   8//config:
   9//config:config FEATURE_NSLOOKUP_BIG
  10//config:       bool "Use internal resolver code instead of libc"
  11//config:       depends on NSLOOKUP
  12//config:       default y
  13//config:
  14//config:config FEATURE_NSLOOKUP_LONG_OPTIONS
  15//config:       bool "Enable long options"
  16//config:       default y
  17//config:       depends on FEATURE_NSLOOKUP_BIG && LONG_OPTS
  18
  19//applet:IF_NSLOOKUP(APPLET(nslookup, BB_DIR_USR_BIN, BB_SUID_DROP))
  20
  21//kbuild:lib-$(CONFIG_NSLOOKUP) += nslookup.o
  22
  23//usage:#define nslookup_trivial_usage
  24//usage:       IF_FEATURE_NSLOOKUP_BIG("[-type=QUERY_TYPE] [-debug] ") "HOST [DNS_SERVER]"
  25//usage:#define nslookup_full_usage "\n\n"
  26//usage:       "Query DNS about HOST"
  27//usage:       IF_FEATURE_NSLOOKUP_BIG("\n")
  28//usage:       IF_FEATURE_NSLOOKUP_BIG("\nQUERY_TYPE: soa,ns,a,"IF_FEATURE_IPV6("aaaa,")"cname,mx,txt,ptr,any")
  29//usage:#define nslookup_example_usage
  30//usage:       "$ nslookup localhost\n"
  31//usage:       "Server:     default\n"
  32//usage:       "Address:    default\n"
  33//usage:       "\n"
  34//usage:       "Name:       debian\n"
  35//usage:       "Address:    127.0.0.1\n"
  36
  37#include <resolv.h>
  38#include <net/if.h>     /* for IFNAMSIZ */
  39//#include <arpa/inet.h>
  40//#include <netdb.h>
  41#include "libbb.h"
  42#include "common_bufsiz.h"
  43
  44
  45#if !ENABLE_FEATURE_NSLOOKUP_BIG
  46
  47/*
  48 * Mini nslookup implementation for busybox
  49 *
  50 * Copyright (C) 1999,2000 by Lineo, inc. and John Beppu
  51 * Copyright (C) 1999,2000,2001 by John Beppu <beppu@codepoet.org>
  52 *
  53 * Correct default name server display and explicit name server option
  54 * added by Ben Zeckel <bzeckel@hmc.edu> June 2001
  55 *
  56 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  57 */
  58
  59/*
  60 * I'm only implementing non-interactive mode;
  61 * I totally forgot nslookup even had an interactive mode.
  62 *
  63 * This applet is the only user of res_init(). Without it,
  64 * you may avoid pulling in _res global from libc.
  65 */
  66
  67/* Examples of 'standard' nslookup output
  68 * $ nslookup yahoo.com
  69 * Server:         128.193.0.10
  70 * Address:        128.193.0.10#53
  71 *
  72 * Non-authoritative answer:
  73 * Name:   yahoo.com
  74 * Address: 216.109.112.135
  75 * Name:   yahoo.com
  76 * Address: 66.94.234.13
  77 *
  78 * $ nslookup 204.152.191.37
  79 * Server:         128.193.4.20
  80 * Address:        128.193.4.20#53
  81 *
  82 * Non-authoritative answer:
  83 * 37.191.152.204.in-addr.arpa     canonical name = 37.32-27.191.152.204.in-addr.arpa.
  84 * 37.32-27.191.152.204.in-addr.arpa       name = zeus-pub2.kernel.org.
  85 *
  86 * Authoritative answers can be found from:
  87 * 32-27.191.152.204.in-addr.arpa  nameserver = ns1.kernel.org.
  88 * 32-27.191.152.204.in-addr.arpa  nameserver = ns2.kernel.org.
  89 * 32-27.191.152.204.in-addr.arpa  nameserver = ns3.kernel.org.
  90 * ns1.kernel.org  internet address = 140.211.167.34
  91 * ns2.kernel.org  internet address = 204.152.191.4
  92 * ns3.kernel.org  internet address = 204.152.191.36
  93 */
  94
  95static int print_host(const char *hostname, const char *header)
  96{
  97        /* We can't use xhost2sockaddr() - we want to get ALL addresses,
  98         * not just one */
  99        struct addrinfo *result = NULL;
 100        int rc;
 101        struct addrinfo hint;
 102
 103        memset(&hint, 0 , sizeof(hint));
 104        /* hint.ai_family = AF_UNSPEC; - zero anyway */
 105        /* Needed. Or else we will get each address thrice (or more)
 106         * for each possible socket type (tcp,udp,raw...): */
 107        hint.ai_socktype = SOCK_STREAM;
 108        // hint.ai_flags = AI_CANONNAME;
 109        rc = getaddrinfo(hostname, NULL /*service*/, &hint, &result);
 110
 111        if (rc == 0) {
 112                struct addrinfo *cur = result;
 113                unsigned cnt = 0;
 114
 115                printf("%-10s %s\n", header, hostname);
 116                // puts(cur->ai_canonname); ?
 117                while (cur) {
 118                        char *dotted, *revhost;
 119                        dotted = xmalloc_sockaddr2dotted_noport(cur->ai_addr);
 120                        revhost = xmalloc_sockaddr2hostonly_noport(cur->ai_addr);
 121
 122                        printf("Address %u: %s%c", ++cnt, dotted, revhost ? ' ' : '\n');
 123                        if (revhost) {
 124                                puts(revhost);
 125                                if (ENABLE_FEATURE_CLEAN_UP)
 126                                        free(revhost);
 127                        }
 128                        if (ENABLE_FEATURE_CLEAN_UP)
 129                                free(dotted);
 130                        cur = cur->ai_next;
 131                }
 132        } else {
 133#if ENABLE_VERBOSE_RESOLUTION_ERRORS
 134                bb_error_msg("can't resolve '%s': %s", hostname, gai_strerror(rc));
 135#else
 136                bb_error_msg("can't resolve '%s'", hostname);
 137#endif
 138        }
 139        if (ENABLE_FEATURE_CLEAN_UP && result)
 140                freeaddrinfo(result);
 141        return (rc != 0);
 142}
 143
 144/* lookup the default nameserver and display it */
 145static void server_print(void)
 146{
 147        char *server;
 148        struct sockaddr *sa;
 149
 150#if ENABLE_FEATURE_IPV6
 151        sa = (struct sockaddr*)_res._u._ext.nsaddrs[0];
 152        if (!sa)
 153#endif
 154                sa = (struct sockaddr*)&_res.nsaddr_list[0];
 155        server = xmalloc_sockaddr2dotted_noport(sa);
 156
 157        print_host(server, "Server:");
 158        if (ENABLE_FEATURE_CLEAN_UP)
 159                free(server);
 160        bb_putchar('\n');
 161}
 162
 163/* alter the global _res nameserver structure to use
 164   an explicit dns server instead of what is in /etc/resolv.conf */
 165static void set_default_dns(const char *server)
 166{
 167        len_and_sockaddr *lsa;
 168
 169        if (!server)
 170                return;
 171
 172        /* NB: this works even with, say, "[::1]:5353"! :) */
 173        lsa = xhost2sockaddr(server, 53);
 174
 175        if (lsa->u.sa.sa_family == AF_INET) {
 176                _res.nscount = 1;
 177                /* struct copy */
 178                _res.nsaddr_list[0] = lsa->u.sin;
 179        }
 180#if ENABLE_FEATURE_IPV6
 181        /* Hoped libc can cope with IPv4 address there too.
 182         * No such luck, glibc 2.4 segfaults even with IPv6,
 183         * maybe I misunderstand how to make glibc use IPv6 addr?
 184         * (uclibc 0.9.31+ should work) */
 185        if (lsa->u.sa.sa_family == AF_INET6) {
 186                // glibc neither SEGVs nor sends any dgrams with this
 187                // (strace shows no socket ops):
 188                //_res.nscount = 0;
 189                _res._u._ext.nscount = 1;
 190                /* store a pointer to part of malloc'ed lsa */
 191                _res._u._ext.nsaddrs[0] = &lsa->u.sin6;
 192                /* must not free(lsa)! */
 193        }
 194#endif
 195}
 196
 197int nslookup_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 198int nslookup_main(int argc, char **argv)
 199{
 200        /* We allow 1 or 2 arguments.
 201         * The first is the name to be looked up and the second is an
 202         * optional DNS server with which to do the lookup.
 203         * More than 3 arguments is an error to follow the pattern of the
 204         * standard nslookup */
 205        if (!argv[1] || argv[1][0] == '-' || argc > 3)
 206                bb_show_usage();
 207
 208        /* initialize DNS structure _res used in printing the default
 209         * name server and in the explicit name server option feature. */
 210        res_init();
 211        /* rfc2133 says this enables IPv6 lookups */
 212        /* (but it also says "may be enabled in /etc/resolv.conf") */
 213        /*_res.options |= RES_USE_INET6;*/
 214
 215        set_default_dns(argv[2]);
 216
 217        server_print();
 218
 219        /* getaddrinfo and friends are free to request a resolver
 220         * reinitialization. Just in case, set_default_dns() again
 221         * after getaddrinfo (in server_print). This reportedly helps
 222         * with bug 675 "nslookup does not properly use second argument"
 223         * at least on Debian Wheezy and Openwrt AA (eglibc based).
 224         */
 225        set_default_dns(argv[2]);
 226
 227        return print_host(argv[1], "Name:");
 228}
 229
 230
 231#else /****** A version from LEDE / OpenWRT ******/
 232
 233/*
 234 * musl compatible nslookup
 235 *
 236 * Copyright (C) 2017 Jo-Philipp Wich <jo@mein.io>
 237 *
 238 * Permission to use, copy, modify, and/or distribute this software for any
 239 * purpose with or without fee is hereby granted, provided that the above
 240 * copyright notice and this permission notice appear in all copies.
 241 *
 242 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 243 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 244 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 245 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 246 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 247 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 248 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 249 */
 250
 251#if 0
 252# define dbg(...) fprintf(stderr, __VA_ARGS__)
 253#else
 254# define dbg(...) ((void)0)
 255#endif
 256
 257struct ns {
 258        const char *name;
 259        len_and_sockaddr *lsa;
 260        //UNUSED: int failures;
 261        int replies;
 262};
 263
 264struct query {
 265        const char *name;
 266        unsigned qlen;
 267//      unsigned latency;
 268//      uint8_t rcode;
 269        unsigned char query[512];
 270//      unsigned char reply[512];
 271};
 272
 273static const struct {
 274        unsigned char type;
 275        char name[7];
 276} qtypes[] = {
 277        { ns_t_soa,   "SOA"   },
 278        { ns_t_ns,    "NS"    },
 279        { ns_t_a,     "A"     },
 280#if ENABLE_FEATURE_IPV6
 281        { ns_t_aaaa,  "AAAA"  },
 282#endif
 283        { ns_t_cname, "CNAME" },
 284        { ns_t_mx,    "MX"    },
 285        { ns_t_txt,   "TXT"   },
 286        { ns_t_ptr,   "PTR"   },
 287        { ns_t_any,   "ANY"   },
 288};
 289
 290static const char *const rcodes[] = {
 291        "NOERROR",    // 0
 292        "FORMERR",    // 1
 293        "SERVFAIL",   // 2
 294        "NXDOMAIN",   // 3
 295        "NOTIMP",     // 4
 296        "REFUSED",    // 5
 297        "YXDOMAIN",   // 6
 298        "YXRRSET",    // 7
 299        "NXRRSET",    // 8
 300        "NOTAUTH",    // 9
 301        "NOTZONE",    // 10
 302        "11",         // 11 not assigned
 303        "12",         // 12 not assigned
 304        "13",         // 13 not assigned
 305        "14",         // 14 not assigned
 306        "15",         // 15 not assigned
 307};
 308
 309#if ENABLE_FEATURE_IPV6
 310static const char v4_mapped[12] = { 0,0,0,0, 0,0,0,0, 0,0,0xff,0xff };
 311#endif
 312
 313struct globals {
 314        unsigned default_port;
 315        unsigned default_retry;
 316        unsigned default_timeout;
 317        unsigned query_count;
 318        unsigned serv_count;
 319        struct ns *server;
 320        struct query *query;
 321        char *search;
 322        smalluint have_search_directive;
 323        smalluint exitcode;
 324} FIX_ALIASING;
 325#define G (*(struct globals*)bb_common_bufsiz1)
 326#define INIT_G() do { \
 327        setup_common_bufsiz(); \
 328        G.default_port = 53; \
 329        G.default_retry = 2; \
 330        G.default_timeout = 5; \
 331} while (0)
 332
 333enum {
 334        OPT_debug = (1 << 0),
 335};
 336
 337static int parse_reply(const unsigned char *msg, size_t len)
 338{
 339        HEADER *header;
 340
 341        ns_msg handle;
 342        ns_rr rr;
 343        int i, n, rdlen;
 344        const char *format = NULL;
 345        char astr[INET6_ADDRSTRLEN], dname[MAXDNAME];
 346        const unsigned char *cp;
 347
 348        header = (HEADER *)msg;
 349        if (!header->aa)
 350                printf("Non-authoritative answer:\n");
 351
 352        if (ns_initparse(msg, len, &handle) != 0) {
 353                //printf("Unable to parse reply: %s\n", strerror(errno));
 354                return -1;
 355        }
 356
 357        for (i = 0; i < ns_msg_count(handle, ns_s_an); i++) {
 358                if (ns_parserr(&handle, ns_s_an, i, &rr) != 0) {
 359                        //printf("Unable to parse resource record: %s\n", strerror(errno));
 360                        return -1;
 361                }
 362
 363                rdlen = ns_rr_rdlen(rr);
 364
 365                switch (ns_rr_type(rr))
 366                {
 367                case ns_t_a:
 368                        if (rdlen != 4) {
 369                                dbg("unexpected A record length %d\n", rdlen);
 370                                return -1;
 371                        }
 372                        inet_ntop(AF_INET, ns_rr_rdata(rr), astr, sizeof(astr));
 373                        printf("Name:\t%s\nAddress: %s\n", ns_rr_name(rr), astr);
 374                        break;
 375
 376#if ENABLE_FEATURE_IPV6
 377                case ns_t_aaaa:
 378                        if (rdlen != 16) {
 379                                dbg("unexpected AAAA record length %d\n", rdlen);
 380                                return -1;
 381                        }
 382                        inet_ntop(AF_INET6, ns_rr_rdata(rr), astr, sizeof(astr));
 383                        /* bind-utils-9.11.3 uses the same format for A and AAAA answers */
 384                        printf("Name:\t%s\nAddress: %s\n", ns_rr_name(rr), astr);
 385                        break;
 386#endif
 387
 388                case ns_t_ns:
 389                        if (!format)
 390                                format = "%s\tnameserver = %s\n";
 391                        /* fall through */
 392
 393                case ns_t_cname:
 394                        if (!format)
 395                                format = "%s\tcanonical name = %s\n";
 396                        /* fall through */
 397
 398                case ns_t_ptr:
 399                        if (!format)
 400                                format = "%s\tname = %s\n";
 401                        if (ns_name_uncompress(ns_msg_base(handle), ns_msg_end(handle),
 402                                        ns_rr_rdata(rr), dname, sizeof(dname)) < 0
 403                        ) {
 404                                //printf("Unable to uncompress domain: %s\n", strerror(errno));
 405                                return -1;
 406                        }
 407                        printf(format, ns_rr_name(rr), dname);
 408                        break;
 409
 410                case ns_t_mx:
 411                        if (rdlen < 2) {
 412                                printf("MX record too short\n");
 413                                return -1;
 414                        }
 415                        n = ns_get16(ns_rr_rdata(rr));
 416                        if (ns_name_uncompress(ns_msg_base(handle), ns_msg_end(handle),
 417                                        ns_rr_rdata(rr) + 2, dname, sizeof(dname)) < 0
 418                        ) {
 419                                //printf("Cannot uncompress MX domain: %s\n", strerror(errno));
 420                                return -1;
 421                        }
 422                        printf("%s\tmail exchanger = %d %s\n", ns_rr_name(rr), n, dname);
 423                        break;
 424
 425                case ns_t_txt:
 426                        if (rdlen < 1) {
 427                                //printf("TXT record too short\n");
 428                                return -1;
 429                        }
 430                        n = *(unsigned char *)ns_rr_rdata(rr);
 431                        if (n > 0) {
 432                                memset(dname, 0, sizeof(dname));
 433                                memcpy(dname, ns_rr_rdata(rr) + 1, n);
 434                                printf("%s\ttext = \"%s\"\n", ns_rr_name(rr), dname);
 435                        }
 436                        break;
 437
 438                case ns_t_soa:
 439                        if (rdlen < 20) {
 440                                dbg("SOA record too short:%d\n", rdlen);
 441                                return -1;
 442                        }
 443
 444                        printf("%s\n", ns_rr_name(rr));
 445
 446                        cp = ns_rr_rdata(rr);
 447                        n = ns_name_uncompress(ns_msg_base(handle), ns_msg_end(handle),
 448                                               cp, dname, sizeof(dname));
 449                        if (n < 0) {
 450                                //printf("Unable to uncompress domain: %s\n", strerror(errno));
 451                                return -1;
 452                        }
 453
 454                        printf("\torigin = %s\n", dname);
 455                        cp += n;
 456
 457                        n = ns_name_uncompress(ns_msg_base(handle), ns_msg_end(handle),
 458                                               cp, dname, sizeof(dname));
 459                        if (n < 0) {
 460                                //printf("Unable to uncompress domain: %s\n", strerror(errno));
 461                                return -1;
 462                        }
 463
 464                        printf("\tmail addr = %s\n", dname);
 465                        cp += n;
 466
 467                        printf("\tserial = %lu\n", ns_get32(cp));
 468                        cp += 4;
 469
 470                        printf("\trefresh = %lu\n", ns_get32(cp));
 471                        cp += 4;
 472
 473                        printf("\tretry = %lu\n", ns_get32(cp));
 474                        cp += 4;
 475
 476                        printf("\texpire = %lu\n", ns_get32(cp));
 477                        cp += 4;
 478
 479                        printf("\tminimum = %lu\n", ns_get32(cp));
 480                        break;
 481
 482                default:
 483                        break;
 484                }
 485        }
 486
 487        return i;
 488}
 489
 490/*
 491 * Function logic borrowed & modified from musl libc, res_msend.c
 492 * G.query_count is always > 0.
 493 */
 494static int send_queries(struct ns *ns)
 495{
 496        unsigned char reply[512];
 497        uint8_t rcode;
 498        len_and_sockaddr *local_lsa;
 499        struct pollfd pfd;
 500        int servfail_retry = 0;
 501        int n_replies = 0;
 502//      int save_idx = 0;
 503        unsigned retry_interval;
 504        unsigned timeout = G.default_timeout * 1000;
 505        unsigned tstart, tsent, tcur;
 506
 507        pfd.events = POLLIN;
 508        pfd.fd = xsocket_type(&local_lsa, ns->lsa->u.sa.sa_family, SOCK_DGRAM);
 509        /*
 510         * local_lsa has "null" address and port 0 now.
 511         * bind() ensures we have a *particular port* selected by kernel
 512         * and remembered in fd, thus later recv(fd)
 513         * receives only packets sent to this port.
 514         */
 515        xbind(pfd.fd, &local_lsa->u.sa, local_lsa->len);
 516        free(local_lsa);
 517        /* Make read/writes know the destination */
 518        xconnect(pfd.fd, &ns->lsa->u.sa, ns->lsa->len);
 519        ndelay_on(pfd.fd);
 520
 521        retry_interval = timeout / G.default_retry;
 522        tstart = tcur = monotonic_ms();
 523        goto send;
 524
 525        while (tcur - tstart < timeout) {
 526                int qn;
 527                int recvlen;
 528
 529                if (tcur - tsent >= retry_interval) {
 530 send:
 531                        for (qn = 0; qn < G.query_count; qn++) {
 532                                if (G.query[qn].qlen == 0)
 533                                        continue; /* this one was replied already */
 534
 535                                if (write(pfd.fd, G.query[qn].query, G.query[qn].qlen) < 0) {
 536                                        bb_perror_msg("write to '%s'", ns->name);
 537                                        n_replies = -1; /* "no go, try next server" */
 538                                        goto ret;
 539                                }
 540                                dbg("query %u sent\n", qn);
 541                        }
 542                        tsent = tcur;
 543                        servfail_retry = 2 * G.query_count;
 544                }
 545
 546                /* Wait for a response, or until time to retry */
 547                if (poll(&pfd, 1, retry_interval - (tcur - tsent)) <= 0)
 548                        goto next;
 549
 550                recvlen = read(pfd.fd, reply, sizeof(reply));
 551                if (recvlen < 0) {
 552                        bb_perror_msg("read");
 553 next:
 554                        tcur = monotonic_ms();
 555                        continue;
 556                }
 557
 558                if (ns->replies++ == 0) {
 559                        printf("Server:\t\t%s\n", ns->name);
 560                        printf("Address:\t%s\n\n",
 561                                auto_string(xmalloc_sockaddr2dotted(&ns->lsa->u.sa))
 562                        );
 563                        /* In "Address", bind-utils-9.11.3 show port after a hash: "1.2.3.4#53" */
 564                        /* Should we do the same? */
 565                }
 566
 567                /* Non-identifiable packet */
 568                if (recvlen < 4) {
 569                        dbg("read is too short:%d\n", recvlen);
 570                        goto next;
 571                }
 572
 573                /* Find which query this answer goes with, if any */
 574//              qn = save_idx;
 575                qn = 0;
 576                for (;;) {
 577                        if (memcmp(reply, G.query[qn].query, 2) == 0) {
 578                                dbg("response matches query %u\n", qn);
 579                                break;
 580                        }
 581                        if (++qn >= G.query_count) {
 582                                dbg("response does not match any query\n");
 583                                goto next;
 584                        }
 585                }
 586
 587                if (G.query[qn].qlen == 0) {
 588                        dbg("dropped duplicate response to query %u\n", qn);
 589                        goto next;
 590                }
 591
 592                rcode = reply[3] & 0x0f;
 593                dbg("query %u rcode:%s\n", qn, rcodes[rcode]);
 594
 595                /* Retry immediately on SERVFAIL */
 596                if (rcode == 2) {
 597                        //UNUSED: ns->failures++;
 598                        if (servfail_retry) {
 599                                servfail_retry--;
 600                                write(pfd.fd, G.query[qn].query, G.query[qn].qlen);
 601                                dbg("query %u resent\n", qn);
 602                                goto next;
 603                        }
 604                }
 605
 606                /* Process reply */
 607                G.query[qn].qlen = 0; /* flag: "reply received" */
 608                tcur = monotonic_ms();
 609#if 1
 610                if (option_mask32 & OPT_debug) {
 611                        printf("Query #%d completed in %ums:\n", qn, tcur - tstart);
 612                }
 613                if (rcode != 0) {
 614                        printf("** server can't find %s: %s\n",
 615                                        G.query[qn].name, rcodes[rcode]);
 616                        G.exitcode = EXIT_FAILURE;
 617                } else {
 618                        if (parse_reply(reply, recvlen) < 0) {
 619                                printf("*** Can't find %s: Parse error\n", G.query[qn].name);
 620                                G.exitcode = EXIT_FAILURE;
 621                        }
 622                }
 623                bb_putchar('\n');
 624                n_replies++;
 625                if (n_replies >= G.query_count)
 626                        goto ret;
 627#else
 628//used to store replies and process them later
 629                G.query[qn].latency = tcur - tstart;
 630                n_replies++;
 631                if (qn != save_idx) {
 632                        /* "wrong" receive buffer, move to correct one */
 633                        memcpy(G.query[qn].reply, G.query[save_idx].reply, recvlen);
 634                        continue;
 635                }
 636                /* G.query[0..save_idx] have replies, move to next one, if exists */
 637                for (;;) {
 638                        save_idx++;
 639                        if (save_idx >= G.query_count)
 640                                goto ret; /* all are full: we have all results */
 641                        if (!G.query[save_idx].rlen)
 642                                break; /* this one is empty */
 643                }
 644#endif
 645        } /* while() */
 646
 647 ret:
 648        close(pfd.fd);
 649
 650        return n_replies;
 651}
 652
 653static void add_ns(const char *addr)
 654{
 655        struct ns *ns;
 656        unsigned count;
 657
 658        dbg("%s: addr:'%s'\n", __func__, addr);
 659
 660        count = G.serv_count++;
 661
 662        G.server = xrealloc_vector(G.server, /*8=2^3:*/ 3, count);
 663        ns = &G.server[count];
 664        ns->name = addr;
 665        ns->lsa = xhost2sockaddr(addr, G.default_port);
 666        /*ns->replies = 0; - already is */
 667        /*ns->failures = 0; - already is */
 668}
 669
 670static void parse_resolvconf(void)
 671{
 672        FILE *resolv;
 673
 674        resolv = fopen("/etc/resolv.conf", "r");
 675        if (resolv) {
 676                char line[512]; /* "search" is defined to be up to 256 chars */
 677
 678                while (fgets(line, sizeof(line), resolv)) {
 679                        char *p, *arg;
 680
 681                        p = strtok(line, " \t\n");
 682                        if (!p)
 683                                continue;
 684                        dbg("resolv_key:'%s'\n", p);
 685                        arg = strtok(NULL, "\n");
 686                        dbg("resolv_arg:'%s'\n", arg);
 687                        if (!arg)
 688                                continue;
 689
 690                        if (strcmp(p, "domain") == 0) {
 691                                /* domain DOM */
 692                                if (!G.have_search_directive)
 693                                        goto set_search;
 694                                continue;
 695                        }
 696                        if (strcmp(p, "search") == 0) {
 697                                /* search DOM1 DOM2... */
 698                                G.have_search_directive = 1;
 699 set_search:
 700                                free(G.search);
 701                                G.search = xstrdup(arg);
 702                                dbg("search='%s'\n", G.search);
 703                                continue;
 704                        }
 705
 706                        if (strcmp(p, "nameserver") != 0)
 707                                continue;
 708
 709                        /* nameserver DNS */
 710                        add_ns(xstrdup(arg));
 711                }
 712
 713                fclose(resolv);
 714        }
 715
 716        if (!G.search) {
 717                /* default search domain is domain part of hostname */
 718                char *h = safe_gethostname();
 719                char *d = strchr(h, '.');
 720                if (d) {
 721                        G.search = d + 1;
 722                        dbg("search='%s' (from hostname)\n", G.search);
 723                }
 724                /* else free(h); */
 725        }
 726
 727        /* Cater for case of "domain ." in resolv.conf */
 728        if (G.search && LONE_CHAR(G.search, '.'))
 729                G.search = NULL;
 730}
 731
 732static void add_query(int type, const char *dname)
 733{
 734        struct query *new_q;
 735        unsigned count;
 736        ssize_t qlen;
 737
 738        count = G.query_count++;
 739
 740        G.query = xrealloc_vector(G.query, /*4=2^2:*/ 2, count);
 741        new_q = &G.query[count];
 742
 743        dbg("new query#%u type %u for '%s'\n", count, type, dname);
 744        new_q->name = dname;
 745
 746        qlen = res_mkquery(QUERY, dname, C_IN, type,
 747                        /*data:*/ NULL, /*datalen:*/ 0,
 748                        /*newrr:*/ NULL,
 749                        new_q->query, sizeof(new_q->query)
 750        );
 751        new_q->qlen = qlen;
 752}
 753
 754static void add_query_with_search(int type, const char *dname)
 755{
 756        char *s;
 757
 758        if (type == T_PTR || !G.search || strchr(dname, '.')) {
 759                add_query(type, dname);
 760                return;
 761        }
 762
 763        s = G.search;
 764        for (;;) {
 765                char *fullname, *e;
 766
 767                e = skip_non_whitespace(s);
 768                fullname = xasprintf("%s.%.*s", dname, (int)(e - s), s);
 769                add_query(type, fullname);
 770                s = skip_whitespace(e);
 771                if (!*s)
 772                        break;
 773        }
 774}
 775
 776static char *make_ptr(const char *addrstr)
 777{
 778        unsigned char addr[16];
 779
 780#if ENABLE_FEATURE_IPV6
 781        if (inet_pton(AF_INET6, addrstr, addr)) {
 782                if (memcmp(addr, v4_mapped, 12) != 0) {
 783                        int i;
 784                        char resbuf[80];
 785                        char *ptr = resbuf;
 786                        for (i = 0; i < 16; i++) {
 787                                *ptr++ = 0x20 | bb_hexdigits_upcase[(unsigned char)addr[15 - i] & 0xf];
 788                                *ptr++ = '.';
 789                                *ptr++ = 0x20 | bb_hexdigits_upcase[(unsigned char)addr[15 - i] >> 4];
 790                                *ptr++ = '.';
 791                        }
 792                        strcpy(ptr, "ip6.arpa");
 793                        return xstrdup(resbuf);
 794                }
 795                return xasprintf("%u.%u.%u.%u.in-addr.arpa",
 796                                addr[15], addr[14], addr[13], addr[12]);
 797        }
 798#endif
 799
 800        if (inet_pton(AF_INET, addrstr, addr)) {
 801                return xasprintf("%u.%u.%u.%u.in-addr.arpa",
 802                        addr[3], addr[2], addr[1], addr[0]);
 803        }
 804
 805        return NULL;
 806}
 807
 808int nslookup_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 809int nslookup_main(int argc UNUSED_PARAM, char **argv)
 810{
 811        unsigned types;
 812        int rc;
 813        int err;
 814
 815        INIT_G();
 816
 817        /* manpage: "Options can also be specified on the command line
 818         * if they precede the arguments and are prefixed with a hyphen."
 819         */
 820        types = 0;
 821        argv++;
 822        for (;;) {
 823                const char *options =
 824// bind-utils-9.11.3 accept these:
 825// class=   cl=
 826// type=    ty= querytype= query= qu= q=
 827// domain=  do=
 828// port=    po=
 829// timeout= t=
 830// retry=   ret=
 831// ndots=
 832// recurse
 833// norecurse
 834// defname
 835// nodefname
 836// vc
 837// novc
 838// debug
 839// nodebug
 840// d2
 841// nod2
 842// search
 843// nosearch
 844// sil
 845// fail
 846// nofail
 847// ver (prints version and exits)
 848                        "type\0"      /* 0 */
 849                        "querytype\0" /* 1 */
 850                        "port\0"      /* 2 */
 851                        "retry\0"     /* 3 */
 852                        "debug\0"     /* 4 */
 853                        "t\0" /* disambiguate with "type": else -t=2 fails */
 854                        "timeout\0"   /* 6 */
 855                        "";
 856                int i;
 857                char *arg;
 858                char *val;
 859
 860                if (!*argv)
 861                        bb_show_usage();
 862                if (argv[0][0] != '-')
 863                        break;
 864
 865                /* Separate out "=val" part */
 866                arg = (*argv++) + 1;
 867                val = strchrnul(arg, '=');
 868                if (*val)
 869                        *val++ = '\0';
 870
 871                i = index_in_substrings(options, arg);
 872                //bb_error_msg("i:%d arg:'%s' val:'%s'", i, arg, val);
 873                if (i < 0)
 874                        bb_show_usage();
 875
 876                if (i <= 1) {
 877                        for (i = 0;; i++) {
 878                                if (i == ARRAY_SIZE(qtypes))
 879                                        bb_error_msg_and_die("invalid query type \"%s\"", val);
 880                                if (strcasecmp(qtypes[i].name, val) == 0)
 881                                        break;
 882                        }
 883                        types |= (1 << i);
 884                        continue;
 885                }
 886                if (i == 2) {
 887                        G.default_port = xatou_range(val, 1, 0xffff);
 888                }
 889                if (i == 3) {
 890                        G.default_retry = xatou_range(val, 1, INT_MAX);
 891                }
 892                if (i == 4) {
 893                        option_mask32 |= OPT_debug;
 894                }
 895                if (i > 4) {
 896                        G.default_timeout = xatou_range(val, 1, INT_MAX / 1000);
 897                }
 898        }
 899
 900        /* Use given DNS server if present */
 901        if (argv[1]) {
 902                if (argv[2])
 903                        bb_show_usage();
 904                add_ns(argv[1]);
 905        } else {
 906                parse_resolvconf();
 907                /* Fall back to localhost if we could not find NS in resolv.conf */
 908                if (G.serv_count == 0)
 909                        add_ns("127.0.0.1");
 910        }
 911
 912        if (types == 0) {
 913                /* No explicit type given, guess query type.
 914                 * If we can convert the domain argument into a ptr (means that
 915                 * inet_pton() could read it) we assume a PTR request, else
 916                 * we issue A+AAAA queries and switch to an output format
 917                 * mimicking the one of the traditional nslookup applet.
 918                 */
 919                char *ptr;
 920
 921                ptr = make_ptr(argv[0]);
 922                if (ptr) {
 923                        add_query(T_PTR, ptr);
 924                } else {
 925                        add_query_with_search(T_A, argv[0]);
 926#if ENABLE_FEATURE_IPV6
 927                        add_query_with_search(T_AAAA, argv[0]);
 928#endif
 929                }
 930        } else {
 931                int c;
 932                for (c = 0; c < ARRAY_SIZE(qtypes); c++) {
 933                        if (types & (1 << c))
 934                                add_query_with_search(qtypes[c].type, argv[0]);
 935                }
 936        }
 937
 938        for (rc = 0; rc < G.serv_count;) {
 939                int c;
 940
 941                c = send_queries(&G.server[rc]);
 942                if (c > 0) {
 943                        /* more than zero replies received */
 944#if 0 /* which version does this? */
 945                        if (option_mask32 & OPT_debug) {
 946                                printf("Replies:\t%d\n", G.server[rc].replies);
 947                                printf("Failures:\t%d\n\n", G.server[rc].failures);
 948                        }
 949#endif
 950                        break;
 951//FIXME: we "break" even though some queries may still be not answered, and other servers may know them?
 952                }
 953                /* c = 0: timed out waiting for replies */
 954                /* c < 0: error (message already printed) */
 955                rc++;
 956                if (rc >= G.serv_count) {
 957//
 958// NB: bind-utils-9.11.3 behavior (all to stdout, not stderr):
 959//
 960// $ nslookup gmail.com 8.8.8.8
 961// ;; connection timed out; no servers could be reached
 962//
 963// Using TCP mode:
 964// $ nslookup -vc gmail.com 8.8.8.8; echo EXITCODE:$?
 965//     <~10 sec>
 966// ;; Connection to 8.8.8.8#53(8.8.8.8) for gmail.com failed: timed out.
 967//     <~10 sec>
 968// ;; Connection to 8.8.8.8#53(8.8.8.8) for gmail.com failed: timed out.
 969//     <~10 sec>
 970// ;; connection timed out; no servers could be reached
 971// ;; Connection to 8.8.8.8#53(8.8.8.8) for gmail.com failed: timed out.
 972//     <empty line>
 973// EXITCODE:1
 974// $ _
 975                        printf(";; connection timed out; no servers could be reached\n\n");
 976                        return EXIT_FAILURE;
 977                }
 978        }
 979
 980        err = 0;
 981        for (rc = 0; rc < G.query_count; rc++) {
 982                if (G.query[rc].qlen) {
 983                        printf("*** Can't find %s: No answer\n", G.query[rc].name);
 984                        err = 1;
 985                }
 986        }
 987        if (err) /* should this affect exicode too? */
 988                bb_putchar('\n');
 989
 990        if (ENABLE_FEATURE_CLEAN_UP) {
 991                free(G.server);
 992                free(G.query);
 993        }
 994
 995        return G.exitcode;
 996}
 997
 998#endif
 999