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