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
74int _rpcpmstart;		/* Started by a port monitor ? */
75static int _rpcfdtype;
76		 /* Whether Stream or Datagram ? */
77static int _rpcaf;
78static 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
86extern void ypprog_1(struct svc_req *, SVCXPRT *);
87extern void ypprog_2(struct svc_req *, SVCXPRT *);
88extern int _rpc_dtablesize(void);
89extern int _rpcsvcstate;	 /* Set when a request is serviced */
90char *progname = "ypserv";
91char *yp_dir = _PATH_YP;
92/*int debug = 0;*/
93int do_dns = 0;
94int resfd;
95
96struct socklistent {
97	int				sle_sock;
98	struct sockaddr_storage		sle_ss;
99	SLIST_ENTRY(socklistent)	sle_next;
100};
101static SLIST_HEAD(, socklistent) sle_head =
102	SLIST_HEAD_INITIALIZER(sle_head);
103
104struct bindaddrlistent {
105	const char			*ble_hostname;
106	SLIST_ENTRY(bindaddrlistent)	ble_next;
107};
108static SLIST_HEAD(, bindaddrlistent) ble_head =
109	SLIST_HEAD_INITIALIZER(ble_head);
110
111static char *servname = "0";
112
113static
114void _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
129pid_t	yp_pid;
130
131static void
132yp_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
182static void
183unregister(void)
184{
185	(void)svc_unreg(YPPROG, YPVERS);
186	(void)svc_unreg(YPPROG, YPOLDVERS);
187}
188
189static void
190reaper(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
217static void
218usage(void)
219{
220	fprintf(stderr, "usage: ypserv [-h addr] [-d] [-n] [-p path] [-P port]\n");
221	exit(1);
222}
223
224static void
225closedown(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
254static int
255create_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
449int
450main(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