yp*(8): Fix building with -fno-common.
[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  * $FreeBSD: src/usr.sbin/rpc.yppasswdd/yppasswdd_main.c,v 1.27 2008/10/30 01:54:31 rafan Exp $
33  */
34
35 #include <sys/param.h>
36 #include <sys/ioctl.h>
37 #include <sys/stat.h>
38 #include <sys/socket.h>
39 #include <sys/time.h>
40 #include <sys/resource.h>
41 #include <netinet/in.h>
42
43 #include <err.h>
44 #include <errno.h>
45 #include <fcntl.h>
46 #include <memory.h>
47 #include <signal.h>
48 #include <stdio.h>
49 #include <stdlib.h> /* getenv, exit */
50 #include <string.h> /* strcmp */
51 #include <syslog.h>
52 #include <unistd.h>
53
54 #include <rpc/rpc.h>
55 #include <rpc/rpc_com.h>
56 #include <rpc/pmap_clnt.h> /* for pmap_unset */
57 #include <rpcsvc/yp.h>
58 #include <rpcsvc/ypclnt.h>
59
60 #include "yppasswd.h"
61 #include "yppasswdd_extern.h"
62 #include "yppasswd_private.h"
63 #include "ypxfr_extern.h"
64 #include "yp_extern.h"
65
66 #ifndef SIG_PF
67 #define SIG_PF void(*)(int)
68 #endif
69
70 #ifdef DEBUG
71 #define RPC_SVC_FG
72 #endif
73
74 #define _RPCSVC_CLOSEDOWN 120
75 int _rpcpmstart = 0;            /* Started by a port monitor ? */
76 static int _rpcfdtype;
77                  /* Whether Stream or Datagram ? */
78         /* States a server can be in wrt request */
79
80 #define _IDLE 0
81 #define _SERVED 1
82 #define _SERVING 2
83 int debug;
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 const 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         signal(SIGALRM, (SIG_PF) closedown);
147         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         debug = 1;
175
176         while ((ch = getopt(argc, argv, "t:d:p:sfamuivh")) != -1) {
177                 switch (ch) {
178                 case 't':
179                         passfile_default = optarg;
180                         break;
181                 case 'd':
182                         yppasswd_domain = optarg;
183                         break;
184                 case 's':
185                         no_chsh++;
186                         break;
187                 case 'f':
188                         no_chfn++;
189                         break;
190                 case 'p':
191                         yp_dir = optarg;
192                         break;
193                 case 'a':
194                         allow_additions++;
195                         break;
196                 case 'm':
197                         multidomain++;
198                         break;
199                 case 'i':
200                         inplace++;
201                         break;
202                 case 'v':
203                         verbose++;
204                         break;
205                 case 'u':
206                         resvport = 0;
207                         break;
208                 default:
209                 case 'h':
210                         usage();
211                         break;
212                 }
213         }
214
215         if (yppasswd_domain == NULL) {
216                 if (yp_get_default_domain(&yppasswd_domain)) {
217                         yp_error("no domain specified and system domain \
218 name isn't set -- aborting");
219                 usage();
220                 }
221         }
222
223         load_securenets();
224
225         if (getrpcport(_localhost, YPPROG, YPVERS, IPPROTO_UDP) <= 0) {
226                 yp_error("no ypserv processes registered with local portmap");
227                 yp_error("this host is not an NIS server -- aborting");
228                 exit(1);
229         }
230
231         if ((mastername = ypxfr_get_master(yppasswd_domain,
232                  _passwd_byname, _localhost, 0)) == NULL) {
233                 yp_error("can't get name of NIS master server for domain %s",
234                                                         yppasswd_domain);
235                 exit(1);
236         }
237
238         if (gethostname((char *)&myname, sizeof(myname)) == -1) {
239                 yp_error("can't get local hostname: %s", strerror(errno));
240                 exit(1);
241         }
242
243         if (strncasecmp(mastername, (char *)&myname, sizeof(myname))) {
244                 yp_error("master of %s is %s, but we are %s",
245                         "passwd.byname", mastername, myname);
246                 yp_error("this host is not the NIS master server for \
247 the %s domain -- aborting", yppasswd_domain);
248                 exit(1);
249         }
250
251         debug = 0;
252
253         if (getsockname(0, (struct sockaddr *)&saddr, &asize) == 0) {
254                 socklen_t ssize = sizeof (int);
255                 if (saddr.sin_family != AF_INET)
256                         exit(1);
257                 if (getsockopt(0, SOL_SOCKET, SO_TYPE,
258                     (char *)&_rpcfdtype, &ssize) == -1)
259                         exit(1);
260                 _rpcpmstart = 1;
261         }
262
263         if (!debug && _rpcpmstart == 0) {
264                 if (daemon(0,0)) {
265                         err(1,"cannot fork");
266                 }
267         }
268         openlog("rpc.yppasswdd", LOG_PID, LOG_DAEMON);
269         memset(&sa, 0, sizeof(sa));
270         sa.sa_flags = SA_NOCLDWAIT;
271         sigaction(SIGCHLD, &sa, NULL);
272
273         rpcb_unset(YPPASSWDPROG, YPPASSWDVERS, NULL);
274         rpcb_unset(MASTER_YPPASSWDPROG, MASTER_YPPASSWDVERS, NULL);
275
276         rpc_control(RPC_SVC_CONNMAXREC_SET, &maxrec);
277
278         if (svc_create(yppasswdprog_1, YPPASSWDPROG, YPPASSWDVERS, "netpath") == 0) {
279                 yp_error("cannot create yppasswd service.");
280                 exit(1);
281         }
282         if (svc_create(master_yppasswdprog_1, MASTER_YPPASSWDPROG,
283             MASTER_YPPASSWDVERS, "netpath") == 0) {
284                 yp_error("cannot create master_yppasswd service.");
285                 exit(1);
286         }
287
288         nconf = NULL;
289         localhandle = setnetconfig();
290         while ((nconf = getnetconfig(localhandle)) != NULL) {
291                 if (nconf->nc_protofmly != NULL &&
292                     strcmp(nconf->nc_protofmly, NC_LOOPBACK) == 0)
293                         break;
294         }
295         if (nconf == NULL) {
296                 yp_error("getnetconfigent unix: %s", nc_sperror());
297                 exit(1);
298         }
299         unlink(sockname);
300         transp = svcunix_create(RPC_ANYSOCK, 0, 0, sockname);
301         if (transp == NULL) {
302                 yp_error("cannot create AF_LOCAL service.");
303                 exit(1);
304         }
305         if (!svc_reg(transp, MASTER_YPPASSWDPROG, MASTER_YPPASSWDVERS,
306             master_yppasswdprog_1, nconf)) {
307                 yp_error("unable to register (MASTER_YPPASSWDPROG, \
308                     MASTER_YPPASSWDVERS, unix).");
309                 exit(1);
310         }
311         endnetconfig(localhandle);
312
313         /* Only root may connect() to the AF_UNIX link. */
314         if (chmod(sockname, 0))
315                 err(1, "chmod of %s failed", sockname);
316
317         if (transp == NULL) {
318                 yp_error("could not create a handle");
319                 exit(1);
320         }
321         if (_rpcpmstart) {
322                 signal(SIGALRM, (SIG_PF) closedown);
323                 alarm(_RPCSVC_CLOSEDOWN/2);
324         }
325
326         /* Unlimited resource limits. */
327         rlim.rlim_cur = rlim.rlim_max = RLIM_INFINITY;
328         setrlimit(RLIMIT_CPU, &rlim);
329         setrlimit(RLIMIT_FSIZE, &rlim);
330         setrlimit(RLIMIT_STACK, &rlim);
331         setrlimit(RLIMIT_DATA, &rlim);
332         setrlimit(RLIMIT_RSS, &rlim);
333
334         /* Don't drop core (not really necessary, but GP's). */
335         rlim.rlim_cur = rlim.rlim_max = 0;
336         setrlimit(RLIMIT_CORE, &rlim);
337
338         /* Turn off signals. */
339         signal(SIGALRM, SIG_IGN);
340         signal(SIGHUP, (SIG_PF) reload);
341         signal(SIGINT, SIG_IGN);
342         signal(SIGPIPE, SIG_IGN);
343         signal(SIGQUIT, SIG_IGN);
344         signal(SIGTERM, (SIG_PF) terminate);
345
346         svc_run();
347         yp_error("svc_run returned");
348         exit(1);
349         /* NOTREACHED */
350 }