1 /* 2 * Copyright (c) 1995 3 * Bill Paul <wpaul@ctr.columbia.edu>. 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. 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 author nor the names of any co-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 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 30 * SUCH DAMAGE. 31 */ 32 33 #include <sys/cdefs.h> 34 __FBSDID("$FreeBSD$"); 35 36 /* 37 * ypserv startup function. 38 * We need out own main() since we have to do some additional work 39 * that rpcgen won't do for us. Most of this file was generated using 40 * rpcgen.new, and later modified. 41 */ 42 43 #include <sys/types.h> 44 #include <sys/mman.h> 45 #include <sys/queue.h> 46 #include <sys/socket.h> 47 #include <sys/wait.h> 48 #include "yp.h" 49 #include <err.h> 50 #include <errno.h> 51 #include <memory.h> 52 #include <stdio.h> 53 #include <signal.h> 54 #include <stdarg.h> 55 #include <stdlib.h> /* getenv, exit */ 56 #include <string.h> /* strcmp */ 57 #include <syslog.h> 58 #include <unistd.h> 59 #ifdef __cplusplus 60 #include <sysent.h> /* getdtablesize, open */ 61 #endif /* __cplusplus */ 62 #include <netinet/in.h> 63 #include <netdb.h> 64 #include "yp_extern.h" 65 #include <netconfig.h> 66 #include <rpc/rpc.h> 67 #include <rpc/rpc_com.h> 68 69 #ifndef SIG_PF 70 #define SIG_PF void(*)(int) 71 #endif 72 73 #define _RPCSVC_CLOSEDOWN 120 74 int _rpcpmstart; /* Started by a port monitor ? */ 75 static int _rpcfdtype; 76 /* Whether Stream or Datagram ? */ 77 static int _rpcaf; 78 static int _rpcfd; 79 80 /* States a server can be in wrt request */ 81 82 #define _IDLE 0 83 #define _SERVED 1 84 #define _SERVING 2 85 86 extern void ypprog_1(struct svc_req *, SVCXPRT *); 87 extern void ypprog_2(struct svc_req *, SVCXPRT *); 88 extern int _rpc_dtablesize(void); 89 extern int _rpcsvcstate; /* Set when a request is serviced */ 90 char *progname = "ypserv"; 91 char *yp_dir = _PATH_YP; 92 /*int debug = 0;*/ 93 int do_dns = 0; 94 int resfd; 95 96 struct socklistent { 97 int sle_sock; 98 struct sockaddr_storage sle_ss; 99 SLIST_ENTRY(socklistent) sle_next; 100 }; 101 static SLIST_HEAD(, socklistent) sle_head = 102 SLIST_HEAD_INITIALIZER(sle_head); 103 104 struct bindaddrlistent { 105 const char *ble_hostname; 106 SLIST_ENTRY(bindaddrlistent) ble_next; 107 }; 108 static SLIST_HEAD(, bindaddrlistent) ble_head = 109 SLIST_HEAD_INITIALIZER(ble_head); 110 111 static char *servname = "0"; 112 113 static 114 void _msgout(char* msg, ...) 115 { 116 va_list ap; 117 118 va_start(ap, msg); 119 if (debug) { 120 if (_rpcpmstart) 121 vsyslog(LOG_ERR, msg, ap); 122 else 123 vwarnx(msg, ap); 124 } else 125 vsyslog(LOG_ERR, msg, ap); 126 va_end(ap); 127 } 128 129 pid_t yp_pid; 130 131 static void 132 yp_svc_run(void) 133 { 134 #ifdef FD_SETSIZE 135 fd_set readfds; 136 #else 137 int readfds; 138 #endif /* def FD_SETSIZE */ 139 int fd_setsize = _rpc_dtablesize(); 140 struct timeval timeout; 141 142 /* Establish the identity of the parent ypserv process. */ 143 yp_pid = getpid(); 144 145 for (;;) { 146 #ifdef FD_SETSIZE 147 readfds = svc_fdset; 148 #else 149 readfds = svc_fds; 150 #endif /* def FD_SETSIZE */ 151 152 FD_SET(resfd, &readfds); 153 154 timeout.tv_sec = RESOLVER_TIMEOUT; 155 timeout.tv_usec = 0; 156 switch (select(fd_setsize, &readfds, NULL, NULL, 157 &timeout)) { 158 case -1: 159 if (errno == EINTR) { 160 continue; 161 } 162 warn("svc_run: - select failed"); 163 return; 164 case 0: 165 if (getpid() == yp_pid) 166 yp_prune_dnsq(); 167 break; 168 default: 169 if (getpid() == yp_pid) { 170 if (FD_ISSET(resfd, &readfds)) { 171 yp_run_dnsq(); 172 FD_CLR(resfd, &readfds); 173 } 174 svc_getreqset(&readfds); 175 } 176 } 177 if (yp_pid != getpid()) 178 _exit(0); 179 } 180 } 181 182 static void 183 unregister(void) 184 { 185 (void)svc_unreg(YPPROG, YPVERS); 186 (void)svc_unreg(YPPROG, YPOLDVERS); 187 } 188 189 static void 190 reaper(int sig) 191 { 192 int status; 193 int saved_errno; 194 195 saved_errno = errno; 196 197 if (sig == SIGHUP) { 198 load_securenets(); 199 #ifdef DB_CACHE 200 yp_flush_all(); 201 #endif 202 errno = saved_errno; 203 return; 204 } 205 206 if (sig == SIGCHLD) { 207 while (wait3(&status, WNOHANG, NULL) > 0) 208 children--; 209 } else { 210 unregister(); 211 exit(0); 212 } 213 errno = saved_errno; 214 return; 215 } 216 217 static void 218 usage(void) 219 { 220 fprintf(stderr, "usage: ypserv [-h addr] [-d] [-n] [-p path] [-P port]\n"); 221 exit(1); 222 } 223 224 static void 225 closedown(int sig) 226 { 227 if (_rpcsvcstate == _IDLE) { 228 extern fd_set svc_fdset; 229 static int size; 230 int i, openfd; 231 232 if (_rpcfdtype == SOCK_DGRAM) { 233 unregister(); 234 exit(0); 235 } 236 if (size == 0) { 237 size = getdtablesize(); 238 } 239 for (i = 0, openfd = 0; i < size && openfd < 2; i++) 240 if (FD_ISSET(i, &svc_fdset)) 241 openfd++; 242 if (openfd <= 1) { 243 unregister(); 244 exit(0); 245 } 246 } 247 if (_rpcsvcstate == _SERVED) 248 _rpcsvcstate = _IDLE; 249 250 (void) signal(SIGALRM, (SIG_PF) closedown); 251 (void) alarm(_RPCSVC_CLOSEDOWN/2); 252 } 253 254 static int 255 create_service(const int sock, const struct netconfig *nconf, 256 const struct __rpc_sockinfo *si) 257 { 258 int error; 259 260 SVCXPRT *transp; 261 struct addrinfo hints, *res, *res0; 262 struct socklistent *slep; 263 struct bindaddrlistent *blep; 264 struct netbuf svcaddr; 265 266 SLIST_INIT(&sle_head); 267 memset(&hints, 0, sizeof(hints)); 268 memset(&svcaddr, 0, sizeof(svcaddr)); 269 270 hints.ai_family = si->si_af; 271 hints.ai_socktype = si->si_socktype; 272 hints.ai_protocol = si->si_proto; 273 274 /* 275 * Build socketlist from bindaddrlist. 276 */ 277 if (sock == RPC_ANYFD) { 278 SLIST_FOREACH(blep, &ble_head, ble_next) { 279 if (blep->ble_hostname == NULL) 280 hints.ai_flags = AI_PASSIVE; 281 else 282 hints.ai_flags = 0; 283 error = getaddrinfo(blep->ble_hostname, servname, 284 &hints, &res0); 285 if (error) { 286 _msgout("getaddrinfo(): %s", 287 gai_strerror(error)); 288 return -1; 289 } 290 for (res = res0; res; res = res->ai_next) { 291 int s; 292 293 s = __rpc_nconf2fd(nconf); 294 if (s < 0) { 295 if (errno == EAFNOSUPPORT) 296 _msgout("unsupported" 297 " transport: %s", 298 nconf->nc_netid); 299 else 300 _msgout("cannot create" 301 " %s socket: %s", 302 nconf->nc_netid, 303 strerror(errno)); 304 freeaddrinfo(res0); 305 return -1; 306 } 307 if (bindresvport_sa(s, res->ai_addr) == -1) { 308 if ((errno != EPERM) || 309 (bind(s, res->ai_addr, 310 res->ai_addrlen) == -1)) { 311 _msgout("cannot bind " 312 "%s socket: %s", 313 nconf->nc_netid, 314 strerror(errno)); 315 freeaddrinfo(res0); 316 close(sock); 317 return -1; 318 } 319 } 320 if (nconf->nc_semantics != NC_TPI_CLTS) 321 listen(s, SOMAXCONN); 322 323 slep = malloc(sizeof(*slep)); 324 if (slep == NULL) { 325 _msgout("malloc failed: %s", 326 strerror(errno)); 327 freeaddrinfo(res0); 328 close(s); 329 return -1; 330 } 331 memset(slep, 0, sizeof(*slep)); 332 memcpy(&slep->sle_ss, res->ai_addr, 333 res->ai_addrlen); 334 slep->sle_sock = s; 335 SLIST_INSERT_HEAD(&sle_head, slep, sle_next); 336 337 /* 338 * If servname == "0", redefine it by using 339 * the bound socket. 340 */ 341 if (strncmp("0", servname, 1) == 0) { 342 struct sockaddr *sap; 343 socklen_t slen; 344 char *sname; 345 346 sname = malloc(NI_MAXSERV); 347 if (sname == NULL) { 348 _msgout("malloc(): %s", 349 strerror(errno)); 350 freeaddrinfo(res0); 351 close(s); 352 return -1; 353 } 354 memset(sname, 0, NI_MAXSERV); 355 356 sap = (struct sockaddr *)&slep->sle_ss; 357 slen = sizeof(*sap); 358 error = getsockname(s, sap, &slen); 359 if (error) { 360 _msgout("getsockname(): %s", 361 strerror(errno)); 362 freeaddrinfo(res0); 363 close(s); 364 free(sname); 365 return -1; 366 } 367 error = getnameinfo(sap, slen, 368 NULL, 0, 369 sname, NI_MAXSERV, 370 NI_NUMERICHOST | NI_NUMERICSERV); 371 if (error) { 372 _msgout("getnameinfo(): %s", 373 strerror(errno)); 374 freeaddrinfo(res0); 375 close(s); 376 free(sname); 377 return -1; 378 } 379 servname = sname; 380 } 381 } 382 freeaddrinfo(res0); 383 } 384 } else { 385 slep = malloc(sizeof(*slep)); 386 if (slep == NULL) { 387 _msgout("malloc failed: %s", strerror(errno)); 388 return -1; 389 } 390 memset(slep, 0, sizeof(*slep)); 391 slep->sle_sock = sock; 392 SLIST_INSERT_HEAD(&sle_head, slep, sle_next); 393 } 394 395 /* 396 * Traverse socketlist and create rpc service handles for each socket. 397 */ 398 SLIST_FOREACH(slep, &sle_head, sle_next) { 399 if (nconf->nc_semantics == NC_TPI_CLTS) 400 transp = svc_dg_create(slep->sle_sock, 0, 0); 401 else 402 transp = svc_vc_create(slep->sle_sock, RPC_MAXDATASIZE, 403 RPC_MAXDATASIZE); 404 if (transp == NULL) { 405 _msgout("unable to create service: %s", 406 nconf->nc_netid); 407 continue; 408 } 409 if (!svc_reg(transp, YPPROG, YPOLDVERS, ypprog_1, NULL)) { 410 svc_destroy(transp); 411 close(slep->sle_sock); 412 _msgout("unable to register (YPPROG, YPOLDVERS, %s):" 413 " %s", nconf->nc_netid, strerror(errno)); 414 continue; 415 } 416 if (!svc_reg(transp, YPPROG, YPVERS, ypprog_2, NULL)) { 417 svc_destroy(transp); 418 close(slep->sle_sock); 419 _msgout("unable to register (YPPROG, YPVERS, %s): %s", 420 nconf->nc_netid, strerror(errno)); 421 continue; 422 } 423 } 424 while(!(SLIST_EMPTY(&sle_head))) 425 SLIST_REMOVE_HEAD(&sle_head, sle_next); 426 427 /* 428 * Register RPC service to rpcbind by using AI_PASSIVE address. 429 */ 430 hints.ai_flags = AI_PASSIVE; 431 error = getaddrinfo(NULL, servname, &hints, &res0); 432 if (error) { 433 _msgout("getaddrinfo(): %s", gai_strerror(error)); 434 return -1; 435 } 436 svcaddr.buf = res0->ai_addr; 437 svcaddr.len = res0->ai_addrlen; 438 439 if (si->si_af == AF_INET) { 440 /* XXX: ignore error intentionally */ 441 rpcb_set(YPPROG, YPOLDVERS, nconf, &svcaddr); 442 } 443 /* XXX: ignore error intentionally */ 444 rpcb_set(YPPROG, YPVERS, nconf, &svcaddr); 445 freeaddrinfo(res0); 446 return 0; 447 } 448 449 int 450 main(int argc, char *argv[]) 451 { 452 int ch; 453 int error; 454 int ntrans; 455 456 void *nc_handle; 457 struct netconfig *nconf; 458 struct __rpc_sockinfo si; 459 struct bindaddrlistent *blep; 460 461 memset(&si, 0, sizeof(si)); 462 SLIST_INIT(&ble_head); 463 464 while ((ch = getopt(argc, argv, "dh:np:P:")) != -1) { 465 switch (ch) { 466 case 'd': 467 debug = ypdb_debug = 1; 468 break; 469 case 'h': 470 blep = malloc(sizeof(*blep)); 471 if (blep == NULL) 472 err(1, "malloc() failed: -h %s", optarg); 473 blep->ble_hostname = optarg; 474 SLIST_INSERT_HEAD(&ble_head, blep, ble_next); 475 break; 476 case 'n': 477 do_dns = 1; 478 break; 479 case 'p': 480 yp_dir = optarg; 481 break; 482 case 'P': 483 servname = optarg; 484 break; 485 default: 486 usage(); 487 } 488 } 489 /* 490 * Add "anyaddr" entry if no -h is specified. 491 */ 492 if (SLIST_EMPTY(&ble_head)) { 493 blep = malloc(sizeof(*blep)); 494 if (blep == NULL) 495 err(1, "malloc() failed"); 496 memset(blep, 0, sizeof(*blep)); 497 SLIST_INSERT_HEAD(&ble_head, blep, ble_next); 498 } 499 500 load_securenets(); 501 yp_init_resolver(); 502 #ifdef DB_CACHE 503 yp_init_dbs(); 504 #endif 505 nc_handle = setnetconfig(); 506 if (nc_handle == NULL) 507 err(1, "cannot read %s", NETCONFIG); 508 if (__rpc_fd2sockinfo(0, &si) != 0) { 509 /* invoked from inetd */ 510 _rpcpmstart = 1; 511 _rpcfdtype = si.si_socktype; 512 _rpcaf = si.si_af; 513 _rpcfd = 0; 514 openlog("ypserv", LOG_PID, LOG_DAEMON); 515 } else { 516 /* standalone mode */ 517 if (!debug) { 518 if (daemon(0,0)) { 519 err(1,"cannot fork"); 520 } 521 openlog("ypserv", LOG_PID, LOG_DAEMON); 522 } 523 _rpcpmstart = 0; 524 _rpcaf = AF_INET; 525 _rpcfd = RPC_ANYFD; 526 unregister(); 527 } 528 529 if (madvise(NULL, 0, MADV_PROTECT) != 0) 530 _msgout("madvise(): %s", strerror(errno)); 531 532 /* 533 * Create RPC service for each transport. 534 */ 535 ntrans = 0; 536 while((nconf = getnetconfig(nc_handle))) { 537 if ((nconf->nc_flag & NC_VISIBLE)) { 538 if (__rpc_nconf2sockinfo(nconf, &si) == 0) { 539 _msgout("cannot get information for %s. " 540 "Ignored.", nconf->nc_netid); 541 continue; 542 } 543 if (_rpcpmstart) { 544 if (si.si_socktype != _rpcfdtype || 545 si.si_af != _rpcaf) 546 continue; 547 } else if (si.si_af != _rpcaf) 548 continue; 549 error = create_service(_rpcfd, nconf, &si); 550 if (error) { 551 endnetconfig(nc_handle); 552 exit(1); 553 } 554 ntrans++; 555 } 556 } 557 endnetconfig(nc_handle); 558 while(!(SLIST_EMPTY(&ble_head))) 559 SLIST_REMOVE_HEAD(&ble_head, ble_next); 560 if (ntrans == 0) { 561 _msgout("no transport is available. Aborted."); 562 exit(1); 563 } 564 if (_rpcpmstart) { 565 (void) signal(SIGALRM, (SIG_PF) closedown); 566 (void) alarm(_RPCSVC_CLOSEDOWN/2); 567 } 568 /* 569 * Make sure SIGPIPE doesn't blow us away while servicing TCP 570 * connections. 571 */ 572 (void) signal(SIGPIPE, SIG_IGN); 573 (void) signal(SIGCHLD, (SIG_PF) reaper); 574 (void) signal(SIGTERM, (SIG_PF) reaper); 575 (void) signal(SIGINT, (SIG_PF) reaper); 576 (void) signal(SIGHUP, (SIG_PF) reaper); 577 yp_svc_run(); 578 _msgout("svc_run returned"); 579 exit(1); 580 /* NOTREACHED */ 581 } 582