xref: /freebsd/usr.sbin/rpc.yppasswdd/yppasswdd_main.c (revision cc16dea626cf2fc80cde667ac4798065108e596c)
1 /*
2  * Copyright (c) 1995, 1996
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 #include <sys/param.h>
37 #include <sys/ioctl.h>
38 #include <sys/stat.h>
39 #include <sys/socket.h>
40 #include <sys/time.h>
41 #include <sys/resource.h>
42 #include <netinet/in.h>
43 
44 #include <err.h>
45 #include <errno.h>
46 #include <fcntl.h>
47 #include <memory.h>
48 #include <signal.h>
49 #include <stdio.h>
50 #include <stdlib.h> /* getenv, exit */
51 #include <string.h> /* strcmp */
52 #include <syslog.h>
53 #include <unistd.h>
54 
55 #include <rpc/rpc.h>
56 #include <rpc/rpc_com.h>
57 #include <rpc/pmap_clnt.h> /* for pmap_unset */
58 #include <rpcsvc/yp.h>
59 #include <rpcsvc/ypclnt.h>
60 
61 #include "yppasswd.h"
62 #include "yppasswdd_extern.h"
63 #include "yppasswd_private.h"
64 #include "ypxfr_extern.h"
65 #include "yp_extern.h"
66 
67 #ifndef SIG_PF
68 #define	SIG_PF void(*)(int)
69 #endif
70 
71 #ifdef DEBUG
72 #define	RPC_SVC_FG
73 #endif
74 
75 #define	_RPCSVC_CLOSEDOWN 120
76 int _rpcpmstart = 0;		/* Started by a port monitor ? */
77 static int _rpcfdtype;
78 		 /* Whether Stream or Datagram ? */
79 	/* States a server can be in wrt request */
80 
81 #define	_IDLE 0
82 #define	_SERVED 1
83 #define	_SERVING 2
84 
85 static char _localhost[] = "localhost";
86 static char _passwd_byname[] = "passwd.byname";
87 extern int _rpcsvcstate;	 /* Set when a request is serviced */
88 static char _progname[] = "rpc.yppasswdd";
89 char *progname = _progname;
90 static char _yp_dir[] = _PATH_YP;
91 char *yp_dir = _yp_dir;
92 static char _passfile_default[] = _PATH_YP "master.passwd";
93 char *passfile_default = _passfile_default;
94 char *passfile;
95 char *yppasswd_domain = NULL;
96 int no_chsh = 0;
97 int no_chfn = 0;
98 int allow_additions = 0;
99 int multidomain = 0;
100 int verbose = 0;
101 int resvport = 1;
102 int inplace = 0;
103 char sockname[] = YP_SOCKNAME;
104 
105 static void
106 terminate(int sig __unused)
107 {
108 	rpcb_unset(YPPASSWDPROG, YPPASSWDVERS, NULL);
109 	rpcb_unset(MASTER_YPPASSWDPROG, MASTER_YPPASSWDVERS, NULL);
110 	unlink(sockname);
111 	exit(0);
112 }
113 
114 static void
115 reload(int sig __unused)
116 {
117 	load_securenets();
118 }
119 
120 static void
121 closedown(int sig __unused)
122 {
123 	if (_rpcsvcstate == _IDLE) {
124 		extern fd_set svc_fdset;
125 		static int size;
126 		int i, openfd;
127 
128 		if (_rpcfdtype == SOCK_DGRAM) {
129 			unlink(sockname);
130 			exit(0);
131 		}
132 		if (size == 0) {
133 			size = getdtablesize();
134 		}
135 		for (i = 0, openfd = 0; i < size && openfd < 2; i++)
136 			if (FD_ISSET(i, &svc_fdset))
137 				openfd++;
138 		if (openfd <= 1) {
139 			unlink(sockname);
140 			exit(0);
141 		}
142 	}
143 	if (_rpcsvcstate == _SERVED)
144 		_rpcsvcstate = _IDLE;
145 
146 	(void) signal(SIGALRM, (SIG_PF) closedown);
147 	(void) alarm(_RPCSVC_CLOSEDOWN/2);
148 }
149 
150 static void
151 usage(void)
152 {
153 	fprintf(stderr, "%s\n%s\n",
154 "usage: rpc.yppasswdd [-t master.passwd file] [-d domain] [-p path] [-s]",
155 "                     [-f] [-m] [-i] [-a] [-v] [-u] [-h]");
156 	exit(1);
157 }
158 
159 int
160 main(int argc, char *argv[])
161 {
162 	struct rlimit rlim;
163 	SVCXPRT *transp = NULL;
164 	struct sockaddr_in saddr;
165 	socklen_t asize = sizeof (saddr);
166 	struct netconfig *nconf;
167 	struct sigaction sa;
168 	void *localhandle;
169 	int ch;
170 	char *mastername;
171 	char myname[MAXHOSTNAMELEN + 2];
172 	int maxrec = RPC_MAXDATASIZE;
173 
174 	extern int debug;
175 
176 	debug = 1;
177 
178 	while ((ch = getopt(argc, argv, "t:d:p:sfamuivh")) != -1) {
179 		switch (ch) {
180 		case 't':
181 			passfile_default = optarg;
182 			break;
183 		case 'd':
184 			yppasswd_domain = optarg;
185 			break;
186 		case 's':
187 			no_chsh++;
188 			break;
189 		case 'f':
190 			no_chfn++;
191 			break;
192 		case 'p':
193 			yp_dir = optarg;
194 			break;
195 		case 'a':
196 			allow_additions++;
197 			break;
198 		case 'm':
199 			multidomain++;
200 			break;
201 		case 'i':
202 			inplace++;
203 			break;
204 		case 'v':
205 			verbose++;
206 			break;
207 		case 'u':
208 			resvport = 0;
209 			break;
210 		default:
211 		case 'h':
212 			usage();
213 			break;
214 		}
215 	}
216 
217 	if (yppasswd_domain == NULL) {
218 		if (yp_get_default_domain(&yppasswd_domain)) {
219 			yp_error("no domain specified and system domain \
220 name isn't set -- aborting");
221 		usage();
222 		}
223 	}
224 
225 	load_securenets();
226 
227 	if (getrpcport(_localhost, YPPROG, YPVERS, IPPROTO_UDP) <= 0) {
228 		yp_error("no ypserv processes registered with local portmap");
229 		yp_error("this host is not an NIS server -- aborting");
230 		exit(1);
231 	}
232 
233 	if ((mastername = ypxfr_get_master(yppasswd_domain,
234 		 _passwd_byname, _localhost, 0)) == NULL) {
235 		yp_error("can't get name of NIS master server for domain %s",
236 			 				yppasswd_domain);
237 		exit(1);
238 	}
239 
240 	if (gethostname((char *)&myname, sizeof(myname)) == -1) {
241 		yp_error("can't get local hostname: %s", strerror(errno));
242 		exit(1);
243 	}
244 
245 	if (strncasecmp(mastername, (char *)&myname, sizeof(myname))) {
246 		yp_error("master of %s is %s, but we are %s",
247 			"passwd.byname", mastername, myname);
248 		yp_error("this host is not the NIS master server for \
249 the %s domain -- aborting", yppasswd_domain);
250 		exit(1);
251 	}
252 
253 	debug = 0;
254 
255 	if (getsockname(0, (struct sockaddr *)&saddr, &asize) == 0) {
256 		socklen_t ssize = sizeof (int);
257 		if (saddr.sin_family != AF_INET)
258 			exit(1);
259 		if (getsockopt(0, SOL_SOCKET, SO_TYPE,
260 		    (char *)&_rpcfdtype, &ssize) == -1)
261 			exit(1);
262 		_rpcpmstart = 1;
263 	}
264 
265 	if (!debug && _rpcpmstart == 0) {
266 		if (daemon(0,0)) {
267 			err(1,"cannot fork");
268 		}
269 	}
270 	openlog("rpc.yppasswdd", LOG_PID, LOG_DAEMON);
271 	memset(&sa, 0, sizeof(sa));
272 	sa.sa_flags = SA_NOCLDWAIT;
273 	sigaction(SIGCHLD, &sa, NULL);
274 
275 	rpcb_unset(YPPASSWDPROG, YPPASSWDVERS, NULL);
276 	rpcb_unset(MASTER_YPPASSWDPROG, MASTER_YPPASSWDVERS, NULL);
277 
278 	rpc_control(RPC_SVC_CONNMAXREC_SET, &maxrec);
279 
280 	if (svc_create(yppasswdprog_1, YPPASSWDPROG, YPPASSWDVERS, "netpath") == 0) {
281 		yp_error("cannot create yppasswd service.");
282 		exit(1);
283 	}
284 	if (svc_create(master_yppasswdprog_1, MASTER_YPPASSWDPROG,
285 	    MASTER_YPPASSWDVERS, "netpath") == 0) {
286 		yp_error("cannot create master_yppasswd service.");
287 		exit(1);
288 	}
289 
290 	nconf = NULL;
291 	localhandle = setnetconfig();
292 	while ((nconf = getnetconfig(localhandle)) != NULL) {
293 		if (nconf->nc_protofmly != NULL &&
294 		    strcmp(nconf->nc_protofmly, NC_LOOPBACK) == 0)
295 			break;
296 	}
297 	if (nconf == NULL) {
298 		yp_error("getnetconfigent unix: %s", nc_sperror());
299 		exit(1);
300 	}
301 	unlink(sockname);
302 	transp = svcunix_create(RPC_ANYSOCK, 0, 0, sockname);
303 	if (transp == NULL) {
304 		yp_error("cannot create AF_LOCAL service.");
305 		exit(1);
306 	}
307 	if (!svc_reg(transp, MASTER_YPPASSWDPROG, MASTER_YPPASSWDVERS,
308 	    master_yppasswdprog_1, nconf)) {
309 		yp_error("unable to register (MASTER_YPPASSWDPROG, \
310 		    MASTER_YPPASSWDVERS, unix).");
311 		exit(1);
312 	}
313 	endnetconfig(localhandle);
314 
315 	/* Only root may connect() to the AF_UNIX link. */
316 	if (chmod(sockname, 0))
317 		err(1, "chmod of %s failed", sockname);
318 
319 	if (transp == (SVCXPRT *)NULL) {
320 		yp_error("could not create a handle");
321 		exit(1);
322 	}
323 	if (_rpcpmstart) {
324 		(void) signal(SIGALRM, (SIG_PF) closedown);
325 		(void) alarm(_RPCSVC_CLOSEDOWN/2);
326 	}
327 
328 	/* Unlimited resource limits. */
329 	rlim.rlim_cur = rlim.rlim_max = RLIM_INFINITY;
330 	(void)setrlimit(RLIMIT_CPU, &rlim);
331 	(void)setrlimit(RLIMIT_FSIZE, &rlim);
332 	(void)setrlimit(RLIMIT_STACK, &rlim);
333 	(void)setrlimit(RLIMIT_DATA, &rlim);
334 	(void)setrlimit(RLIMIT_RSS, &rlim);
335 
336 	/* Don't drop core (not really necessary, but GP's). */
337 	rlim.rlim_cur = rlim.rlim_max = 0;
338 	(void)setrlimit(RLIMIT_CORE, &rlim);
339 
340 	/* Turn off signals. */
341 	(void)signal(SIGALRM, SIG_IGN);
342 	(void)signal(SIGHUP, (SIG_PF) reload);
343 	(void)signal(SIGINT, SIG_IGN);
344 	(void)signal(SIGPIPE, SIG_IGN);
345 	(void)signal(SIGQUIT, SIG_IGN);
346 	(void)signal(SIGTERM, (SIG_PF) terminate);
347 
348 	svc_run();
349 	yp_error("svc_run returned");
350 	exit(1);
351 	/* NOTREACHED */
352 }
353