More shell features.... if-then-else-fi is now basically usable (disable
[oweals/busybox.git] / umount.c
index 9ad6f26c214d38cf107aca088acbc80c39f900f2..6a5e4cad8bc70c5caf0af91ef7e6b251ee3c3dbc 100644 (file)
--- a/umount.c
+++ b/umount.c
@@ -1,8 +1,9 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini umount implementation for busybox
  *
  *
- * Copyright (C) 1999 by Lineo, inc.
+ * Copyright (C) 1999,2000 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
  * This program is free software; you can redistribute it and/or modify
 
 #include "internal.h"
 #include <stdio.h>
-#include <sys/mount.h>
 #include <mntent.h>
-#include <fstab.h>
 #include <errno.h>
 
-#if defined BB_FEATURE_MOUNT_LOOP
-#include <fcntl.h>
-#include <sys/ioctl.h>
-#include <linux/loop.h>
 
-static int del_loop(const char *device);
-#endif
+#define MNT_FORCE              1
+#define MS_MGC_VAL             0xc0ed0000 /* Magic number indicatng "new" flags */
+#define MS_REMOUNT             32      /* Alter flags of a mounted FS.  */
+#define MS_RDONLY              1       /* Mount read-only.  */
 
-static const char umount_usage[] = 
-"umount [flags] filesystem|directory\n\n"
-"Flags:\n"
-"\t-a:\tUnmount all file systems"
-#ifdef BB_MTAB
-" in /etc/mtab\n\t-n:\tDon't erase /etc/mtab entries\n"
-#else
-"\n"
-#endif
-;
+extern int mount (__const char *__special_file, __const char *__dir,
+                       __const char *__fstype, unsigned long int __rwflag,
+                       __const void *__data);
+extern int umount (__const char *__special_file);
+extern int umount2 (__const char *__special_file, int __flags);
+
+struct _mtab_entry_t {
+       char *device;
+       char *mountpt;
+       struct _mtab_entry_t *next;
+};
 
+static struct _mtab_entry_t *mtab_cache = NULL;
 
+
+
+#if defined BB_FEATURE_MOUNT_FORCE
+static int doForce = FALSE;
+#endif
+#if defined BB_FEATURE_MOUNT_LOOP
+static int freeLoop = TRUE;
+#endif
 static int useMtab = TRUE;
 static int umountAll = FALSE;
-extern const char mtab_file[]; /* Defined in utility.c */
+static int doRemount = FALSE;
+extern const char mtab_file[]; /* Defined in utility.c */
+
 
-static int 
-do_umount(const char* name, int useMtab)
+
+/* These functions are here because the getmntent functions do not appear
+ * to be re-entrant, which leads to all sorts of problems when we try to
+ * use them recursively - randolph
+ *
+ * TODO: Perhaps switch to using Glibc's getmntent_r
+ *        -Erik
+ */
+void mtab_read(void)
 {
-    int status;
+       struct _mtab_entry_t *entry = NULL;
+       struct mntent *e;
+       FILE *fp;
 
-#if defined BB_FEATURE_MOUNT_LOOP
-    /* check to see if this is a loop device */
-    struct stat fst;
-    char dev[20];
-    const char *oldname = NULL;
-    int i;
-    
-    if (stat(name, &fst)) {
-       fprintf(stderr, "umount: %s: %s\n", name, strerror(errno));
-       exit(1);
-    }
-    for (i = 0 ; i <= 7 ; i++) {
-       struct stat lst;
-       sprintf(dev, "/dev/loop%d", i);
-       if (stat(dev, &lst))
-           continue;
-       if (lst.st_dev == fst.st_dev) {
-           oldname = name;
-           name = dev;
-           break;
+       if (mtab_cache != NULL)
+               return;
+
+       if ((fp = setmntent(mtab_file, "r")) == NULL) {
+               errorMsg("Cannot open %s\n", mtab_file);
+               return;
+       }
+       while ((e = getmntent(fp))) {
+               entry = xmalloc(sizeof(struct _mtab_entry_t));
+               entry->device = strdup(e->mnt_fsname);
+               entry->mountpt = strdup(e->mnt_dir);
+               entry->next = mtab_cache;
+               mtab_cache = entry;
+       }
+       endmntent(fp);
+}
+
+char *mtab_getinfo(const char *match, const char which)
+{
+       struct _mtab_entry_t *cur = mtab_cache;
+
+       while (cur) {
+               if (strcmp(cur->mountpt, match) == 0 ||
+                       strcmp(cur->device, match) == 0) {
+                       if (which == MTAB_GETMOUNTPT) {
+                               return cur->mountpt;
+                       } else {
+#if !defined BB_MTAB
+                               if (strcmp(cur->device, "/dev/root") == 0) {
+                                       /* Adjusts device to be the real root device,
+                                        * or leaves device alone if it can't find it */
+                                       find_real_root_device_name( cur->device);
+                                       return ( cur->device);
+                               }
+#endif
+                               return cur->device;
+                       }
+               }
+               cur = cur->next;
+       }
+       return NULL;
+}
+
+char *mtab_first(void **iter)
+{
+       struct _mtab_entry_t *mtab_iter;
+
+       if (!iter)
+               return NULL;
+       mtab_iter = mtab_cache;
+       *iter = (void *) mtab_iter;
+       return mtab_next(iter);
+}
+
+char *mtab_next(void **iter)
+{
+       char *mp;
+
+       if (iter == NULL || *iter == NULL)
+               return NULL;
+       mp = ((struct _mtab_entry_t *) (*iter))->mountpt;
+       *iter = (void *) ((struct _mtab_entry_t *) (*iter))->next;
+       return mp;
+}
+
+/* Don't bother to clean up, since exit() does that 
+ * automagically, so we can save a few bytes */
+#ifdef BB_FEATURE_CLEAN_UP
+void mtab_free(void)
+{
+       struct _mtab_entry_t *this, *next;
+
+       this = mtab_cache;
+       while (this) {
+               next = this->next;
+               if (this->device)
+                       free(this->device);
+               if (this->mountpt)
+                       free(this->mountpt);
+               free(this);
+               this = next;
        }
-    }
+}
 #endif
 
-    status = umount(name);
+static int do_umount(const char *name, int useMtab)
+{
+       int status;
+       char *blockDevice = mtab_getinfo(name, MTAB_GETDEVICE);
+
+       if (blockDevice && strcmp(blockDevice, name) == 0)
+               name = mtab_getinfo(blockDevice, MTAB_GETMOUNTPT);
+
+       status = umount(name);
 
 #if defined BB_FEATURE_MOUNT_LOOP
-    if (!strncmp("/dev/loop", name, 9)) { /* this was a loop device, delete it */
-       del_loop(name);
-       if (oldname != NULL)
-           name = oldname;
-    }
+       if (freeLoop == TRUE && blockDevice != NULL && !strncmp("/dev/loop", blockDevice, 9))
+               /* this was a loop device, delete it */
+               del_loop(blockDevice);
 #endif
+#if defined BB_FEATURE_MOUNT_FORCE
+       if (status != 0 && doForce == TRUE) {
+               status = umount2(blockDevice, MNT_FORCE);
+               if (status != 0) {
+                       fatalError("forced umount of %s failed!\n", blockDevice);
+               }
+       }
+#endif
+       if (status != 0 && doRemount == TRUE && errno == EBUSY) {
+               status = mount(blockDevice, name, NULL,
+                                          MS_MGC_VAL | MS_REMOUNT | MS_RDONLY, NULL);
+               if (status == 0) {
+                       errorMsg("%s busy - remounted read-only\n", blockDevice);
+               } else {
+                       errorMsg("Cannot remount %s read-only\n", blockDevice);
+               }
+       }
+       if (status == 0) {
 #if defined BB_MTAB
-    if ( status == 0 ) {
-       if ( useMtab==TRUE )
-           erase_mtab(name);
-       return 0;
-    }
-    else
+               if (useMtab == TRUE)
+                       erase_mtab(name);
 #endif
-       return(status);
+               return (TRUE);
+       }
+       return (FALSE);
 }
 
-static int
-umount_all(int useMtab)
+static int umount_all(int useMtab)
 {
-       int status;
-       struct mntent *m;
-        FILE *mountTable;
-
-        if ((mountTable = setmntent (mtab_file, "r"))) {
-            while ((m = getmntent (mountTable)) != 0) {
-                char *blockDevice = m->mnt_fsname;
-#if ! defined BB_MTAB
-               if (strcmp (blockDevice, "/dev/root") == 0) {
-                   struct fstab* fstabItem;
-                   fstabItem = getfsfile ("/");
-                   if (fstabItem != NULL) {
-                       blockDevice = fstabItem->fs_spec;
-                   }
-               }
-#endif
-               /* Don't umount /proc when doing umount -a */
-                if (strcmp (blockDevice, "proc") == 0)
-                   continue;
-
-               status=do_umount (m->mnt_dir, useMtab);
-               if (status!=0) {
-                   /* Don't bother retrying the umount on busy devices */
-                   if (errno==EBUSY) {
-                       perror(m->mnt_dir); 
+       int status = TRUE;
+       char *mountpt;
+       void *iter;
+
+       for (mountpt = mtab_first(&iter); mountpt; mountpt = mtab_next(&iter)) {
+               /* Never umount /proc on a umount -a */
+               if (strstr(mountpt, "proc")!= NULL)
                        continue;
-                   }
-                   status=do_umount (blockDevice, useMtab);
-                   if (status!=0) {
-                       printf ("Couldn't umount %s on %s (type %s): %s\n", 
-                               blockDevice, m->mnt_dir, m->mnt_type, strerror(errno));
-                   }
+               status = do_umount(mountpt, useMtab);
+               if (status != 0) {
+                       /* Don't bother retrying the umount on busy devices */
+                       if (errno == EBUSY) {
+                               perror(mountpt);
+                               continue;
+                       }
+                       status = do_umount(mountpt, useMtab);
+                       if (status != 0) {
+                               printf("Couldn't umount %s on %s: %s\n",
+                                          mountpt, mtab_getinfo(mountpt, MTAB_GETDEVICE),
+                                          strerror(errno));
+                       }
                }
-            }
-            endmntent (mountTable);
-        }
-        return( TRUE);
+       }
+       return (status);
 }
 
-extern int
-umount_main(int argc, char** argv)
+extern int umount_main(int argc, char **argv)
 {
-    if (argc < 2) {
-       usage( umount_usage); 
-    }
-
-    /* Parse any options */
-    while (--argc > 0 && **(++argv) == '-') {
-       while (*++(*argv)) switch (**argv) {
-           case 'a':
-               umountAll = TRUE;
-               break;
+       if (argc < 2) {
+               usage(umount_usage);
+       }
+#ifdef BB_FEATURE_CLEAN_UP
+       atexit(mtab_free);
+#endif
+
+       /* Parse any options */
+       while (--argc > 0 && **(++argv) == '-') {
+               while (*++(*argv))
+                       switch (**argv) {
+                       case 'a':
+                               umountAll = TRUE;
+                               break;
+#if defined BB_FEATURE_MOUNT_LOOP
+                       case 'l':
+                               freeLoop = FALSE;
+                               break;
+#endif
 #ifdef BB_MTAB
-           case 'n':
-               useMtab = FALSE;
-               break;
+                       case 'n':
+                               useMtab = FALSE;
+                               break;
 #endif
-           default:
-               usage( umount_usage);
+#ifdef BB_FEATURE_MOUNT_FORCE
+                       case 'f':
+                               doForce = TRUE;
+                               break;
+#endif
+                       case 'r':
+                               doRemount = TRUE;
+                               break;
+                       case 'v':
+                               break; /* ignore -v */
+                       default:
+                               usage(umount_usage);
+                       }
        }
-    }
-
 
-    if(umountAll==TRUE) {
-       exit(umount_all(useMtab));
-    }
-    if ( do_umount(*argv,useMtab) == 0 )
-       exit (TRUE);
-    else {
+       mtab_read();
+       if (umountAll == TRUE) {
+               exit(umount_all(useMtab));
+       }
+       if (do_umount(*argv, useMtab) == 0)
+               exit(TRUE);
        perror("umount");
-       exit(FALSE);
-    }
+       return(FALSE);
 }
 
-#if defined BB_FEATURE_MOUNT_LOOP
-static int del_loop(const char *device)
-{
-       int fd;
-
-       if ((fd = open(device, O_RDONLY)) < 0) {
-               perror(device);
-               exit(1);
-       }
-       if (ioctl(fd, LOOP_CLR_FD, 0) < 0) {
-               perror("ioctl: LOOP_CLR_FD");
-               exit(1);
-       }
-       close(fd);
-       return(0);
-}
-#endif