networking/nslookup.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini nslookup implementation for busybox
   4  *
   5  * Copyright (C) 1999,2000 by Lineo, inc. and John Beppu
   6  * Copyright (C) 1999,2000,2001 by John Beppu <beppu@codepoet.org>
   7  *
   8  * Correct default name server display and explicit name server option
   9  * added by Ben Zeckel <bzeckel@hmc.edu> June 2001
  10  *
  11  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  12  */
  13
  14 #include <resolv.h>
  15 #include "busybox.h"
  16
  17 /*
  18  *  I'm only implementing non-interactive mode;
  19  *  I totally forgot nslookup even had an interactive mode.
  20  */
  21
  22 /* Examples of 'standard' nslookup output
  23  * $ nslookup yahoo.com
  24  * Server:         128.193.0.10
  25  * Address:        128.193.0.10#53
  26  *
  27  * Non-authoritative answer:
  28  * Name:   yahoo.com
  29  * Address: 216.109.112.135
  30  * Name:   yahoo.com
  31  * Address: 66.94.234.13
  32  *
  33  * $ nslookup 204.152.191.37
  34  * Server:         128.193.4.20
  35  * Address:        128.193.4.20#53
  36  *
  37  * Non-authoritative answer:
  38  * 37.191.152.204.in-addr.arpa     canonical name = 37.32-27.191.152.204.in-addr.arpa.
  39  * 37.32-27.191.152.204.in-addr.arpa       name = zeus-pub2.kernel.org.
  40  *
  41  * Authoritative answers can be found from:
  42  * 32-27.191.152.204.in-addr.arpa  nameserver = ns1.kernel.org.
  43  * 32-27.191.152.204.in-addr.arpa  nameserver = ns2.kernel.org.
  44  * 32-27.191.152.204.in-addr.arpa  nameserver = ns3.kernel.org.
  45  * ns1.kernel.org  internet address = 140.211.167.34
  46  * ns2.kernel.org  internet address = 204.152.191.4
  47  * ns3.kernel.org  internet address = 204.152.191.36
  48  */
  49
  50 /*static int sockaddr_to_dotted(struct sockaddr *saddr, char *buf, int buflen)
  51 {
  52         if (buflen <= 0) return -1;
  53         buf[0] = '\0';
  54         if (saddr->sa_family == AF_INET) {
  55                 inet_ntop(AF_INET, &((struct sockaddr_in*)saddr)->sin_addr, buf, buflen);
  56                 return 0;
  57         }
  58         if (saddr->sa_family == AF_INET6) {
  59                 inet_ntop(AF_INET6, &((struct sockaddr_in6*)saddr)->sin6_addr, buf, buflen);
  60                 return 0;
  61         }
  62         return -1;
  63 }
  64 */
  65
  66 static int print_host(const char *hostname, const char *header)
  67 {
  68 #if 0
  69         char str[128];  /* IPv6 address will fit, hostnames hopefully too */
  70         struct addrinfo *result = NULL;
  71         int rc;
  72         struct addrinfo hint;
  73
  74         memset(&hint, 0 , sizeof(hint));
  75         /* hint.ai_family = AF_UNSPEC; - zero anyway */
  76         /* Needed. Or else we will get each address thrice (or more)
  77          * for each possible socket type (tcp,udp,raw...): */
  78         hint.ai_socktype = SOCK_STREAM;
  79         // hint.ai_flags = AI_CANONNAME;
  80         rc = getaddrinfo(hostname, NULL /*service*/, &hint, &result);
  81
  82         if (!rc) {
  83                 struct addrinfo *cur = result;
  84                 // printf("%s\n", cur->ai_canonname); ?
  85                 while (cur) {
  86                         sockaddr_to_dotted(cur->ai_addr, str, sizeof(str));
  87                         printf("%s  %s\nAddress: %s", header, hostname, str);
  88                         str[0] = ' ';
  89                         if (getnameinfo(cur->ai_addr, cur->ai_addrlen, str+1, sizeof(str)-1, NULL, 0, NI_NAMEREQD))
  90                                 str[0] = '\0';
  91                         puts(str);
  92                         cur = cur->ai_next;
  93                 }
  94         } else {
  95                 bb_error_msg("getaddrinfo('%s') failed: %s", hostname, gai_strerror(rc));
  96         }
  97         freeaddrinfo(result);
  98         return (rc != 0);
  99
 100 #else
 101         /* We can't use host2sockaddr() - we want to get ALL addresses,
 102          * not just one */
 103
 104         struct addrinfo *result = NULL;
 105         int rc;
 106         struct addrinfo hint;
 107
 108         memset(&hint, 0 , sizeof(hint));
 109         /* hint.ai_family = AF_UNSPEC; - zero anyway */
 110         /* Needed. Or else we will get each address thrice (or more)
 111          * for each possible socket type (tcp,udp,raw...): */
 112         hint.ai_socktype = SOCK_STREAM;
 113         // hint.ai_flags = AI_CANONNAME;
 114         rc = getaddrinfo(hostname, NULL /*service*/, &hint, &result);
 115
 116         if (!rc) {
 117                 struct addrinfo *cur = result;
 118                 unsigned cnt = 0;
 119
 120                 printf("%-10s %s\n", header, hostname);
 121                 // printf("%s\n", cur->ai_canonname); ?
 122                 while (cur) {
 123                         char *dotted, *revhost;
 124                         dotted = xmalloc_sockaddr2dotted_noport(cur->ai_addr, cur->ai_addrlen);
 125                         revhost = xmalloc_sockaddr2hostonly_noport(cur->ai_addr, cur->ai_addrlen);
 126
 127                         printf("Address %u: %s%c", ++cnt, dotted, revhost ? ' ' : '\n');
 128                         if (revhost) {
 129                                 puts(revhost);
 130                                 if (ENABLE_FEATURE_CLEAN_UP)
 131                                         free(revhost);
 132                         }
 133                         if (ENABLE_FEATURE_CLEAN_UP)
 134                                 free(dotted);
 135                         cur = cur->ai_next;
 136                 }
 137         } else {
 138 #if ENABLE_VERBOSE_RESOLUTION_ERRORS
 139                 bb_error_msg("getaddrinfo('%s') failed: %s", hostname, gai_strerror(rc));
 140 #else
 141                 bb_error_msg("can't resolve '%s'", hostname);
 142 #endif
 143         }
 144         if (ENABLE_FEATURE_CLEAN_UP)
 145                 freeaddrinfo(result);
 146         return (rc != 0);
 147 #endif
 148 }
 149
 150
 151 /* lookup the default nameserver and display it */
 152 static void server_print(void)
 153 {
 154         char *server;
 155
 156         server = xmalloc_sockaddr2dotted_noport((struct sockaddr*)&_res.nsaddr_list[0],
 157                         sizeof(struct sockaddr_in));
 158         /* I honestly don't know what to do if DNS server has _IPv6 address_.
 159          * Probably it is listed in
 160          * _res._u._ext_.nsaddrs[MAXNS] (of type "struct sockaddr_in6*" each)
 161          * but how to find out whether resolver uses
 162          * _res.nsaddr_list[] or _res._u._ext_.nsaddrs[], or both?
 163          * Looks like classic design from hell, BIND-grade. Hard to surpass. */
 164         print_host(server, "Server:");
 165         if (ENABLE_FEATURE_CLEAN_UP)
 166                 free(server);
 167         puts("");
 168 }
 169
 170
 171 /* alter the global _res nameserver structure to use
 172    an explicit dns server instead of what is in /etc/resolv.h */
 173 static void set_default_dns(char *server)
 174 {
 175         struct in_addr server_in_addr;
 176
 177         if (inet_pton(AF_INET, server, &server_in_addr) > 0) {
 178                 _res.nscount = 1;
 179                 _res.nsaddr_list[0].sin_addr = server_in_addr;
 180         }
 181 }
 182
 183
 184 int nslookup_main(int argc, char **argv)
 185 {
 186         /* We allow 1 or 2 arguments.
 187          * The first is the name to be looked up and the second is an
 188          * optional DNS server with which to do the lookup.
 189          * More than 3 arguments is an error to follow the pattern of the
 190          * standard nslookup */
 191
 192         if (argc < 2 || *argv[1] == '-' || argc > 3)
 193                 bb_show_usage();
 194
 195         /* initialize DNS structure _res used in printing the default
 196          * name server and in the explicit name server option feature. */
 197         res_init();
 198         /* rfc2133 says this enables IPv6 lookups */
 199         /* (but it also says "may be enabled in /etc/resolv.conf|) */
 200         /*_res.options |= RES_USE_INET6;*/
 201
 202         if(argc == 3)
 203                 set_default_dns(argv[2]);
 204
 205         server_print();
 206         return print_host(argv[1], "Name:");
 207 }