0842d56bf13fc0851d75eb74bac9a9e0465d5765
[dragonfly.git] / usr.sbin / rpc.ypupdated / ypupdated_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.ypupdated/ypupdated_main.c,v 1.10 2004/05/24 12:28:27 stefanf Exp $
33  */
34
35 #include "ypupdate_prot.h"
36 #include <stdio.h>
37 #include <stdlib.h> /* getenv, exit */
38 #include <rpc/pmap_clnt.h> /* for pmap_unset */
39 #include <rpc/rpc_com.h>
40 #include <string.h> /* strcmp */
41 #include <signal.h>
42 #ifdef __cplusplus
43 #include <sysent.h> /* getdtablesize, open */
44 #endif /* __cplusplus */
45 #include <memory.h>
46 #include <sys/socket.h>
47 #include <netinet/in.h>
48 #include <syslog.h>
49 #include <sys/wait.h>
50 #include <errno.h>
51 #include <err.h>
52 #include <unistd.h>
53 #include "ypupdated_extern.h"
54 #include "yp_extern.h"
55
56 #ifndef SIG_PF
57 #define SIG_PF void(*)(int)
58 #endif
59
60 #ifdef DEBUG
61 #define RPC_SVC_FG
62 #endif
63
64 #define _RPCSVC_CLOSEDOWN 120
65 int _rpcpmstart;                /* Started by a port monitor ? */
66 static int _rpcfdtype;
67                  /* Whether Stream or Datagram ? */
68         /* States a server can be in wrt request */
69
70 #define _IDLE 0
71 #define _SERVED 1
72 #define _SERVING 2
73
74 extern int _rpcsvcstate;         /* Set when a request is serviced */
75
76 char *progname = "rpc.ypupdated";
77 const char *yp_dir = "/var/yp/";
78
79 static void
80 _msgout(char *msg)
81 {
82 #ifdef RPC_SVC_FG
83         if (_rpcpmstart)
84                 syslog(LOG_ERR, "%s", msg);
85         else
86                 warnx("%s", msg);
87 #else
88         syslog(LOG_ERR, "%s", msg);
89 #endif
90 }
91
92 static void
93 closedown(int sig)
94 {
95         if (_rpcsvcstate == _IDLE) {
96                 extern fd_set svc_fdset;
97                 static int size;
98                 int i, openfd;
99
100                 if (_rpcfdtype == SOCK_DGRAM)
101                         exit(0);
102                 if (size == 0) {
103                         size = getdtablesize();
104                 }
105                 for (i = 0, openfd = 0; i < size && openfd < 2; i++)
106                         if (FD_ISSET(i, &svc_fdset))
107                                 openfd++;
108                 if (openfd <= 1)
109                         exit(0);
110         }
111         if (_rpcsvcstate == _SERVED)
112                 _rpcsvcstate = _IDLE;
113
114         signal(SIGALRM, (SIG_PF) closedown);
115         alarm(_RPCSVC_CLOSEDOWN/2);
116 }
117
118 static void
119 ypupdated_svc_run(void)
120 {
121 #ifdef FD_SETSIZE
122         fd_set readfds;
123 #else
124         int readfds;
125 #endif /* def FD_SETSIZE */
126         extern int forked;
127         int pid;
128         int fd_setsize = _rpc_dtablesize();
129
130         /* Establish the identity of the parent ypupdated process. */
131         pid = getpid();
132
133         for (;;) {
134 #ifdef FD_SETSIZE
135                 readfds = svc_fdset;
136 #else
137                 readfds = svc_fds;
138 #endif /* def FD_SETSIZE */
139                 switch (select(fd_setsize, &readfds, NULL, NULL, NULL)) {
140                 case -1:
141                         if (errno == EINTR) {
142                                 continue;
143                         }
144                         warn("svc_run: - select failed");
145                         return;
146                 case 0:
147                         continue;
148                 default:
149                         svc_getreqset(&readfds);
150                         if (forked && pid != getpid())
151                                 exit(0);
152                 }
153         }
154 }
155
156 static void
157 reaper(int sig)
158 {
159         int status;
160
161         if (sig == SIGHUP) {
162 #ifdef foo
163                 load_securenets();
164 #endif
165                 return;
166         }
167
168         if (sig == SIGCHLD) {
169                 while (wait3(&status, WNOHANG, NULL) > 0)
170                         children--;
171         } else {
172                 pmap_unset(YPU_PROG, YPU_VERS);
173                 exit(0);
174         }
175 }
176
177 __dead2 void
178 usage(void)
179 {
180         fprintf(stderr, "rpc.ypupdatedd [-p path]\n");
181         exit(0);
182 }
183
184 int
185 main(int argc, char *argv[])
186 {
187         SVCXPRT *transp = NULL;
188         int sock;
189         int proto = 0;
190         struct sockaddr_in saddr;
191         int asize = sizeof (saddr);
192         int ch;
193
194         while ((ch = getopt(argc, argv, "p:h")) != -1) {
195                 switch (ch) {
196                 case 'p':
197                         yp_dir = optarg;
198                         break;
199                 default:
200                         usage();
201                         break;
202                 }
203         }
204 #ifdef foo
205         load_securenets();
206 #endif
207
208         if (svc_auth_reg(AUTH_DES, _svcauth_des) == -1) {
209                 yp_error("failed to register AUTH_DES flavor");
210                 exit(1);
211         }
212
213         if (getsockname(0, (struct sockaddr *)&saddr, &asize) == 0) {
214                 int ssize = sizeof (int);
215
216                 if (saddr.sin_family != AF_INET)
217                         exit(1);
218                 if (getsockopt(0, SOL_SOCKET, SO_TYPE,
219                                 (char *)&_rpcfdtype, &ssize) == -1)
220                         exit(1);
221                 sock = 0;
222                 _rpcpmstart = 1;
223                 proto = 0;
224                 openlog("rpc.ypupdatedd", LOG_PID, LOG_DAEMON);
225         } else {
226 #ifndef RPC_SVC_FG
227                 if (daemon(0,0)) {
228                         err(1, "cannot fork");
229                 }
230                 openlog("rpc.ypupdated", LOG_PID, LOG_DAEMON);
231 #endif
232                 sock = RPC_ANYSOCK;
233                 pmap_unset(YPU_PROG, YPU_VERS);
234         }
235
236         if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_DGRAM)) {
237                 transp = svcudp_create(sock);
238                 if (transp == NULL) {
239                         _msgout("cannot create udp service.");
240                         exit(1);
241                 }
242                 if (!_rpcpmstart)
243                         proto = IPPROTO_UDP;
244                 if (!svc_register(transp, YPU_PROG, YPU_VERS, ypu_prog_1, proto)) {
245                         _msgout("unable to register (YPU_PROG, YPU_VERS, udp).");
246                         exit(1);
247                 }
248         }
249
250         if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_STREAM)) {
251                 transp = svctcp_create(sock, 0, 0);
252                 if (transp == NULL) {
253                         _msgout("cannot create tcp service.");
254                         exit(1);
255                 }
256                 if (!_rpcpmstart)
257                         proto = IPPROTO_TCP;
258                 if (!svc_register(transp, YPU_PROG, YPU_VERS, ypu_prog_1, proto)) {
259                         _msgout("unable to register (YPU_PROG, YPU_VERS, tcp).");
260                         exit(1);
261                 }
262         }
263
264         if (transp == NULL) {
265                 _msgout("could not create a handle");
266                 exit(1);
267         }
268         if (_rpcpmstart) {
269                 signal(SIGALRM, (SIG_PF) closedown);
270                 alarm(_RPCSVC_CLOSEDOWN/2);
271         }
272
273         signal(SIGPIPE, SIG_IGN);
274         signal(SIGCHLD, (SIG_PF) reaper);
275         signal(SIGTERM, (SIG_PF) reaper);
276         signal(SIGINT, (SIG_PF) reaper);
277         signal(SIGHUP, (SIG_PF) reaper);
278
279         ypupdated_svc_run();
280         _msgout("svc_run returned");
281         exit(1);
282         /* NOTREACHED */
283 }