umount: cleanup and code shrink
[oweals/busybox.git] / util-linux / umount.c
index 48efbd3494352a8a245dadb42885822072c80bc0..5597d9eba65e87339d81c23f85bd33b0c9b8b88b 100644 (file)
+/* vi: set sw=4 ts=4: */
 /*
  * Mini umount implementation for busybox
  *
- * Copyright (C) 1998 by Erik Andersen <andersee@debian.org>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
+ * Copyright (C) 2005 by Rob Landley <rob@landley.net>
  *
+ * Licensed under GPLv2, see file LICENSE in this source tree.
  */
-
-#include "internal.h"
-#include <stdio.h>
-#include <sys/mount.h>
 #include <mntent.h>
-#include <fstab.h>
-#include <errno.h>
-
-static const char umount_usage[] = 
-"Usage: umount filesystem\n"
-"   or: umount directory\n"
-"   or: umount -a"
-"to unmount all mounted file systems.\n";
+#include <sys/mount.h>
+#include "libbb.h"
 
-static int
-umount_all()
+#if defined(__dietlibc__)
+// TODO: This does not belong here.
+/* 16.12.2006, Sampo Kellomaki (sampo@iki.fi)
+ * dietlibc-0.30 does not have implementation of getmntent_r() */
+static struct mntent *getmntent_r(FILE* stream, struct mntent* result,
+               char* buffer UNUSED_PARAM, int bufsize UNUSED_PARAM)
 {
-       int status;
-       struct mntent *m;
-        FILE *mountTable;
-
-        if ((mountTable = setmntent ("/proc/mounts", "r"))) {
-            while ((m = getmntent (mountTable)) != 0) {
-                char *blockDevice = m->mnt_fsname;
-                if (strcmp (blockDevice, "/dev/root") == 0)
-                    blockDevice = (getfsfile ("/"))->fs_spec;
-               status=umount (m->mnt_dir);
-               if (status!=0) {
-                   /* Don't bother retrying the umount on busy devices */
-                   if (errno==EBUSY) {
-                       perror(m->mnt_dir); 
-                       continue;
-                   }
-                   printf ("Trying to umount %s failed: %s\n", 
-                           m->mnt_dir, strerror(errno)); 
-                   printf ("Instead trying to umount %s\n", blockDevice); 
-                   status=umount (blockDevice);
-                   if (status!=0) {
-                       printf ("Couldn't umount %s on %s (type %s): %s\n", 
-                               blockDevice, m->mnt_dir, m->mnt_type, strerror(errno));
-                   }
-               }
-            }
-            endmntent (mountTable);
-        }
-        return( TRUE);
+       struct mntent* ment = getmntent(stream);
+       return memcpy(result, ment, sizeof(*ment));
 }
+#endif
 
-extern int
-umount_main(int argc, char * * argv)
+/* ignored: -v -d -t -i */
+#define OPTION_STRING           "fldnra" "vdt:i"
+#define OPT_FORCE               (1 << 0) // Same as MNT_FORCE
+#define OPT_LAZY                (1 << 1) // Same as MNT_DETACH
+#define OPT_FREELOOP            (1 << 2)
+#define OPT_NO_MTAB             (1 << 3)
+#define OPT_REMOUNT             (1 << 4)
+#define OPT_ALL                 (ENABLE_FEATURE_UMOUNT_ALL ? (1 << 5) : 0)
+
+int umount_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int umount_main(int argc UNUSED_PARAM, char **argv)
 {
+       int doForce;
+       struct mntent me;
+       FILE *fp;
+       char *fstype = NULL;
+       int status = EXIT_SUCCESS;
+       unsigned opt;
+       struct mtab_list {
+               char *dir;
+               char *device;
+               struct mtab_list *next;
+       } *mtl, *m;
+
+       opt = getopt32(argv, OPTION_STRING, &fstype);
+       //argc -= optind;
+       argv += optind;
+
+       // MNT_FORCE and MNT_DETACH (from linux/fs.h) must match
+       // OPT_FORCE and OPT_LAZY, otherwise this trick won't work:
+       doForce = MAX((opt & OPT_FORCE), (opt & OPT_LAZY));
+
+       /* Get a list of mount points from mtab.  We read them all in now mostly
+        * for umount -a (so we don't have to worry about the list changing while
+        * we iterate over it, or about getting stuck in a loop on the same failing
+        * entry.  Notice that this also naturally reverses the list so that -a
+        * umounts the most recent entries first. */
+       m = mtl = NULL;
 
-    if (argc < 2) {
-       usage( umount_usage); 
-    }
-    argc--;
-    argv++;
-
-    /* Parse any options */
-    while (**argv == '-') {
-       while (*++(*argv)) switch (**argv) {
-           case 'a':
-               exit ( umount_all() );
-               break;
-           default:
-               usage( umount_usage);
+       // If we're umounting all, then m points to the start of the list and
+       // the argument list should be empty (which will match all).
+       fp = setmntent(bb_path_mtab_file, "r");
+       if (!fp) {
+               if (opt & OPT_ALL)
+                       bb_error_msg_and_die("can't open '%s'", bb_path_mtab_file);
+       } else {
+               while (getmntent_r(fp, &me, bb_common_bufsiz1, sizeof(bb_common_bufsiz1))) {
+                       /* Match fstype if passed */
+                       if (!match_fstype(&me, fstype))
+                               continue;
+                       m = xzalloc(sizeof(*m));
+                       m->next = mtl;
+                       m->device = xstrdup(me.mnt_fsname);
+                       m->dir = xstrdup(me.mnt_dir);
+                       mtl = m;
+               }
+               endmntent(fp);
        }
-    }
-    if ( umount(*argv) == 0 )
-            exit (TRUE);
-    else {
-       perror("umount");
-       exit( FALSE);
-    }
-}
 
+       // If we're not umounting all, we need at least one argument.
+       if (!(opt & OPT_ALL) && !fstype) {
+               if (!argv[0])
+                       bb_show_usage();
+               m = NULL;
+       }
+
+       // Loop through everything we're supposed to umount, and do so.
+       for (;;) {
+               int curstat;
+               char *zapit = *argv;
+               char *path;
+
+               // Do we already know what to umount this time through the loop?
+               if (m)
+                       path = xstrdup(m->dir);
+               // For umount -a, end of mtab means time to exit.
+               else if (opt & OPT_ALL)
+                       break;
+               // Use command line argument (and look it up in mtab list)
+               else {
+                       if (!zapit)
+                               break;
+                       argv++;
+                       path = xmalloc_realpath(zapit);
+                       if (path) {
+                               for (m = mtl; m; m = m->next)
+                                       if (strcmp(path, m->dir) == 0 || strcmp(path, m->device) == 0)
+                                               break;
+                       }
+               }
+               // If we couldn't find this sucker in /etc/mtab, punt by passing our
+               // command line argument straight to the umount syscall.  Otherwise,
+               // umount the directory even if we were given the block device.
+               if (m) zapit = m->dir;
+
+               // Let's ask the thing nicely to unmount.
+               curstat = umount(zapit);
+
+               // Force the unmount, if necessary.
+               if (curstat && doForce)
+                       curstat = umount2(zapit, doForce);
+
+               // If still can't umount, maybe remount read-only?
+               if (curstat) {
+                       if ((opt & OPT_REMOUNT) && errno == EBUSY && m) {
+                               // Note! Even if we succeed here, later we should not
+                               // free loop device or erase mtab entry!
+                               const char *msg = "%s busy - remounted read-only";
+                               curstat = mount(m->device, zapit, NULL, MS_REMOUNT|MS_RDONLY, NULL);
+                               if (curstat) {
+                                       msg = "can't remount %s read-only";
+                                       status = EXIT_FAILURE;
+                               }
+                               bb_error_msg(msg, m->device);
+                       } else {
+                               status = EXIT_FAILURE;
+                               bb_perror_msg("can't %sumount %s", (doForce ? "forcibly " : ""), zapit);
+                       }
+               } else {
+                       // De-allocate the loop device.  This ioctl should be ignored on
+                       // any non-loop block devices.
+                       if (ENABLE_FEATURE_MOUNT_LOOP && (opt & OPT_FREELOOP) && m)
+                               del_loop(m->device);
+                       if (ENABLE_FEATURE_MTAB_SUPPORT && !(opt & OPT_NO_MTAB) && m)
+                               erase_mtab(m->dir);
+               }
+
+               // Find next matching mtab entry for -a or umount /dev
+               // Note this means that "umount /dev/blah" will unmount all instances
+               // of /dev/blah, not just the most recent.
+               if (m) {
+                       while ((m = m->next) != NULL)
+                               // NB: if m is non-NULL, path is non-NULL as well
+                               if ((opt & OPT_ALL) || strcmp(path, m->device) == 0)
+                                       break;
+               }
+               free(path);
+       }
+
+       // Free mtab list if necessary
+       if (ENABLE_FEATURE_CLEAN_UP) {
+               while (mtl) {
+                       m = mtl->next;
+                       free(mtl->device);
+                       free(mtl->dir);
+                       free(mtl);
+                       mtl = m;
+               }
+       }
+
+       return status;
+}