Add the DragonFly cvs id and perform general cleanups on cvs/rcs/sccs ids. Most
[dragonfly.git] / sbin / mount_portal / mount_portal.c
1 /*
2  * Copyright (c) 1992, 1993, 1994
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software donated to Berkeley by
6  * Jan-Simon Pendry.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. All advertising materials mentioning features or use of this software
17  *    must display the following acknowledgement:
18  *      This product includes software developed by the University of
19  *      California, Berkeley and its contributors.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  *
36  * @(#) Copyright (c) 1992, 1993, 1994 The Regents of the University of California.  All rights reserved.
37  * @(#)mount_portal.c   8.6 (Berkeley) 4/26/95
38  * $FreeBSD: src/sbin/mount_portal/mount_portal.c,v 1.16 1999/10/09 11:54:11 phk Exp $
39  * $DragonFly: src/sbin/mount_portal/mount_portal.c,v 1.2 2003/06/17 04:27:33 dillon Exp $
40  */
41
42 #include <sys/param.h>
43 #include <sys/wait.h>
44 #include <sys/socket.h>
45 #include <sys/un.h>
46 #include <sys/stat.h>
47 #include <sys/syslog.h>
48 #include <sys/mount.h>
49
50 #include <err.h>
51 #include <errno.h>
52 #include <stdio.h>
53 #include <stdlib.h>
54 #include <string.h>
55 #include <sysexits.h>
56 #include <unistd.h>
57
58 #include "mntopts.h"
59 #include "pathnames.h"
60 #include "portald.h"
61
62 struct mntopt mopts[] = {
63         MOPT_STDOPTS,
64         { NULL }
65 };
66
67 static void usage __P((void)) __dead2;
68
69 static sig_atomic_t readcf;     /* Set when SIGHUP received */
70
71 static void sighup(sig)
72 int sig;
73 {
74         readcf ++;
75 }
76
77 static void sigchld(sig)
78 int sig;
79 {
80         pid_t pid;
81
82         while ((pid = waitpid((pid_t) -1, (int *) 0, WNOHANG)) > 0)
83                 ;
84         /* wrtp - waitpid _doesn't_ return 0 when no children! */
85 #ifdef notdef
86         if (pid < 0 && errno != ECHILD)
87                 syslog(LOG_WARNING, "waitpid: %s", strerror(errno));
88 #endif
89 }
90
91 int
92 main(argc, argv)
93         int argc;
94         char *argv[];
95 {
96         struct portal_args args;
97         struct sockaddr_un un;
98         char *conf;
99         char mountpt[MAXPATHLEN];
100         int mntflags = 0;
101         char tag[32];
102         struct vfsconf vfc;
103         mode_t um;
104
105         qelem q;
106         int rc;
107         int so;
108         int error = 0;
109
110         /*
111          * Crack command line args
112          */
113         int ch;
114
115         while ((ch = getopt(argc, argv, "o:")) != -1) {
116                 switch (ch) {
117                 case 'o':
118                         getmntopts(optarg, mopts, &mntflags, 0);
119                         break;
120                 default:
121                         error = 1;
122                         break;
123                 }
124         }
125
126         if (optind != (argc - 2))
127                 error = 1;
128
129         if (error)
130                 usage();
131
132         /*
133          * Get config file and mount point
134          */
135         conf = argv[optind];
136
137         /* resolve the mountpoint with realpath(3) */
138         (void)checkpath(argv[optind+1], mountpt);
139
140         /*
141          * Construct the listening socket
142          */
143         un.sun_family = AF_UNIX;
144         if (sizeof(_PATH_TMPPORTAL) >= sizeof(un.sun_path)) {
145                 errx(EX_SOFTWARE, "portal socket name too long");
146         }
147         strcpy(un.sun_path, _PATH_TMPPORTAL);
148         mktemp(un.sun_path);
149         un.sun_len = strlen(un.sun_path);
150
151         so = socket(AF_UNIX, SOCK_STREAM, 0);
152         if (so < 0) {
153                 err(EX_OSERR, "socket");
154         }
155         um = umask(077);
156         (void) unlink(un.sun_path);
157         if (bind(so, (struct sockaddr *) &un, sizeof(un)) < 0)
158                 err(1, NULL);
159
160         (void) unlink(un.sun_path);
161         (void) umask(um);
162
163         (void) listen(so, 5);
164
165         args.pa_socket = so;
166         sprintf(tag, "portal:%d", getpid());
167         args.pa_config = tag;
168
169         error = getvfsbyname("portal", &vfc);
170         if (error && vfsisloadable("portal")) {
171                 if (vfsload("portal"))
172                         err(EX_OSERR, "vfsload(portal)");
173                 endvfsent();
174                 error = getvfsbyname("portal", &vfc);
175         }
176         if (error)
177                 errx(EX_OSERR, "portal filesystem is not available");
178
179         rc = mount(vfc.vfc_name, mountpt, mntflags, &args);
180         if (rc < 0)
181                 err(1, NULL);
182
183         /*
184          * Everything is ready to go - now is a good time to fork
185          */
186 #ifndef DEBUG
187         daemon(0, 0);
188 #endif
189
190         /*
191          * Start logging (and change name)
192          */
193         openlog("portald", LOG_CONS|LOG_PID, LOG_DAEMON);
194
195         q.q_forw = q.q_back = &q;
196         readcf = 1;
197
198         signal(SIGCHLD, sigchld);
199         signal(SIGHUP, sighup);
200
201         /*
202          * Just loop waiting for new connections and activating them
203          */
204         for (;;) {
205                 struct sockaddr_un un2;
206                 int len2 = sizeof(un2);
207                 int so2;
208                 pid_t pid;
209                 fd_set fdset;
210                 int rc;
211
212                 /*
213                  * Check whether we need to re-read the configuration file
214                  */
215                 if (readcf) {
216 #ifdef DEBUG
217                         printf ("re-reading configuration file\n");
218 #endif
219                         readcf = 0;
220                         conf_read(&q, conf);
221                         continue;
222                 }
223
224                 /*
225                  * Accept a new connection
226                  * Will get EINTR if a signal has arrived, so just
227                  * ignore that error code
228                  */
229                 FD_ZERO(&fdset);
230                 FD_SET(so, &fdset);
231                 rc = select(so+1, &fdset, (fd_set *) 0, (fd_set *) 0, (struct timeval *) 0);
232                 if (rc < 0) {
233                         if (errno == EINTR)
234                                 continue;
235                         syslog(LOG_ERR, "select: %s", strerror(errno));
236                         exit(EX_OSERR);
237                 }
238                 if (rc == 0)
239                         break;
240                 so2 = accept(so, (struct sockaddr *) &un2, &len2);
241                 if (so2 < 0) {
242                         /*
243                          * The unmount function does a shutdown on the socket
244                          * which will generated ECONNABORTED on the accept.
245                          */
246                         if (errno == ECONNABORTED)
247                                 break;
248                         if (errno != EINTR) {
249                                 syslog(LOG_ERR, "accept: %s", strerror(errno));
250                                 exit(EX_OSERR);
251                         }
252                         continue;
253                 }
254
255                 /*
256                  * Now fork a new child to deal with the connection
257                  */
258         eagain:;
259                 switch (pid = fork()) {
260                 case -1:
261                         if (errno == EAGAIN) {
262                                 sleep(1);
263                                 goto eagain;
264                         }
265                         syslog(LOG_ERR, "fork: %s", strerror(errno));
266                         break;
267                 case 0:
268                         (void) close(so);
269                         activate(&q, so2);
270                         exit(0);
271                 default:
272                         (void) close(so2);
273                         break;
274                 }
275         }
276         syslog(LOG_INFO, "%s unmounted", mountpt);
277         exit(0);
278 }
279
280 static void
281 usage()
282 {
283         (void)fprintf(stderr,
284                 "usage: mount_portal [-o options] config mount-point\n");
285         exit(EX_USAGE);
286 }