usr.sbin/rpc.umntall/rpc.umntall.c

   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.13 2005/05/27 00:05:16 mux Exp $
  27  */
  28
  29 #include <sys/param.h>
  30 #include <sys/ucred.h>
  31 #include <sys/mount.h>
  32
  33 #include <rpc/rpc.h>
  34 #include <rpc/pmap_clnt.h>
  35 #include <vfs/nfs/rpcv2.h>
  36
  37 #include <err.h>
  38 #include <netdb.h>
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string.h>
  42 #include <unistd.h>
  43 #include <resolv.h>
  44
  45 #include "mounttab.h"
  46
  47 int verbose;
  48 int fastopt;
  49
  50 static int do_umount (char *, char *);
  51 static int do_umntall (char *);
  52 static int is_mounted (char *, char *);
  53 static void usage (void);
  54
  55 int     xdr_dir (XDR *, char *);
  56
  57 int
  58 main(int argc, char **argv) {
  59         int ch, keep, success, pathlen;
  60         time_t expire, now;
  61         char *host, *path;
  62         struct mtablist *mtab;
  63
  64         expire = 0;
  65         host = path = NULL;
  66         success = keep = verbose = 0;
  67         while ((ch = getopt(argc, argv, "h:kp:ve:")) != -1)
  68                 switch (ch) {
  69                 case 'h':
  70                         host = optarg;
  71                         break;
  72                 case 'e':
  73                         expire = atoi(optarg);
  74                         break;
  75                 case 'k':
  76                         keep = 1;
  77                         break;
  78                 case 'p':
  79                         path = optarg;
  80                         break;
  81                 case 'v':
  82                         verbose = 1;
  83                         break;
  84                 case '?':
  85                         usage();
  86                 default:
  87                         break;
  88                 }
  89         argc -= optind;
  90         argv += optind;
  91
  92         /* Default expiretime is one day */
  93         if (expire == 0)
  94                 expire = 86400;
  95         time(&now);
  96
  97         /* Read PATH_MOUNTTAB. */
  98         if (!read_mtab()) {
  99                 if (verbose)
 100                         warnx("no mounttab entries (%s does not exist)",
 101                             PATH_MOUNTTAB);
 102                 mtabhead = NULL;
 103         }
 104
 105         if (host == NULL && path == NULL) {
 106                 /* Check each entry and do any necessary unmount RPCs. */
 107                 for (mtab = mtabhead; mtab != NULL; mtab = mtab->mtab_next) {
 108                         if (*mtab->mtab_host == '\0')
 109                                 continue;
 110                         if (mtab->mtab_time + expire < now) {
 111                                 /* Clear expired entry. */
 112                                 if (verbose)
 113                                         warnx("remove expired entry %s:%s",
 114                                             mtab->mtab_host, mtab->mtab_dirp);
 115                                 bzero(mtab->mtab_host,
 116                                     sizeof(mtab->mtab_host));
 117                                 continue;
 118                         }
 119                         if (keep && is_mounted(mtab->mtab_host,
 120                             mtab->mtab_dirp)) {
 121                                 if (verbose)
 122                                         warnx("skip entry %s:%s",
 123                                             mtab->mtab_host, mtab->mtab_dirp);
 124                                 continue;
 125                         }
 126                         if (do_umount(mtab->mtab_host, mtab->mtab_dirp)) {
 127                                 if (verbose)
 128                                         warnx("umount RPC for %s:%s succeeded",
 129                                             mtab->mtab_host, mtab->mtab_dirp);
 130                                 /* Remove all entries for this host + path. */
 131                                 clean_mtab(mtab->mtab_host, mtab->mtab_dirp,
 132                                     verbose);
 133                         }
 134                 }
 135                 success = 1;
 136         } else {
 137                 if (host == NULL && path != NULL)
 138                         /* Missing hostname. */
 139                         usage();
 140                 if (path == NULL) {
 141                         /* Do a RPC UMNTALL for this specific host */
 142                         success = do_umntall(host);
 143                         if (verbose && success)
 144                                 warnx("umntall RPC for %s succeeded", host);
 145                 } else {
 146                         /* Do a RPC UMNTALL for this specific mount */
 147                         for (pathlen = strlen(path);
 148                             pathlen > 1 && path[pathlen - 1] == '/'; pathlen--)
 149                                 path[pathlen - 1] = '\0';
 150                         success = do_umount(host, path);
 151                         if (verbose && success)
 152                                 warnx("umount RPC for %s:%s succeeded", host,
 153                                     path);
 154                 }
 155                 /* If successful, remove any corresponding mounttab entries. */
 156                 if (success)
 157                         clean_mtab(host, path, verbose);
 158         }
 159         /* Write and unlink PATH_MOUNTTAB if necessary */
 160         if (success)
 161                 success = write_mtab(verbose);
 162         free_mtab();
 163         exit (success ? 0 : 1);
 164 }
 165
 166 /*
 167  * Send a RPC_MNT UMNTALL request to hostname.
 168  * XXX This works for all mountd implementations,
 169  * but produces a RPC IOERR on non FreeBSD systems.
 170  */
 171 int
 172 do_umntall(char *hostname) {
 173         enum clnt_stat clnt_stat;
 174         struct timeval try;
 175         CLIENT *clp;
 176
 177         try.tv_sec = 3;
 178         try.tv_usec = 0;
 179         clp = clnt_create_timed(hostname, RPCPROG_MNT, RPCMNT_VER1, "udp",
 180             &try);
 181         if (clp == NULL) {
 182                 warnx("%s: %s", hostname, clnt_spcreateerror("RPCPROG_MNT"));
 183                 return (0);
 184         }
 185         clp->cl_auth = authunix_create_default();
 186         clnt_stat = clnt_call(clp, RPCMNT_UMNTALL,
 187             (xdrproc_t)xdr_void, (caddr_t)0,
 188             (xdrproc_t)xdr_void, (caddr_t)0, try);
 189         if (clnt_stat != RPC_SUCCESS)
 190                 warnx("%s: %s", hostname, clnt_sperror(clp, "RPCMNT_UMNTALL"));
 191         auth_destroy(clp->cl_auth);
 192         clnt_destroy(clp);
 193         return (clnt_stat == RPC_SUCCESS);
 194 }
 195
 196 /*
 197  * Send a RPC_MNT UMOUNT request for dirp to hostname.
 198  */
 199 int
 200 do_umount(char *hostname, char *dirp) {
 201         enum clnt_stat clnt_stat;
 202         struct timeval try;
 203         CLIENT *clp;
 204
 205         try.tv_sec = 3;
 206         try.tv_usec = 0;
 207         clp = clnt_create_timed(hostname, RPCPROG_MNT, RPCMNT_VER1, "udp",
 208             &try);
 209         if (clp  == NULL) {
 210                 warnx("%s: %s", hostname, clnt_spcreateerror("RPCPROG_MNT"));
 211                 return (0);
 212         }
 213         clp->cl_auth = authsys_create_default();
 214         clnt_stat = clnt_call(clp, RPCMNT_UMOUNT, (xdrproc_t)xdr_dir, dirp,
 215             (xdrproc_t)xdr_void, (caddr_t)0, try);
 216         if (clnt_stat != RPC_SUCCESS)
 217                 warnx("%s: %s", hostname, clnt_sperror(clp, "RPCMNT_UMOUNT"));
 218         auth_destroy(clp->cl_auth);
 219         clnt_destroy(clp);
 220         return (clnt_stat == RPC_SUCCESS);
 221 }
 222
 223 /*
 224  * Check if the entry is still/already mounted.
 225  */
 226 int
 227 is_mounted(char *hostname, char *dirp) {
 228         struct statfs *mntbuf;
 229         char name[MNAMELEN + 1];
 230         size_t bufsize;
 231         int mntsize, i;
 232
 233         if (strlen(hostname) + strlen(dirp) >= MNAMELEN)
 234                 return (0);
 235         snprintf(name, sizeof(name), "%s:%s", hostname, dirp);
 236         mntsize = getfsstat(NULL, 0, MNT_NOWAIT);
 237         if (mntsize <= 0)
 238                 return (0);
 239         bufsize = (mntsize + 1) * sizeof(struct statfs);
 240         if ((mntbuf = malloc(bufsize)) == NULL)
 241                 err(1, "malloc");
 242         mntsize = getfsstat(mntbuf, (long)bufsize, MNT_NOWAIT);
 243         for (i = mntsize - 1; i >= 0; i--) {
 244                 if (strcmp(mntbuf[i].f_mntfromname, name) == 0) {
 245                         free(mntbuf);
 246                         return (1);
 247                 }
 248         }
 249         free(mntbuf);
 250         return (0);
 251 }
 252
 253 /*
 254  * xdr routines for mount rpc's
 255  */
 256 int
 257 xdr_dir(XDR *xdrsp, char *dirp) {
 258         return (xdr_string(xdrsp, &dirp, RPCMNT_PATHLEN));
 259 }
 260
 261 static void
 262 usage(void) {
 263         fprintf(stderr, "%s\n",
 264             "usage: rpc.umntall [-kv] [-e expire] [-h host] [-p path]");
 265         exit(1);
 266 }