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