pscan: size optimization (Tito <farmatito@tiscali.it>)
[oweals/busybox.git] / util-linux / umount.c
index 71ddbd4ed024b7b8d29d86338feff133d9c2e6eb..7063a474d1b7822be6c014cda98c406c6af71e0c 100644 (file)
@@ -5,39 +5,31 @@
  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  * Copyright (C) 2005 by Rob Landley <rob@landley.net>
  *
- * This program is licensed under the GNU General Public license (GPL)
- * version 2 or later, see http://www.fsf.org/licensing/licenses/gpl.html
- * or the file "LICENSE" in the busybox source tarball for the full text.
- *
+ * Licensed under GPL version 2, see file LICENSE in this tarball for details.
  */
 
-#include <limits.h>
-#include <stdio.h>
 #include <mntent.h>
-#include <errno.h>
-#include <string.h>
-#include <stdlib.h>
-#include <sys/mount.h>
 #include <getopt.h>
-#include "busybox.h"
+#include "libbb.h"
 
-#define OPTION_STRING          "flDnrva"
+#define OPTION_STRING          "flDnravdt:"
 #define OPT_FORCE                      1
 #define OPT_LAZY                       2
 #define OPT_DONTFREELOOP       4
 #define OPT_NO_MTAB                    8
 #define OPT_REMOUNT                    16
-#define OPT_IGNORED                    32      // -v is ignored
-#define OPT_ALL                                (ENABLE_FEATURE_UMOUNT_ALL ? 64 : 0)
+#define OPT_ALL                                (ENABLE_FEATURE_UMOUNT_ALL ? 32 : 0)
 
+int umount_main(int argc, char **argv);
 int umount_main(int argc, char **argv)
 {
        int doForce;
        char path[2*PATH_MAX];
        struct mntent me;
        FILE *fp;
-       int status=EXIT_SUCCESS;
-       unsigned long opt;
+       char *fstype = 0;
+       int status = EXIT_SUCCESS;
+       unsigned opt;
        struct mtab_list {
                char *dir;
                char *device;
@@ -46,7 +38,7 @@ int umount_main(int argc, char **argv)
 
        /* Parse any options */
 
-       opt = bb_getopt_ulflags (argc, argv, OPTION_STRING);
+       opt = getopt32(argc, argv, OPTION_STRING, &fstype);
 
        argc -= optind;
        argv += optind;
@@ -59,90 +51,104 @@ int umount_main(int argc, char **argv)
         * entry.  Notice that this also naturally reverses the list so that -a
         * umounts the most recent entries first. */
 
-       m=mtl=0;
-       if(opt & OPT_ALL) {
-
-               /* 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). */
-
-               if(!(fp = setmntent(bb_path_mtab_file, "r")))
-                       bb_error_msg_and_die("Cannot open %s", bb_path_mtab_file);
-               while (getmntent_r(fp,&me,path,sizeof(path))) {
-                       m=xmalloc(sizeof(struct mtab_list));
-                       m->next=mtl;
-                       m->device=bb_xstrdup(me.mnt_fsname);
-                       m->dir=bb_xstrdup(me.mnt_dir);
-                       mtl=m;
+       m = mtl = 0;
+
+       /* 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("cannot open %s", bb_path_mtab_file);
+       } else {
+               while (getmntent_r(fp, &me, path, sizeof(path))) {
+                       /* Match fstype if passed */
+                       if (fstype && match_fstype(&me, fstype))
+                               continue;
+                       m = xmalloc(sizeof(struct mtab_list));
+                       m->next = mtl;
+                       m->device = xstrdup(me.mnt_fsname);
+                       m->dir = xstrdup(me.mnt_dir);
+                       mtl = m;
                }
                endmntent(fp);
+       }
+
+       /* If we're not umounting all, we need at least one argument. */
+       if (!(opt & OPT_ALL) && !fstype) {
+               m = 0;
+               if (!argc) bb_show_usage();
+       }
 
-       /* If we're not mounting all, we need at least one argument. */
-       } else if(argc <= 0) bb_show_usage();
-       
        // Loop through everything we're supposed to umount, and do so.
-       for(;;) {
+       for (;;) {
                int curstat;
+               char *zapit = *argv;
 
                // Do we already know what to umount this time through the loop?
-               if(m) safe_strncpy(path,m->dir,PATH_MAX);
+               if (m) safe_strncpy(path, m->dir, PATH_MAX);
                // For umount -a, end of mtab means time to exit.
-               else if(opt & OPT_ALL) break;
+               else if (opt & OPT_ALL) break;
                // Get next command line argument (and look it up in mtab list)
-               else if(!argc--) break;
+               else if (!argc--) break;
                else {
-                       // Get next command line argument (and look it up in mtab list)
-                       realpath(*argv++, path);
-                       if (ENABLE_FEATURE_MTAB_SUPPORT)
-                               for(m = mtl; m; m = m->next)
-                                       if(!strcmp(path, m->dir) || !strcmp(path, m->device))
-                                               break;
+                       argv++;
+                       realpath(zapit, path);
+                       for (m = mtl; m; m = m->next)
+                               if (!strcmp(path, m->dir) || !strcmp(path, m->device))
+                                       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(path);
+               curstat = umount(zapit);
 
                // Force the unmount, if necessary.
-               if(curstat && doForce) {
-                       curstat = umount2(path, doForce);
-                       if(curstat)
-                               bb_error_msg_and_die("forced umount of %s failed!", path);
+               if (curstat && doForce) {
+                       curstat = umount2(zapit, doForce);
+                       if (curstat)
+                               bb_error_msg("forced umount of %s failed!", zapit);
                }
 
                // If still can't umount, maybe remount read-only?
                if (curstat && (opt & OPT_REMOUNT) && errno == EBUSY && m) {
-                       curstat = mount(m->device, path, NULL, MS_REMOUNT|MS_RDONLY, NULL);
-                       bb_error_msg(curstat ? "Cannot remount %s read-only" :
+                       curstat = mount(m->device, zapit, NULL, MS_REMOUNT|MS_RDONLY, NULL);
+                       bb_error_msg(curstat ? "cannot remount %s read-only" :
                                                 "%s busy - remounted read-only", m->device);
                }
 
-               /* De-allocate the loop device.  This ioctl should be ignored on any
-                * non-loop block devices. */
-               if(ENABLE_FEATURE_MOUNT_LOOP && !(opt & OPT_DONTFREELOOP) && m)
-                       del_loop(m->device);
-
-               if(curstat) {
-                       /* Yes, the ENABLE is redundant here, but the optimizer for ARM
-                        * can't do simple constant propagation in local variables... */
-                       if(ENABLE_FEATURE_MTAB_SUPPORT && !(opt & OPT_NO_MTAB) && m)
-                               erase_mtab(m->dir);
+               if (curstat) {
                        status = EXIT_FAILURE;
-                       bb_perror_msg("Couldn't umount %s", path);
+                       bb_perror_msg("cannot umount %s", 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_DONTFREELOOP) && 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
-               while(m && (m = m->next))
-                       if((opt & OPT_ALL) || !strcmp(path,m->device))
+               // Note this means that "umount /dev/blah" will unmount all instances
+               // of /dev/blah, not just the most recent.
+               while (m && (m = m->next))
+                       if ((opt & OPT_ALL) || !strcmp(path, m->device))
                                break;
        }
 
        // Free mtab list if necessary
 
-       if(ENABLE_FEATURE_CLEAN_UP) {
-               while(mtl) {
-                       m=mtl->next;
+       if (ENABLE_FEATURE_CLEAN_UP) {
+               while (mtl) {
+                       m = mtl->next;
                        free(mtl->device);
                        free(mtl->dir);
                        free(mtl);
-                       mtl=m;
+                       mtl = m;
                }
        }