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