1 /*
2 * Copyright (c) 1980, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 * @(#) Copyright (c) 1980, 1993 The Regents of the University of California. All rights reserved.
30 * @(#)whois.c 8.1 (Berkeley) 6/6/93
31 * $FreeBSD: src/usr.bin/whois/whois.c,v 1.15.2.11 2003/02/25 20:59:41 roberto Exp $
32 */
33
34 #include <sys/types.h>
35 #include <sys/socket.h>
36
37 #include <arpa/inet.h>
38 #include <netinet/in.h>
39
40 #include <ctype.h>
41 #include <err.h>
42 #include <netdb.h>
43 #include <stdarg.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <sysexits.h>
48 #include <unistd.h>
49
50 #define ABUSEHOST "whois.abuse.net"
51 #define NICHOST "whois.crsnic.net"
52 #define INICHOST "whois.networksolutions.com"
53 #define DNICHOST "whois.nic.mil"
54 #define GNICHOST "whois.nic.gov"
55 #define ANICHOST "whois.arin.net"
56 #define LNICHOST "whois.lacnic.net"
57 #define KNICHOST "whois.krnic.net"
58 #define RNICHOST "whois.ripe.net"
59 #define PNICHOST "whois.apnic.net"
60 #define RUNICHOST "whois.ripn.net"
61 #define MNICHOST "whois.ra.net"
62 #define QNICHOST_TAIL ".whois-servers.net"
63 #define SNICHOST "whois.6bone.net"
64 #define IANAHOST "whois.iana.org"
65 #define GERMNICHOST "de.whois-servers.net"
66 #define BNICHOST "whois.registro.br"
67 #define WHOIS_SERVER_ID "Whois Server: "
68 #define WHOIS_ORG_SERVER_ID "Registrant Street1:Whois Server:"
69
70 #define DEFAULT_PORT "whois"
71 #define WHOIS_RECURSE 0x01
72 #define WHOIS_QUICK 0x02
73
74 #define ishost(h) (isalnum(h) || h == '.' || h == '-')
75
76 static const char *ip_whois[] = { LNICHOST, RNICHOST, PNICHOST, BNICHOST, NULL };
77 static const char *port = DEFAULT_PORT;
78
79 static char *choose_server(const char *);
80 static struct addrinfo *gethostinfo(const char *, int);
81 static void s_asprintf(char **, const char *, ...) __printflike(2, 3);
82 static void usage(void);
83 static void whois(const char *, const char *, int);
84
85 int
main(int argc,char * argv[])86 main(int argc, char *argv[])
87 {
88 const char *country, *host;
89 char *qnichost;
90 int ch, flags, use_qnichost;
91
92 #ifdef SOCKS
93 SOCKSinit(argv[0]);
94 #endif
95
96 country = host = qnichost = NULL;
97 flags = use_qnichost = 0;
98 while ((ch = getopt(argc, argv, "aAbc:dgh:ilkImp:QrR6")) != -1) {
99 switch (ch) {
100 case 'a':
101 host = ANICHOST;
102 break;
103 case 'A':
104 host = PNICHOST;
105 break;
106 case 'b':
107 host = ABUSEHOST;
108 break;
109 case 'c':
110 country = optarg;
111 break;
112 case 'd':
113 host = DNICHOST;
114 break;
115 case 'g':
116 host = GNICHOST;
117 break;
118 case 'h':
119 host = optarg;
120 break;
121 case 'i':
122 host = INICHOST;
123 break;
124 case 'I':
125 host = IANAHOST;
126 break;
127 case 'k':
128 host = KNICHOST;
129 break;
130 case 'l':
131 host = LNICHOST;
132 break;
133 case 'm':
134 host = MNICHOST;
135 break;
136 case 'p':
137 port = optarg;
138 break;
139 case 'Q':
140 flags |= WHOIS_QUICK;
141 break;
142 case 'r':
143 host = RNICHOST;
144 break;
145 case 'R':
146 host = RUNICHOST;
147 break;
148 case '6':
149 host = SNICHOST;
150 break;
151 case '?':
152 default:
153 usage();
154 /* NOTREACHED */
155 }
156 }
157 argc -= optind;
158 argv += optind;
159
160 if (!argc || (country != NULL && host != NULL))
161 usage();
162
163 /*
164 * If no host or country is specified determine the top level domain
165 * from the query. If the TLD is a number, query ARIN. Otherwise, use
166 * TLD.whois-server.net. If the domain does not contain '.', fall
167 * back to NICHOST.
168 */
169 if (host == NULL && country == NULL) {
170 use_qnichost = 1;
171 host = NICHOST;
172 if (!(flags & WHOIS_QUICK))
173 flags |= WHOIS_RECURSE;
174 }
175 while (argc-- > 0) {
176 if (country != NULL) {
177 s_asprintf(&qnichost, "%s%s", country, QNICHOST_TAIL);
178 whois(*argv, qnichost, flags);
179 } else if (use_qnichost)
180 if ((qnichost = choose_server(*argv)) != NULL)
181 whois(*argv, qnichost, flags);
182 if (qnichost == NULL)
183 whois(*argv, host, flags);
184 free(qnichost);
185 qnichost = NULL;
186 argv++;
187 }
188 exit(0);
189 }
190
191 /*
192 * This function will remove any trailing periods from domain, after which it
193 * returns a pointer to newly allocated memory containing the whois server to
194 * be queried, or a NULL if the correct server couldn't be determined. The
195 * caller must remember to free(3) the allocated memory.
196 */
197 static char *
choose_server(const char * domain)198 choose_server(const char *domain)
199 {
200 char *pos, *retval;
201
202 for (pos = strchr(domain, '\0'); pos > domain && *--pos == '.';)
203 *pos = '\0';
204 if (*domain == '\0')
205 errx(EX_USAGE, "can't search for a null string");
206 while (pos > domain && *pos != '.')
207 --pos;
208 if (pos <= domain)
209 return (NULL);
210 if (isdigit(*++pos))
211 s_asprintf(&retval, "%s", ANICHOST);
212 else
213 s_asprintf(&retval, "%s%s", pos, QNICHOST_TAIL);
214 return (retval);
215 }
216
217 static struct addrinfo *
gethostinfo(const char * host,int exit_on_error)218 gethostinfo(const char *host, int exit_on_error)
219 {
220 struct addrinfo hints, *res;
221 int error;
222
223 memset(&hints, 0, sizeof(hints));
224 hints.ai_flags = 0;
225 hints.ai_family = AF_UNSPEC;
226 hints.ai_socktype = SOCK_STREAM;
227 error = getaddrinfo(host, port, &hints, &res);
228 if (error) {
229 if (error == EAI_SERVICE)
230 warnx("bad port: %s", port);
231 else
232 warnx("%s: %s", host, gai_strerror(error));
233 if (exit_on_error)
234 exit(EX_NOHOST);
235 return (NULL);
236 }
237 return (res);
238 }
239
240 /*
241 * Wrapper for asprintf(3) that exits on error.
242 */
243 static void
s_asprintf(char ** ret,const char * format,...)244 s_asprintf(char **ret, const char *format, ...)
245 {
246
247 va_list ap;
248
249 va_start(ap, format);
250 if (vasprintf(ret, format, ap) == -1) {
251 va_end(ap);
252 err(EX_OSERR, "vasprintf()");
253 }
254 va_end(ap);
255 }
256
257 static void
whois(const char * query,const char * hostname,int flags)258 whois(const char *query, const char *hostname, int flags)
259 {
260 FILE *sfi, *sfo;
261 struct addrinfo *hostres, *res;
262 char *buf, *host, *nhost, *p;
263 int i, s = -1;
264 size_t c, len;
265
266 hostres = gethostinfo(hostname, 1);
267 for (res = hostres; res; res = res->ai_next) {
268 s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
269 if (s < 0)
270 continue;
271 if (connect(s, res->ai_addr, res->ai_addrlen) == 0)
272 break;
273 close(s);
274 }
275 freeaddrinfo(hostres);
276 if (res == NULL)
277 err(EX_OSERR, "connect()");
278
279 sfi = fdopen(s, "r");
280 sfo = fdopen(s, "w");
281 if (sfi == NULL || sfo == NULL)
282 err(EX_OSERR, "fdopen()");
283 if (strcmp(hostname, GERMNICHOST) == 0) {
284 fprintf(sfo, "-T dn,ace -C US-ASCII %s\r\n", query);
285 } else {
286 fprintf(sfo, "%s\r\n", query);
287 }
288 fflush(sfo);
289 nhost = NULL;
290 while ((buf = fgetln(sfi, &len)) != NULL) {
291 while (len > 0 && isspace(buf[len - 1]))
292 buf[--len] = '\0';
293 printf("%.*s\n", (int)len, buf);
294
295 if ((flags & WHOIS_RECURSE) && nhost == NULL) {
296 host = strnstr(buf, WHOIS_SERVER_ID, len);
297 if (host != NULL) {
298 host += sizeof(WHOIS_SERVER_ID) - 1;
299 for (p = host; p < buf + len; p++) {
300 if (!ishost(*p)) {
301 *p = '\0';
302 break;
303 }
304 }
305 s_asprintf(&nhost, "%.*s",
306 (int)(buf + len - host), host);
307 } else if ((host =
308 strnstr(buf, WHOIS_ORG_SERVER_ID, len)) != NULL) {
309 host += sizeof(WHOIS_ORG_SERVER_ID) - 1;
310 for (p = host; p < buf + len; p++) {
311 if (!ishost(*p)) {
312 *p = '\0';
313 break;
314 }
315 }
316 s_asprintf(&nhost, "%.*s",
317 (int)(buf + len - host), host);
318 } else if (strcmp(hostname, ANICHOST) == 0) {
319 for (c = 0; c <= len; c++)
320 buf[c] = tolower((int)buf[c]);
321 for (i = 0; ip_whois[i] != NULL; i++) {
322 if (strnstr(buf, ip_whois[i], len) !=
323 NULL) {
324 s_asprintf(&nhost, "%s",
325 ip_whois[i]);
326 break;
327 }
328 }
329 }
330 }
331 }
332 if (nhost != NULL) {
333 whois(query, nhost, 0);
334 free(nhost);
335 }
336 }
337
338 static void
usage(void)339 usage(void)
340 {
341 fprintf(stderr,
342 "usage: whois [-aAbdgilkImQrR6] [-c country-code | -h hostname] "
343 "[-p port] name ...\n");
344 exit(EX_USAGE);
345 }
346