This depmod script will create a modules.dep file, just like the depmod
[oweals/busybox.git] / nslookup.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini nslookup implementation for busybox
4  *
5  * Copyright (C) 1999,2000,2001 by Lineo, inc.
6  * Written by John Beppu <beppu@lineo.com>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21  *
22  */
23
24 #include <ctype.h>
25 #include <errno.h>
26 #include <stdio.h>
27 #include <string.h>
28 #include <stdlib.h>
29
30 #include <netdb.h>
31 #include <sys/socket.h>
32 #include <sys/types.h>
33 #include <netinet/in.h>
34 #include <resolv.h>
35 #include <arpa/inet.h>
36 #include "busybox.h"
37
38 /*
39  |  I'm only implementing non-interactive mode;
40  |  I totally forgot nslookup even had an interactive mode.
41  |
42  |  [ TODO ]
43  |  + find out how to use non-default name servers
44  */
45
46 /* only works for IPv4 */
47 static int addr_fprint(char *addr)
48 {
49         u_int8_t split[4];
50         u_int32_t ip;
51         u_int32_t *x = (u_int32_t *) addr;
52
53         ip = ntohl(*x);
54         split[0] = (ip & 0xff000000) >> 24;
55         split[1] = (ip & 0x00ff0000) >> 16;
56         split[2] = (ip & 0x0000ff00) >> 8;
57         split[3] = (ip & 0x000000ff);
58         printf("%d.%d.%d.%d", split[0], split[1], split[2], split[3]);
59         return 0;
60 }
61
62 /* takes the NULL-terminated array h_addr_list, and
63  * prints its contents appropriately
64  */
65 static int addr_list_fprint(char **h_addr_list)
66 {
67         int i, j;
68         char *addr_string = (h_addr_list[1])
69                 ? "Addresses: " : "Address:   ";
70
71         printf("%s ", addr_string);
72         for (i = 0, j = 0; h_addr_list[i]; i++, j++) {
73                 addr_fprint(h_addr_list[i]);
74
75                 /* real nslookup does this */
76                 if (j == 4) {
77                         if (h_addr_list[i + 1]) {
78                                 printf("\n          ");
79                         }
80                         j = 0;
81                 } else {
82                         if (h_addr_list[i + 1]) {
83                                 printf(", ");
84                         }
85                 }
86
87         }
88         printf("\n");
89         return 0;
90 }
91
92 /* print the results as nslookup would */
93 static struct hostent *hostent_fprint(struct hostent *host)
94 {
95         if (host) {
96                 printf("Name:       %s\n", host->h_name);
97                 addr_list_fprint(host->h_addr_list);
98         } else {
99                 printf("*** Unknown host\n");
100         }
101         return host;
102 }
103
104 /* changes a c-string matching the perl regex \d+\.\d+\.\d+\.\d+
105  * into a u_int32_t
106  */
107 static u_int32_t str_to_addr(const char *addr)
108 {
109         u_int32_t split[4];
110         u_int32_t ip;
111
112         sscanf(addr, "%d.%d.%d.%d",
113                    &split[0], &split[1], &split[2], &split[3]);
114
115         /* assuming sscanf worked */
116         ip = (split[0] << 24) |
117                 (split[1] << 16) | (split[2] << 8) | (split[3]);
118
119         return htonl(ip);
120 }
121
122 /* gethostbyaddr wrapper */
123 static struct hostent *gethostbyaddr_wrapper(const char *address)
124 {
125         struct in_addr addr;
126
127         addr.s_addr = str_to_addr(address);
128         return gethostbyaddr((char *) &addr, 4, AF_INET);       /* IPv4 only for now */
129 }
130
131 #ifdef __UCLIBC__
132 #warning FIXME after fixing uClibc to define struct _res 
133 static inline void server_print(void)
134 {
135        printf("Server:     %s\n", "default");
136        printf("Address:    %s\n\n", "default");
137 }
138 #else
139 /* lookup the default nameserver and display it */
140 static inline void server_print(void)
141 {
142         struct sockaddr_in def = _res.nsaddr_list[0];
143         char *ip = inet_ntoa(def.sin_addr);
144
145         hostent_fprint(gethostbyaddr_wrapper(ip));
146         printf("\n");
147 }
148 #endif  
149
150 /* naive function to check whether char *s is an ip address */
151 static int is_ip_address(const char *s)
152 {
153         while (*s) {
154                 if ((isdigit(*s)) || (*s == '.')) {
155                         s++;
156                         continue;
157                 }
158                 return 0;
159         }
160         return 1;
161 }
162
163 /* ________________________________________________________________________ */
164 int nslookup_main(int argc, char **argv)
165 {
166         struct hostent *host;
167
168         if (argc < 2 || *argv[1]=='-') {
169                 show_usage();
170         }
171
172         res_init();
173         server_print();
174         if (is_ip_address(argv[1])) {
175                 host = gethostbyaddr_wrapper(argv[1]);
176         } else {
177                 host = gethostbyname(argv[1]);
178         }
179         hostent_fprint(host);
180         return EXIT_SUCCESS;
181 }
182
183 /* $Id: nslookup.c,v 1.24 2001/07/06 17:51:29 andersen Exp $ */