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