1 /*
2  * Copyright (c) 1999 Martin Blapp
3  * 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  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD: src/usr.sbin/rpc.umntall/rpc.umntall.c,v 1.3.2.1 2001/12/13 01:27:20 iedowse Exp $
27  * $DragonFly: src/usr.sbin/rpc.umntall/rpc.umntall.c,v 1.6 2005/04/02 20:49:56 joerg Exp $
28  */
29 
30 #include <sys/param.h>
31 #include <sys/ucred.h>
32 #include <sys/mount.h>
33 
34 #include <rpc/rpc.h>
35 #include <rpc/pmap_clnt.h>
36 #include <nfs/rpcv2.h>
37 
38 #include <err.h>
39 #include <netdb.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <unistd.h>
44 #include <resolv.h>
45 
46 #include "mounttab.h"
47 
48 int verbose;
49 int fastopt;
50 
51 static int do_umount (char *, char *);
52 static int do_umntall (char *);
53 static int is_mounted (char *, char *);
54 static void usage (void);
55 static struct hostent *gethostbyname_quick(const char *name);
56 
57 int	xdr_dir (XDR *, char *);
58 
59 int
60 main(int argc, char **argv) {
61 	int ch, keep, success, pathlen;
62 	time_t expire, now;
63 	char *host, *path;
64 	struct mtablist *mtab;
65 
66 	expire = 0;
67 	host = path = NULL;
68 	success = keep = verbose = 0;
69 	while ((ch = getopt(argc, argv, "h:kp:vfe:")) != -1)
70 		switch (ch) {
71 		case 'h':
72 			host = optarg;
73 			break;
74 		case 'e':
75 			expire = atoi(optarg);
76 			break;
77 		case 'k':
78 			keep = 1;
79 			break;
80 		case 'p':
81 			path = optarg;
82 			break;
83 		case 'v':
84 			verbose = 1;
85 			break;
86 		case 'f':
87 			fastopt = 1;
88 			break;
89 		case '?':
90 			usage();
91 		default:
92 			break;
93 		}
94 	argc -= optind;
95 	argv += optind;
96 
97 	/* Default expiretime is one day */
98 	if (expire == 0)
99 		expire = 86400;
100 	time(&now);
101 
102 	/* Read PATH_MOUNTTAB. */
103 	if (!read_mtab()) {
104 		if (verbose)
105 			warnx("no mounttab entries (%s does not exist)",
106 			    PATH_MOUNTTAB);
107 		mtabhead = NULL;
108 	}
109 
110 	if (host == NULL && path == NULL) {
111 		/* Check each entry and do any necessary unmount RPCs. */
112 		for (mtab = mtabhead; mtab != NULL; mtab = mtab->mtab_next) {
113 			if (*mtab->mtab_host == '\0')
114 				continue;
115 			if (mtab->mtab_time + expire < now) {
116 				/* Clear expired entry. */
117 				if (verbose)
118 					warnx("remove expired entry %s:%s",
119 					    mtab->mtab_host, mtab->mtab_dirp);
120 				bzero(mtab->mtab_host,
121 				    sizeof(mtab->mtab_host));
122 				continue;
123 			}
124 			if (keep && is_mounted(mtab->mtab_host,
125 			    mtab->mtab_dirp)) {
126 				if (verbose)
127 					warnx("skip entry %s:%s",
128 					    mtab->mtab_host, mtab->mtab_dirp);
129 				continue;
130 			}
131 			if (do_umount(mtab->mtab_host, mtab->mtab_dirp)) {
132 				if (verbose)
133 					warnx("umount RPC for %s:%s succeeded",
134 					    mtab->mtab_host, mtab->mtab_dirp);
135 				/* Remove all entries for this host + path. */
136 				clean_mtab(mtab->mtab_host, mtab->mtab_dirp,
137 				    verbose);
138 			}
139 		}
140 		success = 1;
141 	} else {
142 		if (host == NULL && path != NULL)
143 			/* Missing hostname. */
144 			usage();
145 		if (path == NULL) {
146 			/* Do a RPC UMNTALL for this specific host */
147 			success = do_umntall(host);
148 			if (verbose && success)
149 				warnx("umntall RPC for %s succeeded", host);
150 		} else {
151 			/* Do a RPC UMNTALL for this specific mount */
152 			for (pathlen = strlen(path);
153 			    pathlen > 1 && path[pathlen - 1] == '/'; pathlen--)
154 				path[pathlen - 1] = '\0';
155 			success = do_umount(host, path);
156 			if (verbose && success)
157 				warnx("umount RPC for %s:%s succeeded", host,
158 				    path);
159 		}
160 		/* If successful, remove any corresponding mounttab entries. */
161 		if (success)
162 			clean_mtab(host, path, verbose);
163 	}
164 	/* Write and unlink PATH_MOUNTTAB if necessary */
165 	if (success)
166 		success = write_mtab(verbose);
167 	free_mtab();
168 	exit (success ? 0 : 1);
169 }
170 
171 /*
172  * Send a RPC_MNT UMNTALL request to hostname.
173  * XXX This works for all mountd implementations,
174  * but produces a RPC IOERR on non FreeBSD systems.
175  */
176 int
177 do_umntall(char *hostname) {
178 	enum clnt_stat clnt_stat;
179 	struct hostent *hp;
180 	struct sockaddr_in saddr;
181 	struct timeval pertry, try;
182 	int so;
183 	CLIENT *clp;
184 
185 	if ((hp = gethostbyname_quick(hostname)) == NULL) {
186 		warnx("gethostbyname(%s) failed", hostname);
187 		return (0);
188 	}
189 	memset(&saddr, 0, sizeof(saddr));
190 	saddr.sin_family = AF_INET;
191 	saddr.sin_port = 0;
192 	memmove(&saddr.sin_addr, hp->h_addr, MIN(hp->h_length,
193 	    (int)sizeof(saddr.sin_addr)));
194 	pertry.tv_sec = 3;
195 	pertry.tv_usec = 0;
196 	if (fastopt)
197 	    pmap_getport_timeout(NULL, &pertry);
198 	so = RPC_ANYSOCK;
199 	clp = clntudp_create(&saddr, RPCPROG_MNT, RPCMNT_VER1, pertry, &so);
200 	if (clp == NULL) {
201 		warnx("%s: %s", hostname, clnt_spcreateerror("RPCPROG_MNT"));
202 		return (0);
203 	}
204 	clp->cl_auth = authunix_create_default();
205 	try.tv_sec = 3;
206 	try.tv_usec = 0;
207 	clnt_stat = clnt_call(clp, RPCMNT_UMNTALL, xdr_void, (caddr_t)0,
208 	    xdr_void, (caddr_t)0, try);
209 	if (clnt_stat != RPC_SUCCESS)
210 		warnx("%s: %s", hostname, clnt_sperror(clp, "RPCMNT_UMNTALL"));
211 	auth_destroy(clp->cl_auth);
212 	clnt_destroy(clp);
213 	return (clnt_stat == RPC_SUCCESS);
214 }
215 
216 /*
217  * Send a RPC_MNT UMOUNT request for dirp to hostname.
218  */
219 int
220 do_umount(char *hostname, char *dirp) {
221 	enum clnt_stat clnt_stat;
222 	struct hostent *hp;
223 	struct sockaddr_in saddr;
224 	struct timeval pertry, try;
225 	CLIENT *clp;
226 	int so;
227 
228 	if ((hp = gethostbyname_quick(hostname)) == NULL) {
229 		warnx("gethostbyname(%s) failed", hostname);
230 		return (0);
231 	}
232 	memset(&saddr, 0, sizeof(saddr));
233 	saddr.sin_family = AF_INET;
234 	saddr.sin_port = 0;
235 	memmove(&saddr.sin_addr, hp->h_addr, MIN(hp->h_length,
236 		(int)sizeof(saddr.sin_addr)));
237 	pertry.tv_sec = 3;
238 	pertry.tv_usec = 0;
239 	if (fastopt)
240 	    pmap_getport_timeout(NULL, &pertry);
241 	so = RPC_ANYSOCK;
242 	clp = clntudp_create(&saddr, RPCPROG_MNT, RPCMNT_VER1, pertry, &so);
243 	if (clp  == NULL) {
244 		warnx("%s: %s", hostname, clnt_spcreateerror("RPCPROG_MNT"));
245 		return (0);
246 	}
247 	clp->cl_auth = authunix_create_default();
248 	try.tv_sec = 3;
249 	try.tv_usec = 0;
250 	clnt_stat = clnt_call(clp, RPCMNT_UMOUNT, xdr_dir, dirp,
251 	    xdr_void, (caddr_t)0, try);
252 	if (clnt_stat != RPC_SUCCESS)
253 		warnx("%s: %s", hostname, clnt_sperror(clp, "RPCMNT_UMOUNT"));
254 	auth_destroy(clp->cl_auth);
255 	clnt_destroy(clp);
256 	return (clnt_stat == RPC_SUCCESS);
257 }
258 
259 /*
260  * Check if the entry is still/already mounted.
261  */
262 int
263 is_mounted(char *hostname, char *dirp) {
264 	struct statfs *mntbuf;
265 	char name[MNAMELEN + 1];
266 	size_t bufsize;
267 	int mntsize, i;
268 
269 	if (strlen(hostname) + strlen(dirp) >= MNAMELEN)
270 		return (0);
271 	snprintf(name, sizeof(name), "%s:%s", hostname, dirp);
272 	mntsize = getfsstat(NULL, 0, MNT_NOWAIT);
273 	if (mntsize <= 0)
274 		return (0);
275 	bufsize = (mntsize + 1) * sizeof(struct statfs);
276 	if ((mntbuf = malloc(bufsize)) == NULL)
277 		err(1, "malloc");
278 	mntsize = getfsstat(mntbuf, (long)bufsize, MNT_NOWAIT);
279 	for (i = mntsize - 1; i >= 0; i--) {
280 		if (strcmp(mntbuf[i].f_mntfromname, name) == 0) {
281 			free(mntbuf);
282 			return (1);
283 		}
284 	}
285 	free(mntbuf);
286 	return (0);
287 }
288 
289 /*
290  * rpc.umntall is often called at boot.  If the network or target host has
291  * issues we want to limit resolver stalls so rpc.umntall doesn't stall
292  * the boot sequence forever.
293  */
294 struct hostent *
295 gethostbyname_quick(const char *name)
296 {
297     struct hostent *he;
298     int save_retrans;
299     int save_retry;
300 
301     save_retrans = _res.retrans;
302     save_retry = _res.retry;
303     if (fastopt) {
304 	_res.retrans = 1;
305 	_res.retry = 2;
306     }
307     he = gethostbyname(name);
308     if (fastopt) {
309 	_res.retrans = save_retrans;
310 	_res.retry = save_retry;
311     }
312     return(he);
313 }
314 
315 /*
316  * xdr routines for mount rpc's
317  */
318 int
319 xdr_dir(XDR *xdrsp, char *dirp) {
320 	return (xdr_string(xdrsp, &dirp, RPCMNT_PATHLEN));
321 }
322 
323 static void
324 usage() {
325 	fprintf(stderr, "%s\n",
326 	    "usage: rpc.umntall [-kv] [-e expire] [-h host] [-p path]");
327 	exit(1);
328 }
329