2 * Copyright (c) 1995, 1996
3 * Bill Paul <wpaul@ctr.columbia.edu>. All rights reserved.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by Bill Paul.
16 * 4. 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.
20 * THIS SOFTWARE IS PROVIDED BY Bill Paul 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 Bill Paul 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
32 * $FreeBSD: src/usr.sbin/ypserv/yp_dnslookup.c,v 1.16.2.1 2002/02/15 00:47:00 des Exp $
33 * $DragonFly: src/usr.sbin/ypserv/yp_dnslookup.c,v 1.2 2003/06/17 04:30:04 dillon Exp $
37 * Do standard and reverse DNS lookups using the resolver library.
38 * Take care of all the dirty work here so the main program only has to
39 * pass us a pointer to an array of characters.
41 * We have to use direct resolver calls here otherwise the YP server
42 * could end up looping by calling itself over and over again until
43 * it disappeared up its own belly button.
46 #include <sys/param.h>
47 #include <sys/socket.h>
49 #include <sys/fcntl.h>
50 #include <sys/queue.h>
51 #include <netinet/in.h>
52 #include <arpa/inet.h>
53 #include <arpa/nameser.h>
64 #include <rpcsvc/yp.h>
65 #include "yp_extern.h"
67 static char *parse(hp)
70 static char result[MAXHOSTNAMELEN * 2];
77 len = 16 + strlen(hp->h_name);
78 for (i = 0; hp->h_aliases[i]; i++)
79 len += strlen(hp->h_aliases[i]) + 1;
82 if (len > sizeof(result))
85 bzero(result, sizeof(result));
87 bcopy(hp->h_addr, &addr, sizeof(struct in_addr));
88 snprintf(result, sizeof(result), "%s %s", inet_ntoa(addr), hp->h_name);
90 for (i = 0; hp->h_aliases[i]; i++) {
92 strcat(result, hp->h_aliases[i]);
95 return ((char *)&result);
98 #define MAXPACKET 1024
104 extern struct hostent *__dns_getanswer(char *, int, char *, int);
106 static CIRCLEQ_HEAD(dns_qhead, circleq_dnsentry) qhead;
108 struct circleq_dnsentry {
111 struct sockaddr_in client_addr;
112 unsigned long ypvers;
116 unsigned short prot_type;
120 CIRCLEQ_ENTRY(circleq_dnsentry) links;
123 static int pending = 0;
125 int yp_init_resolver()
127 CIRCLEQ_INIT(&qhead);
128 if (!(_res.options & RES_INIT) && res_init() == -1) {
129 yp_error("res_init failed");
132 if ((resfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
133 yp_error("couldn't create socket");
136 if (fcntl(resfd, F_SETFL, O_NONBLOCK) == -1) {
137 yp_error("couldn't make resolver socket non-blocking");
143 static struct circleq_dnsentry *yp_malloc_dnsent()
145 register struct circleq_dnsentry *q;
147 q = (struct circleq_dnsentry *)malloc(sizeof(struct circleq_dnsentry));
150 yp_error("failed to malloc() circleq dns entry");
160 static unsigned long yp_send_dns_query(name, type)
171 bzero(buf, sizeof(buf));
173 n = res_mkquery(QUERY,name,C_IN,type,NULL,0,NULL,buf,sizeof(buf));
176 yp_error("res_mkquery failed");
180 hptr = (HEADER *)&buf;
181 id = ntohs(hptr->id);
183 for (ns = 0; ns < _res.nscount; ns++) {
184 rval = sendto(resfd, buf, n, 0,
185 (struct sockaddr *)&_res.nsaddr_list[ns],
186 sizeof(struct sockaddr));
188 yp_error("sendto failed");
196 static struct circleq_dnsentry *yp_find_dnsqent(id, type)
200 register struct circleq_dnsentry *q;
202 for (q = qhead.cqh_first; q != (void *)&qhead; q = q->links.cqe_next) {
218 static void yp_send_dns_reply(q, buf)
219 struct circleq_dnsentry *q;
222 ypresponse result_v1;
223 ypresp_val result_v2;
225 struct sockaddr_in client_addr;
230 * Set up correct reply struct and
231 * XDR filter depending on ypvers.
235 bzero((char *)&result_v2, sizeof(result_v2));
238 result_v2.stat = YP_NOKEY;
240 result_v2.val.valdat_len = strlen(buf);
241 result_v2.val.valdat_val = buf;
242 result_v2.stat = YP_TRUE;
244 result = (char *)&result_v2;
245 xdrfunc = (xdrproc_t)xdr_ypresp_val;
249 * The odds are we will _never_ execute this
250 * particular code, but we include it anyway
251 * for the sake of completeness.
253 bzero((char *)&result_v1, sizeof(result_v1));
254 result_v1.yp_resptype = YPRESP_VAL;
255 # define YPVAL ypresponse_u.yp_resp_valtype
258 result_v1.YPVAL.stat = YP_NOKEY;
260 result_v1.YPVAL.val.valdat_len = strlen(buf);
261 result_v1.YPVAL.val.valdat_val = buf;
262 result_v1.YPVAL.stat = YP_TRUE;
264 result = (char *)&result_v1;
265 xdrfunc = (xdrproc_t)xdr_ypresponse;
268 yp_error("bad YP program version (%lu)!", q->ypvers);
274 yp_error("sending dns reply to %s (%lu)",
275 inet_ntoa(q->client_addr.sin_addr), q->id);
277 * XXX This is disgusting. There's basically one transport
278 * handle for UDP, but we're holding off on replying to a
279 * client until we're ready, by which time we may have received
280 * several other queries from other clients with different
281 * transaction IDs. So to make the delayed response thing work,
282 * we have to save the transaction ID and client address of
283 * each request, then jam them into the transport handle when
284 * we're ready to send a reply. Then after we've send the reply,
285 * we put the old transaction ID and remote address back the
286 * way we found 'em. This is _INCREDIBLY_ non-portable; it's
287 * not even supported by the RPC library.
290 * XXX Don't frob the transaction ID for TCP handles.
292 if (q->prot_type == SOCK_DGRAM)
293 xid = svcudp_set_xid(q->xprt, q->xid);
294 client_addr = q->xprt->xp_raddr;
295 q->xprt->xp_raddr = q->client_addr;
297 if (!svc_sendreply(q->xprt, xdrfunc, result))
298 yp_error("svc_sendreply failed");
301 * Now that we sent the reply,
302 * put the handle back the way it was.
304 if (q->prot_type == SOCK_DGRAM)
305 svcudp_set_xid(q->xprt, xid);
306 q->xprt->xp_raddr = client_addr;
312 * Decrement TTL on all queue entries, possibly nuking
313 * any that have been around too long without being serviced.
317 register struct circleq_dnsentry *q, *n;
320 while (q != (void *)&qhead) {
322 n = q->links.cqe_next;
324 CIRCLEQ_REMOVE(&qhead, q, links);
339 * Data is pending on the DNS socket; check for valid replies
340 * to our queries and dispatch them to waiting clients.
344 register struct circleq_dnsentry *q;
345 char buf[sizeof(HEADER) + MAXPACKET];
346 char retrybuf[MAXHOSTNAMELEN];
347 struct sockaddr_in sin;
351 struct hostent *hent;
354 yp_error("running dns queue");
356 bzero(buf, sizeof(buf));
358 len = sizeof(struct sockaddr_in);
359 rval = recvfrom(resfd, buf, sizeof(buf), 0,
360 (struct sockaddr *)&sin, &len);
363 yp_error("recvfrom failed: %s", strerror(errno));
368 * We may have data left in the socket that represents
369 * replies to earlier queries that we don't care about
370 * anymore. If there are no lookups pending or the packet
371 * ID doesn't match any of the queue IDs, just drop it
374 hptr = (HEADER *)&buf;
376 (q = yp_find_dnsqent(ntohs(hptr->id), BY_DNS_ID)) == NULL) {
382 yp_error("got dns reply from %s", inet_ntoa(sin.sin_addr));
384 hent = __dns_getanswer(buf, rval, q->name, q->type);
387 * If the lookup failed, try appending one of the domains
388 * from resolv.conf. If we have no domains to test, the
392 if ((h_errno == TRY_AGAIN || h_errno == NO_RECOVERY)
393 && q->domain && *q->domain) {
394 snprintf(retrybuf, sizeof(retrybuf), "%s.%s",
395 q->name, *q->domain);
397 yp_error("retrying with: %s", retrybuf);
398 q->id = yp_send_dns_query(retrybuf, q->type);
404 if (q->type == T_PTR) {
405 hent->h_addr = (char *)&q->addr.s_addr;
406 hent->h_length = sizeof(struct in_addr);
410 /* Got an answer ready for a client -- send it off. */
411 yp_send_dns_reply(q, parse(hent));
413 CIRCLEQ_REMOVE(&qhead, q, links);
417 /* Decrement TTLs on other entries while we're here. */
424 * Queue and transmit an asynchronous DNS hostname lookup.
426 ypstat yp_async_lookup_name(rqstp, name)
427 struct svc_req *rqstp;
430 register struct circleq_dnsentry *q;
433 /* Check for SOCK_DGRAM or SOCK_STREAM -- we need to know later */
434 type = -1; len = sizeof(type);
435 if (getsockopt(rqstp->rq_xprt->xp_sock, SOL_SOCKET,
436 SO_TYPE, &type, &len) == -1) {
437 yp_error("getsockopt failed: %s", strerror(errno));
441 /* Avoid transmitting dupe requests. */
442 if (type == SOCK_DGRAM &&
443 yp_find_dnsqent(svcudp_get_xid(rqstp->rq_xprt),BY_RPC_XID) != NULL)
446 if ((q = yp_malloc_dnsent()) == NULL)
451 q->xprt = rqstp->rq_xprt;
452 q->ypvers = rqstp->rq_vers;
454 if (q->prot_type == SOCK_DGRAM)
455 q->xid = svcudp_get_xid(q->xprt);
456 q->client_addr = q->xprt->xp_raddr;
457 q->domain = _res.dnsrch;
458 q->id = yp_send_dns_query(name, q->type);
461 yp_error("DNS query failed");
466 q->name = strdup(name);
467 CIRCLEQ_INSERT_HEAD(&qhead, q, links);
471 yp_error("queueing async DNS name lookup (%d)", q->id);
478 * Queue and transmit an asynchronous DNS IP address lookup.
480 ypstat yp_async_lookup_addr(rqstp, addr)
481 struct svc_req *rqstp;
484 register struct circleq_dnsentry *q;
485 char buf[MAXHOSTNAMELEN];
489 /* Check for SOCK_DGRAM or SOCK_STREAM -- we need to know later */
490 type = -1; len = sizeof(type);
491 if (getsockopt(rqstp->rq_xprt->xp_sock, SOL_SOCKET,
492 SO_TYPE, &type, &len) == -1) {
493 yp_error("getsockopt failed: %s", strerror(errno));
497 /* Avoid transmitting dupe requests. */
498 if (type == SOCK_DGRAM &&
499 yp_find_dnsqent(svcudp_get_xid(rqstp->rq_xprt),BY_RPC_XID) != NULL)
502 if ((q = yp_malloc_dnsent()) == NULL)
505 if (sscanf(addr, "%d.%d.%d.%d", &a, &b, &c, &d) != 4)
508 snprintf(buf, sizeof(buf), "%d.%d.%d.%d.in-addr.arpa", d, c, b, a);
511 yp_error("DNS address is: %s", buf);
515 q->xprt = rqstp->rq_xprt;
516 q->ypvers = rqstp->rq_vers;
519 if (q->prot_type == SOCK_DGRAM)
520 q->xid = svcudp_get_xid(q->xprt);
521 q->client_addr = q->xprt->xp_raddr;
522 q->id = yp_send_dns_query(buf, q->type);
525 yp_error("DNS query failed");
530 inet_aton(addr, &q->addr);
531 q->name = strdup(buf);
532 CIRCLEQ_INSERT_HEAD(&qhead, q, links);
536 yp_error("queueing async DNS address lookup (%d)", q->id);