usr.sbin/ypserv/yp_dnslookup.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-4-Clause
   3  *
   4  * Copyright (c) 1995, 1996
   5  *      Bill Paul <wpaul@ctr.columbia.edu>. All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. All advertising materials mentioning features or use of this software
  16  *    must display the following acknowledgement:
  17  *      This product includes software developed by Bill Paul.
  18  * 4. Neither the name of the University nor the names of its contributors
  19  *    may be used to endorse or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY Bill Paul AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL Bill Paul OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  */
  34
  35 #include <sys/cdefs.h>
  36 /*
  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.
  40  *
  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.
  44  */
  45
  46 #include <sys/param.h>
  47 #include <sys/socket.h>
  48 #include <sys/time.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>
  54
  55 #include <ctype.h>
  56 #include <errno.h>
  57 #include <netdb.h>
  58 #include <stdio.h>
  59 #include <stdlib.h>
  60 #include <string.h>
  61 #include <resolv.h>
  62 #include <unistd.h>
  63
  64 #include <rpcsvc/yp.h>
  65 #include "yp_extern.h"
  66
  67 static char *
  68 parse(struct hostent *hp)
  69 {
  70         static char result[MAXHOSTNAMELEN * 2];
  71         int i;
  72         size_t len;
  73         char addr[46];
  74
  75         if (hp == NULL)
  76                 return(NULL);
  77
  78         if (inet_ntop(hp->h_addrtype, hp->h_addr, addr, sizeof(addr)) == NULL)
  79                 return(NULL);
  80
  81         len = strlen(addr) + 1 + strlen(hp->h_name);
  82         for (i = 0; hp->h_aliases[i]; i++)
  83                 len += strlen(hp->h_aliases[i]) + 1;
  84         len++;
  85
  86         if (len > sizeof(result))
  87                 return(NULL);
  88
  89         bzero(result, sizeof(result));
  90         snprintf(result, sizeof(result), "%s %s", addr, hp->h_name);
  91         for (i = 0; hp->h_aliases[i]; i++) {
  92                 strcat(result, " ");
  93                 strcat(result, hp->h_aliases[i]);
  94         }
  95
  96         return ((char *)&result);
  97 }
  98
  99 #define MAXPACKET (64*1024)
 100 #define DEF_TTL 50
 101
 102 #define BY_DNS_ID 1
 103 #define BY_RPC_XID 2
 104
 105 extern struct hostent *__dns_getanswer(char *, int, char *, int);
 106
 107 static TAILQ_HEAD(dns_qhead, circleq_dnsentry) qhead;
 108
 109 struct circleq_dnsentry {
 110         SVCXPRT *xprt;
 111         unsigned long xid;
 112         struct sockaddr_in client_addr;
 113         unsigned long ypvers;
 114         unsigned long id;
 115         unsigned long ttl;
 116         unsigned long type;
 117         unsigned short prot_type;
 118         char **domain;
 119         char *name;
 120         int addrtype;
 121         int addrlen;
 122         uint32_t addr[4];       /* IPv4 or IPv6 */
 123         TAILQ_ENTRY(circleq_dnsentry) links;
 124 };
 125
 126 static int pending = 0;
 127
 128 int
 129 yp_init_resolver(void)
 130 {
 131         TAILQ_INIT(&qhead);
 132         if (!(_res.options & RES_INIT) && res_init() == -1) {
 133                 yp_error("res_init failed");
 134                 return(1);
 135         }
 136         if ((resfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
 137                 yp_error("couldn't create socket");
 138                 return(1);
 139         }
 140         if (fcntl(resfd, F_SETFL, O_NONBLOCK) == -1) {
 141                 yp_error("couldn't make resolver socket non-blocking");
 142                 return(1);
 143         }
 144         return(0);
 145 }
 146
 147 static struct
 148 circleq_dnsentry *yp_malloc_dnsent(void)
 149 {
 150         register struct circleq_dnsentry *q;
 151
 152         q = malloc(sizeof(struct circleq_dnsentry));
 153
 154         if (q == NULL) {
 155                 yp_error("failed to malloc() circleq dns entry");
 156                 return(NULL);
 157         }
 158
 159         return(q);
 160 }
 161
 162 /*
 163  * Transmit a query.
 164  */
 165 static unsigned long
 166 yp_send_dns_query(char *name, int type)
 167 {
 168         char buf[MAXPACKET];
 169         int n;
 170         HEADER *hptr;
 171         int ns;
 172         int rval;
 173         unsigned long id;
 174
 175         bzero(buf, sizeof(buf));
 176
 177         n = res_mkquery(QUERY,name,C_IN,type,NULL,0,NULL,buf,sizeof(buf));
 178
 179         if (n <= 0) {
 180                 yp_error("res_mkquery failed for %s type %d", name, type);
 181                 return(0);
 182         }
 183
 184         hptr = (HEADER *)&buf;
 185         id = ntohs(hptr->id);
 186
 187         for (ns = 0; ns < _res.nscount; ns++) {
 188                 rval = sendto(resfd, buf, n, 0,
 189                         (struct sockaddr *)&_res.nsaddr_list[ns],
 190                                 sizeof(struct sockaddr));
 191                 if (rval == -1) {
 192                         yp_error("sendto failed");
 193                         return(0);
 194                 }
 195         }
 196
 197         return(id);
 198 }
 199
 200 static struct circleq_dnsentry *
 201 yp_find_dnsqent(unsigned long id, int type)
 202 {
 203         register struct circleq_dnsentry *q;
 204
 205         TAILQ_FOREACH(q, &qhead, links) {
 206                 switch (type) {
 207                 case BY_RPC_XID:
 208                         if (id == q->xid)
 209                                 return(q);
 210                         break;
 211                 case BY_DNS_ID:
 212                 default:
 213                         if (id == q->id)
 214                                 return(q);
 215                         break;
 216                 }
 217         }
 218         return (NULL);
 219 }
 220
 221 static void
 222 yp_send_dns_reply(struct circleq_dnsentry *q, char *buf)
 223 {
 224         ypresponse result_v1;
 225         ypresp_val result_v2;
 226         unsigned long xid;
 227         struct sockaddr_in client_addr;
 228         xdrproc_t xdrfunc;
 229         char *result;
 230
 231         /*
 232          * Set up correct reply struct and
 233          * XDR filter depending on ypvers.
 234          */
 235         switch (q->ypvers) {
 236         case YPVERS:
 237                 bzero((char *)&result_v2, sizeof(result_v2));
 238
 239                 if (buf == NULL)
 240                         result_v2.stat = YP_NOKEY;
 241                 else {
 242                         result_v2.val.valdat_len = strlen(buf);
 243                         result_v2.val.valdat_val = buf;
 244                         result_v2.stat = YP_TRUE;
 245                 }
 246                 result = (char *)&result_v2;
 247                 xdrfunc = (xdrproc_t)xdr_ypresp_val;
 248                 break;
 249         case YPOLDVERS:
 250                 /*
 251                  * The odds are we will _never_ execute this
 252                  * particular code, but we include it anyway
 253                  * for the sake of completeness.
 254                  */
 255                 bzero((char *)&result_v1, sizeof(result_v1));
 256                 result_v1.yp_resptype = YPRESP_VAL;
 257
 258 #define YPVAL ypresponse_u.yp_resp_valtype
 259                 if (buf == NULL)
 260                         result_v1.YPVAL.stat = YP_NOKEY;
 261                 else {
 262                         result_v1.YPVAL.val.valdat_len = strlen(buf);
 263                         result_v1.YPVAL.val.valdat_val = buf;
 264                         result_v1.YPVAL.stat = YP_TRUE;
 265                 }
 266                 result = (char *)&result_v1;
 267                 xdrfunc = (xdrproc_t)xdr_ypresponse;
 268                 break;
 269         default:
 270                 yp_error("bad YP program version (%lu)!", q->ypvers);
 271                         return;
 272                 break;
 273         }
 274
 275         if (debug)
 276                 yp_error("sending dns reply to %s (%lu)",
 277                         inet_ntoa(q->client_addr.sin_addr), q->id);
 278         /*
 279          * XXX This is disgusting. There's basically one transport
 280          * handle for UDP, but we're holding off on replying to a
 281          * client until we're ready, by which time we may have received
 282          * several other queries from other clients with different
 283          * transaction IDs. So to make the delayed response thing work,
 284          * we have to save the transaction ID and client address of
 285          * each request, then jam them into the transport handle when
 286          * we're ready to send a reply. Then after we've send the reply,
 287          * we put the old transaction ID and remote address back the
 288          * way we found 'em. This is _INCREDIBLY_ non-portable; it's
 289          * not even supported by the RPC library.
 290          */
 291         /*
 292          * XXX Don't frob the transaction ID for TCP handles.
 293          */
 294         if (q->prot_type == SOCK_DGRAM)
 295                 xid = svcudp_set_xid(q->xprt, q->xid);
 296         client_addr = q->xprt->xp_raddr;
 297         q->xprt->xp_raddr = q->client_addr;
 298
 299         if (!svc_sendreply(q->xprt, xdrfunc, result))
 300                 yp_error("svc_sendreply failed");
 301
 302         /*
 303          * Now that we sent the reply,
 304          * put the handle back the way it was.
 305          */
 306         if (q->prot_type == SOCK_DGRAM)
 307                 svcudp_set_xid(q->xprt, xid);
 308         q->xprt->xp_raddr = client_addr;
 309 }
 310
 311 /*
 312  * Decrement TTL on all queue entries, possibly nuking
 313  * any that have been around too long without being serviced.
 314  */
 315 void
 316 yp_prune_dnsq(void)
 317 {
 318         register struct circleq_dnsentry *q, *n;
 319
 320         q = TAILQ_FIRST(&qhead);
 321         while (q != NULL) {
 322                 q->ttl--;
 323                 n = TAILQ_NEXT(q, links);
 324                 if (!q->ttl) {
 325                         TAILQ_REMOVE(&qhead, q, links);
 326                         free(q->name);
 327                         free(q);
 328                         pending--;
 329                 }
 330                 q = n;
 331         }
 332
 333         if (pending < 0)
 334                 pending = 0;
 335 }
 336
 337 /*
 338  * Data is pending on the DNS socket; check for valid replies
 339  * to our queries and dispatch them to waiting clients.
 340  */
 341 void
 342 yp_run_dnsq(void)
 343 {
 344         register struct circleq_dnsentry *q;
 345         char buf[sizeof(HEADER) + MAXPACKET];
 346         struct sockaddr_in sin;
 347         socklen_t len;
 348         int rval;
 349         HEADER *hptr;
 350         struct hostent *hent;
 351
 352         if (debug)
 353                 yp_error("running dns queue");
 354
 355         bzero(buf, sizeof(buf));
 356
 357         len = sizeof(struct sockaddr_in);
 358         rval = recvfrom(resfd, buf, sizeof(buf), 0,
 359                         (struct sockaddr *)&sin, &len);
 360
 361         if (rval == -1) {
 362                 yp_error("recvfrom failed: %s", strerror(errno));
 363                 return;
 364         }
 365
 366         /*
 367          * We may have data left in the socket that represents
 368          * replies to earlier queries that we don't care about
 369          * anymore. If there are no lookups pending or the packet
 370          * ID doesn't match any of the queue IDs, just drop it
 371          * on the floor.
 372          */
 373         hptr = (HEADER *)&buf;
 374         if (!pending ||
 375                 (q = yp_find_dnsqent(ntohs(hptr->id), BY_DNS_ID)) == NULL) {
 376                 /* ignore */
 377                 return;
 378         }
 379
 380         if (debug)
 381                 yp_error("got dns reply from %s", inet_ntoa(sin.sin_addr));
 382
 383         hent = __dns_getanswer(buf, rval, q->name, q->type);
 384
 385         if (hent != NULL) {
 386                 if (q->type == T_PTR) {
 387                         hent->h_addr = (char *)q->addr;
 388                         hent->h_addrtype = q->addrtype;
 389                         hent->h_length = q->addrlen;
 390                 }
 391         }
 392
 393         /* Got an answer ready for a client -- send it off. */
 394         yp_send_dns_reply(q, parse(hent));
 395         pending--;
 396         TAILQ_REMOVE(&qhead, q, links);
 397         free(q->name);
 398         free(q);
 399
 400         /* Decrement TTLs on other entries while we're here. */
 401         yp_prune_dnsq();
 402 }
 403
 404 /*
 405  * Queue and transmit an asynchronous DNS hostname lookup.
 406  */
 407 ypstat
 408 yp_async_lookup_name(struct svc_req *rqstp, char *name, int af)
 409 {
 410         register struct circleq_dnsentry *q;
 411         socklen_t len;
 412         int type;
 413
 414         /* Check for SOCK_DGRAM or SOCK_STREAM -- we need to know later */
 415         type = -1;
 416         len = sizeof(type);
 417         if (getsockopt(rqstp->rq_xprt->xp_fd, SOL_SOCKET,
 418                                         SO_TYPE, &type, &len) == -1) {
 419                 yp_error("getsockopt failed: %s", strerror(errno));
 420                 return(YP_YPERR);
 421         }
 422
 423         /* Avoid transmitting dupe requests. */
 424         if (type == SOCK_DGRAM &&
 425             yp_find_dnsqent(svcudp_get_xid(rqstp->rq_xprt),BY_RPC_XID) != NULL)
 426                 return(YP_TRUE);
 427
 428         if ((q = yp_malloc_dnsent()) == NULL)
 429                 return(YP_YPERR);
 430
 431         q->type = (af == AF_INET) ? T_A : T_AAAA;
 432         q->ttl = DEF_TTL;
 433         q->xprt = rqstp->rq_xprt;
 434         q->ypvers = rqstp->rq_vers;
 435         q->prot_type = type;
 436         if (q->prot_type == SOCK_DGRAM)
 437                 q->xid = svcudp_get_xid(q->xprt);
 438         q->client_addr = q->xprt->xp_raddr;
 439         q->domain = _res.dnsrch;
 440         q->id = yp_send_dns_query(name, q->type);
 441
 442         if (q->id == 0) {
 443                 yp_error("DNS query failed");
 444                 free(q);
 445                 return(YP_YPERR);
 446         }
 447
 448         q->name = strdup(name);
 449         TAILQ_INSERT_HEAD(&qhead, q, links);
 450         pending++;
 451
 452         if (debug)
 453                 yp_error("queueing async DNS name lookup (%lu)", q->id);
 454
 455         yp_prune_dnsq();
 456         return(YP_TRUE);
 457 }
 458
 459 /*
 460  * Queue and transmit an asynchronous DNS IP address lookup.
 461  */
 462 ypstat
 463 yp_async_lookup_addr(struct svc_req *rqstp, char *addr, int af)
 464 {
 465         register struct circleq_dnsentry *q;
 466         char buf[MAXHOSTNAMELEN], *qp;
 467         uint32_t abuf[4];       /* IPv4 or IPv6 */
 468         u_char *uaddr = (u_char *)abuf;
 469         socklen_t len;
 470         int type, n;
 471
 472         /* Check for SOCK_DGRAM or SOCK_STREAM -- we need to know later */
 473         type = -1;
 474         len = sizeof(type);
 475         if (getsockopt(rqstp->rq_xprt->xp_fd, SOL_SOCKET,
 476                                         SO_TYPE, &type, &len) == -1) {
 477                 yp_error("getsockopt failed: %s", strerror(errno));
 478                 return(YP_YPERR);
 479         }
 480
 481         /* Avoid transmitting dupe requests. */
 482         if (type == SOCK_DGRAM &&
 483             yp_find_dnsqent(svcudp_get_xid(rqstp->rq_xprt),BY_RPC_XID) != NULL)
 484                 return(YP_TRUE);
 485
 486         switch (af) {
 487         case AF_INET:
 488                 if (inet_aton(addr, (struct in_addr *)uaddr) != 1)
 489                         return(YP_NOKEY);
 490                 snprintf(buf, sizeof(buf), "%u.%u.%u.%u.in-addr.arpa",
 491                     (uaddr[3] & 0xff), (uaddr[2] & 0xff),
 492                     (uaddr[1] & 0xff), (uaddr[0] & 0xff));
 493                 len = INADDRSZ;
 494                 break;
 495         case AF_INET6:
 496                 if (inet_pton(af, addr, uaddr) != 1)
 497                         return(YP_NOKEY);
 498                 qp = buf;
 499                 for (n = IN6ADDRSZ - 1; n >= 0; n--) {
 500                         qp += (size_t)sprintf(qp, "%x.%x.", uaddr[n] & 0xf,
 501                             (uaddr[n] >> 4) & 0xf);
 502                 }
 503                 strlcat(buf, "ip6.arpa", sizeof(buf));
 504                 len = IN6ADDRSZ;
 505                 break;
 506         default:
 507                 return(YP_YPERR);
 508         }
 509
 510         if ((q = yp_malloc_dnsent()) == NULL)
 511                 return(YP_YPERR);
 512
 513         if (debug)
 514                 yp_error("DNS address is: %s", buf);
 515
 516         q->type = T_PTR;
 517         q->ttl = DEF_TTL;
 518         q->xprt = rqstp->rq_xprt;
 519         q->ypvers = rqstp->rq_vers;
 520         q->domain = NULL;
 521         q->prot_type = type;
 522         if (q->prot_type == SOCK_DGRAM)
 523                 q->xid = svcudp_get_xid(q->xprt);
 524         q->client_addr = q->xprt->xp_raddr;
 525         q->id = yp_send_dns_query(buf, q->type);
 526
 527         if (q->id == 0) {
 528                 yp_error("DNS query failed");
 529                 free(q);
 530                 return(YP_YPERR);
 531         }
 532
 533         memcpy(q->addr, uaddr, len);
 534         q->addrlen = len;
 535         q->addrtype = af;
 536         q->name = strdup(buf);
 537         TAILQ_INSERT_HEAD(&qhead, q, links);
 538         pending++;
 539
 540         if (debug)
 541                 yp_error("queueing async DNS address lookup (%lu)", q->id);
 542
 543         yp_prune_dnsq();
 544         return(YP_TRUE);
 545 }