1/*      $NetBSD: whois.c,v 1.34 2009/04/14 08:00:48 lukem Exp $   */
2/*	$OpenBSD: whois.c,v 1.28 2003/09/18 22:16:15 fgsch Exp $	*/
3
4/*
5 * Copyright (c) 1980, 1993
6 *	The Regents of the University of California.  All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 * 3. Neither the name of the University nor the names of its contributors
17 *    may be used to endorse or promote products derived from this software
18 *    without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 */
32
33#include <sys/cdefs.h>
34
35#ifndef lint
36__COPYRIGHT("@(#) Copyright (c) 1980, 1993\
37 The Regents of the University of California.  All rights reserved.");
38#endif /* not lint */
39
40#ifndef lint
41#if 0
42static const char sccsid[] = "@(#)whois.c	8.1 (Berkeley) 6/6/93";
43#else
44__RCSID("$NetBSD: whois.c,v 1.34 2009/04/14 08:00:48 lukem Exp $");
45#endif
46#endif /* not lint */
47
48#include <sys/types.h>
49#include <sys/socket.h>
50
51#include <netinet/in.h>
52#include <arpa/inet.h>
53#include <netdb.h>
54
55#include <ctype.h>
56#include <err.h>
57#include <stdio.h>
58#include <stdlib.h>
59#include <string.h>
60#include <unistd.h>
61#include <errno.h>
62
63#define	ANICHOST	"whois.arin.net"
64#define	BNICHOST	"whois.registro.br"
65#define	CNICHOST	"whois.corenic.net"
66#define	DNICHOST	"whois.nic.mil"
67#define	FNICHOST	"whois.afrinic.net"
68#define	GNICHOST	"whois.nic.gov"
69#define	INICHOST	"whois.networksolutions.com"
70#define	LNICHOST	"whois.lacnic.net"
71#define	MNICHOST	"whois.ra.net"
72#define	NICHOST		"whois.crsnic.net"
73#define	PNICHOST	"whois.apnic.net"
74#define	QNICHOST_TAIL	".whois-servers.net"
75#define	RNICHOST	"whois.ripe.net"
76#define	RUNICHOST	"whois.ripn.net"
77#define	SNICHOST	"whois.6bone.net"
78
79#define	WHOIS_PORT	"whois"
80#define	WHOIS_SERVER_ID	"Whois Server:"
81
82#define WHOIS_RECURSE		0x01
83#define WHOIS_QUICK		0x02
84
85static const char *port_whois = WHOIS_PORT;
86static const char *ip_whois[] =
87    { LNICHOST, RNICHOST, PNICHOST, FNICHOST, BNICHOST, NULL };
88
89static void usage(void) __dead;
90static int whois(const char *, const char *, const char *, int);
91static const char *choose_server(const char *, const char *);
92
93int
94main(int argc, char *argv[])
95{
96	int ch, flags, rval;
97	const char *host, *name, *country;
98
99#ifdef SOCKS
100	SOCKSinit(argv[0]);
101#endif
102	country = host = NULL;
103	flags = rval = 0;
104	while ((ch = getopt(argc, argv, "6Aac:dfgh:ilmp:qQRr")) != -1)
105		switch(ch) {
106		case 'a':
107			host = ANICHOST;
108			break;
109		case 'A':
110			host = PNICHOST;
111			break;
112		case 'c':
113			country = optarg;
114			break;
115		case 'd':
116			host = DNICHOST;
117			break;
118		case 'f':
119			host = FNICHOST;
120			break;
121		case 'g':
122			host = GNICHOST;
123			break;
124		case 'h':
125			host = optarg;
126			break;
127		case 'i':
128			host = INICHOST;
129			break;
130		case 'l':
131			host = LNICHOST;
132			break;
133		case 'm':
134			host = MNICHOST;
135			break;
136		case 'p':
137			port_whois = optarg;
138			break;
139		case 'q':
140			/* deprecated, now the default */
141			break;
142		case 'Q':
143			flags |= WHOIS_QUICK;
144			break;
145		case 'r':
146			host = RNICHOST;
147			break;
148		case 'R':
149			host = RUNICHOST;
150			break;
151		case '6':
152			host = SNICHOST;
153			break;
154		default:
155			usage();
156		}
157	argc -= optind;
158	argv += optind;
159
160	if (!argc || (country != NULL && host != NULL))
161		usage();
162
163	if (host == NULL && country == NULL && !(flags & WHOIS_QUICK))
164		flags |= WHOIS_RECURSE;
165	for (name = *argv; (name = *argv) != NULL; argv++)
166		rval += whois(name, host ? host : choose_server(name, country),
167		    port_whois, flags);
168	return rval;
169}
170
171static int
172whois(const char *query, const char *server, const char *port, int flags)
173{
174	FILE *sfi, *sfo;
175	char *buf, *p, *nhost, *nbuf = NULL;
176	size_t len;
177	int i, s, error;
178	const char *reason = NULL, *fmt;
179	struct addrinfo hints, *res, *ai;
180
181	(void)memset(&hints, 0, sizeof(hints));
182	hints.ai_flags = 0;
183	hints.ai_family = AF_UNSPEC;
184	hints.ai_socktype = SOCK_STREAM;
185	error = getaddrinfo(server, port, &hints, &res);
186	if (error) {
187		warnx("%s: %s", server, gai_strerror(error));
188		return (1);
189	}
190
191	for (s = -1, ai = res; ai != NULL; ai = ai->ai_next) {
192		s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
193		if (s < 0) {
194			error = errno;
195			reason = "socket";
196			continue;
197		}
198		if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0) {
199			error = errno;
200			reason = "connect";
201			close(s);
202			s = -1;
203			continue;
204		}
205		break;	/*okay*/
206	}
207	if (s < 0) {
208		if (reason) {
209			errno = error;
210			warn("%s: %s", server, reason);
211		} else
212			warnx("Unknown error in connection attempt");
213		freeaddrinfo(res);
214		return (1);
215	}
216
217	if (strcmp(server, "whois.denic.de") == 0 ||
218	    strcmp(server, "de.whois-servers.net") == 0)
219		fmt = "-T dn,ace -C ISO-8859-1";
220	else
221		fmt = "";
222
223	sfi = fdopen(s, "r");
224	sfo = fdopen(s, "w");
225	if (sfi == NULL || sfo == NULL)
226		err(1, "fdopen");
227	(void)fprintf(sfo, "%s%s\r\n", fmt, query);
228	(void)fflush(sfo);
229	nhost = NULL;
230	while ((buf = fgetln(sfi, &len)) != NULL) {
231		p = buf + len - 1;
232		if (isspace((unsigned char)*p)) {
233			do
234				*p = '\0';
235			while (p > buf && isspace((unsigned char)*--p));
236		} else {
237			if ((nbuf = malloc(len + 1)) == NULL)
238				err(1, "malloc");
239			(void)memcpy(nbuf, buf, len);
240			nbuf[len] = '\0';
241			buf = nbuf;
242		}
243		(void)puts(buf);
244
245		if (nhost != NULL || !(flags & WHOIS_RECURSE))
246			continue;
247
248		if ((p = strstr(buf, WHOIS_SERVER_ID))) {
249			p += sizeof(WHOIS_SERVER_ID) - 1;
250			while (isblank((unsigned char)*p))
251				p++;
252			if ((len = strcspn(p, " \t\n\r"))) {
253				if ((nhost = malloc(len + 1)) == NULL)
254					err(1, "malloc");
255				(void)memcpy(nhost, p, len);
256				nhost[len] = '\0';
257			}
258		} else if (strcmp(server, ANICHOST) == 0) {
259			for (p = buf; *p != '\0'; p++)
260				*p = tolower((unsigned char)*p);
261			for (i = 0; ip_whois[i] != NULL; i++) {
262				if (strstr(buf, ip_whois[i]) != NULL) {
263					nhost = strdup(ip_whois[i]);
264					if (nhost == NULL)
265						err(1, "strdup");
266					break;
267				}
268			}
269		}
270	}
271	if (nbuf != NULL)
272		free(nbuf);
273
274	if (nhost != NULL) {
275		error = whois(query, nhost, port, 0);
276		free(nhost);
277	}
278	freeaddrinfo(res);
279	(void)fclose(sfi);
280	(void)fclose(sfo);
281	return (error);
282}
283
284/*
285 * If no country is specified determine the top level domain from the query.
286 * If the TLD is a number, query ARIN, otherwise, use TLD.whois-server.net.
287 * If the domain does not contain '.', check to see if it is an NSI handle
288 * (starts with '!') or a CORE handle (COCO-[0-9]+ or COHO-[0-9]+).
289 * Fall back to NICHOST for the non-handle case.
290 */
291static const char *
292choose_server(const char *name, const char *country)
293{
294	static char *server;
295	char *nserver;
296	const char *qhead;
297	char *ep;
298	size_t len;
299
300	if (country != NULL)
301		qhead = country;
302	else if ((qhead = strrchr(name, '.')) == NULL) {
303		if (*name == '!')
304			return (INICHOST);
305		else if ((strncasecmp(name, "COCO-", 5) == 0 ||
306		    strncasecmp(name, "COHO-", 5) == 0) &&
307		    strtol(name + 5, &ep, 10) > 0 && *ep == '\0')
308			return (CNICHOST);
309		else
310			return (NICHOST);
311	} else if (isdigit((unsigned char)*(++qhead)))
312		return (ANICHOST);
313	len = strlen(qhead) + sizeof(QNICHOST_TAIL);
314	if ((nserver = realloc(server, len)) == NULL)
315		err(1, "realloc");
316	server = nserver;
317	(void)strlcpy(server, qhead, len);
318	(void)strlcat(server, QNICHOST_TAIL, len);
319	return (server);
320}
321
322static void
323usage(void)
324{
325	(void)fprintf(stderr,
326	    "usage: %s [-6AadgilmQRr] [-c country-code | -h hostname] "
327		"[-p port] name ...\n", getprogname());
328	exit(1);
329}
330